mirror of
https://github.com/famedly/ansible-collection-matrix
synced 2025-01-09 08:08:43 +00:00
352 lines
13 KiB
Python
352 lines
13 KiB
Python
from __future__ import absolute_import, division, print_function, annotations
|
|
|
|
import types
|
|
|
|
import pytest
|
|
from _pytest.monkeypatch import MonkeyPatch
|
|
|
|
from ansible_collections.famedly.matrix.plugins.module_utils import matrix
|
|
from ansible_collections.famedly.matrix.plugins.modules import matrix_member
|
|
|
|
from ansible_collections.famedly.matrix.tests.unit.mock_nio.MatrixNioBase import (
|
|
MatrixNioBase,
|
|
)
|
|
from ansible_collections.famedly.matrix.tests.unit.mock_nio.MatrixNioSuccess import (
|
|
MatrixNioSuccess,
|
|
)
|
|
from ansible_collections.famedly.matrix.tests.unit.mock_nio.utils.RoomSimulator import (
|
|
RoomSimulator,
|
|
)
|
|
from ansible_collections.famedly.matrix.tests.unit.mock_nio.room import failure
|
|
|
|
from ansible_collections.famedly.matrix.tests.unit.utils import (
|
|
AnsibleExitJson,
|
|
AnsibleFailJson,
|
|
assert_expression,
|
|
set_module_args,
|
|
exit_json,
|
|
fail_json,
|
|
)
|
|
|
|
|
|
class TestAnsibleModuleMatrixMember:
|
|
@staticmethod
|
|
def patchAnsibleNioModule(
|
|
monkeypatch: MonkeyPatch, mock_class: type(MatrixNioBase)
|
|
):
|
|
# Mock ansible functions
|
|
monkeypatch.setattr(matrix.AnsibleModule, "exit_json", exit_json)
|
|
monkeypatch.setattr(matrix.AnsibleModule, "fail_json", fail_json)
|
|
# Mock MatrixNio
|
|
for method in MatrixNioBase.__dict__:
|
|
if isinstance(
|
|
getattr(mock_class, method),
|
|
(types.FunctionType, types.BuiltinFunctionType),
|
|
):
|
|
monkeypatch.setattr(
|
|
matrix.AsyncClient, method, getattr(mock_class, method)
|
|
)
|
|
monkeypatch.setattr(matrix.AsyncClient, "logged_in", False)
|
|
# Setup RoomSimulator
|
|
simulator = RoomSimulator()
|
|
simulator.add_room("!myroomid:matrix.example.tld")
|
|
simulator.m_room_create("!myroomid:matrix.example.tld", "@admin:example.tld")
|
|
simulator.m_room_member(
|
|
room_id="!myroomid:matrix.example.tld",
|
|
membership="join",
|
|
sender="@myuser:matrix.example.tld",
|
|
state_key="@user1:matrix.example.tld",
|
|
)
|
|
simulator.m_room_member(
|
|
room_id="!myroomid:matrix.example.tld",
|
|
membership="join",
|
|
sender="@myuser:matrix.example.tld",
|
|
state_key="@user2:matrix.example.tld",
|
|
)
|
|
simulator.m_room_member(
|
|
room_id="!myroomid:matrix.example.tld",
|
|
membership="ban",
|
|
sender="@myuser:matrix.example.tld",
|
|
state_key="@user0:matrix.example.tld",
|
|
)
|
|
monkeypatch.setenv("ROOM_SIMULATOR", simulator.export())
|
|
|
|
def test_check_mode(self, monkeypatch):
|
|
self.patchAnsibleNioModule(monkeypatch, MatrixNioSuccess)
|
|
set_module_args(
|
|
{
|
|
"hs_url": "matrix.example.tld",
|
|
"user_id": "myuser",
|
|
"password": "supersecretpassword",
|
|
"room_id": "!myroomid:matrix.example.tld",
|
|
"state": "member",
|
|
"user_ids": ["@user1:matrix.example.tld", "@user2:matrix.example.tld"],
|
|
},
|
|
check_mode=True
|
|
)
|
|
with pytest.raises(AnsibleExitJson) as result:
|
|
matrix_member.main()
|
|
ansible_result = result.value.result
|
|
assert_expression(ansible_result["changed"] is True)
|
|
assert_expression(ansible_result["banned"] == [])
|
|
assert_expression(ansible_result["unbanned"] == [])
|
|
assert_expression(ansible_result["kicked"] == [])
|
|
assert_expression(ansible_result["invited"] == [])
|
|
assert_expression(
|
|
list(ansible_result["members"])
|
|
== ["@user1:matrix.example.tld", "@user2:matrix.example.tld"]
|
|
)
|
|
|
|
def test_no_changes(self, monkeypatch):
|
|
self.patchAnsibleNioModule(monkeypatch, MatrixNioSuccess)
|
|
set_module_args(
|
|
{
|
|
"hs_url": "matrix.example.tld",
|
|
"user_id": "myuser",
|
|
"password": "supersecretpassword",
|
|
"room_id": "!myroomid:matrix.example.tld",
|
|
"state": "member",
|
|
"user_ids": ["@user1:matrix.example.tld", "@user2:matrix.example.tld"],
|
|
}
|
|
)
|
|
with pytest.raises(AnsibleExitJson) as result:
|
|
matrix_member.main()
|
|
ansible_result = result.value.result
|
|
assert_expression(ansible_result["changed"] is False)
|
|
assert_expression(ansible_result["banned"] == [])
|
|
assert_expression(ansible_result["unbanned"] == [])
|
|
assert_expression(ansible_result["kicked"] == [])
|
|
assert_expression(ansible_result["invited"] == [])
|
|
assert_expression(
|
|
list(ansible_result["members"])
|
|
== ["@user1:matrix.example.tld", "@user2:matrix.example.tld"]
|
|
)
|
|
|
|
def test_add_user(self, monkeypatch):
|
|
self.patchAnsibleNioModule(monkeypatch, MatrixNioSuccess)
|
|
set_module_args(
|
|
{
|
|
"hs_url": "matrix.example.tld",
|
|
"user_id": "myuser",
|
|
"password": "supersecretpassword",
|
|
"room_id": "!myroomid:matrix.example.tld",
|
|
"state": "member",
|
|
"user_ids": [
|
|
"@user1:matrix.example.tld",
|
|
"@user2:matrix.example.tld",
|
|
"@user3:matrix.example.tld",
|
|
],
|
|
}
|
|
)
|
|
with pytest.raises(AnsibleExitJson) as result:
|
|
matrix_member.main()
|
|
ansible_result = result.value.result
|
|
assert_expression(ansible_result["changed"] is True)
|
|
assert_expression(ansible_result["banned"] == [])
|
|
assert_expression(ansible_result["unbanned"] == [])
|
|
assert_expression(ansible_result["kicked"] == [])
|
|
assert_expression(ansible_result["invited"] == ["@user3:matrix.example.tld"])
|
|
assert_expression(
|
|
list(ansible_result["members"])
|
|
== [
|
|
"@user1:matrix.example.tld",
|
|
"@user2:matrix.example.tld",
|
|
"@user3:matrix.example.tld",
|
|
]
|
|
)
|
|
|
|
def test_kick_user(self, monkeypatch):
|
|
self.patchAnsibleNioModule(monkeypatch, MatrixNioSuccess)
|
|
set_module_args(
|
|
{
|
|
"hs_url": "matrix.example.tld",
|
|
"user_id": "myuser",
|
|
"password": "supersecretpassword",
|
|
"room_id": "!myroomid:matrix.example.tld",
|
|
"state": "kicked",
|
|
"user_ids": ["@user1:matrix.example.tld"],
|
|
}
|
|
)
|
|
with pytest.raises(AnsibleExitJson) as result:
|
|
matrix_member.main()
|
|
ansible_result = result.value.result
|
|
assert_expression(ansible_result["changed"] is True)
|
|
assert_expression(ansible_result["banned"] == [])
|
|
assert_expression(ansible_result["unbanned"] == [])
|
|
assert_expression(ansible_result["kicked"] == ["@user1:matrix.example.tld"])
|
|
assert_expression(ansible_result["invited"] == [])
|
|
assert_expression(
|
|
list(ansible_result["members"]) == ["@user2:matrix.example.tld"]
|
|
)
|
|
|
|
def test_ban_user(self, monkeypatch):
|
|
self.patchAnsibleNioModule(monkeypatch, MatrixNioSuccess)
|
|
set_module_args(
|
|
{
|
|
"hs_url": "matrix.example.tld",
|
|
"user_id": "myuser",
|
|
"password": "supersecretpassword",
|
|
"room_id": "!myroomid:matrix.example.tld",
|
|
"state": "banned",
|
|
"user_ids": ["@user1:matrix.example.tld"],
|
|
}
|
|
)
|
|
with pytest.raises(AnsibleExitJson) as result:
|
|
matrix_member.main()
|
|
ansible_result = result.value.result
|
|
assert_expression(ansible_result["changed"] is True)
|
|
assert_expression(ansible_result["banned"] == ["@user1:matrix.example.tld"])
|
|
assert_expression(ansible_result["unbanned"] == [])
|
|
assert_expression(ansible_result["kicked"] == [])
|
|
assert_expression(ansible_result["invited"] == [])
|
|
assert_expression(
|
|
list(ansible_result["members"]) == ["@user2:matrix.example.tld"]
|
|
)
|
|
|
|
def test_unban_user(self, monkeypatch):
|
|
self.patchAnsibleNioModule(monkeypatch, MatrixNioSuccess)
|
|
set_module_args(
|
|
{
|
|
"hs_url": "matrix.example.tld",
|
|
"user_id": "myuser",
|
|
"password": "supersecretpassword",
|
|
"room_id": "!myroomid:matrix.example.tld",
|
|
"state": "member",
|
|
"user_ids": ["@user0:matrix.example.tld"],
|
|
}
|
|
)
|
|
with pytest.raises(AnsibleExitJson) as result:
|
|
matrix_member.main()
|
|
ansible_result = result.value.result
|
|
assert_expression(ansible_result["changed"] is True)
|
|
assert_expression(ansible_result["banned"] == [])
|
|
assert_expression(ansible_result["unbanned"] == ["@user0:matrix.example.tld"])
|
|
assert_expression(ansible_result["kicked"] == [])
|
|
assert_expression(ansible_result["invited"] == ["@user0:matrix.example.tld"])
|
|
assert_expression(
|
|
list(ansible_result["members"])
|
|
== [
|
|
"@user1:matrix.example.tld",
|
|
"@user2:matrix.example.tld",
|
|
"@user0:matrix.example.tld",
|
|
]
|
|
)
|
|
|
|
def test_exclusive(self, monkeypatch):
|
|
self.patchAnsibleNioModule(monkeypatch, MatrixNioSuccess)
|
|
set_module_args(
|
|
{
|
|
"hs_url": "matrix.example.tld",
|
|
"user_id": "myuser",
|
|
"password": "supersecretpassword",
|
|
"room_id": "!myroomid:matrix.example.tld",
|
|
"state": "member",
|
|
"exclusive": True,
|
|
"user_ids": ["@user2:matrix.example.tld"],
|
|
}
|
|
)
|
|
with pytest.raises(AnsibleExitJson) as result:
|
|
matrix_member.main()
|
|
ansible_result = result.value.result
|
|
assert_expression(ansible_result["changed"] is True)
|
|
assert_expression(ansible_result["banned"] == [])
|
|
assert_expression(ansible_result["unbanned"] == [])
|
|
assert_expression(ansible_result["kicked"] == ["@user1:matrix.example.tld"])
|
|
assert_expression(ansible_result["invited"] == [])
|
|
assert_expression(
|
|
list(ansible_result["members"]) == ["@user2:matrix.example.tld"]
|
|
)
|
|
|
|
def test_add_user_fail(self, monkeypatch):
|
|
class TestClass(failure.RoomInvite, MatrixNioSuccess):
|
|
pass
|
|
|
|
self.patchAnsibleNioModule(monkeypatch, TestClass)
|
|
set_module_args(
|
|
{
|
|
"hs_url": "matrix.example.tld",
|
|
"user_id": "myuser",
|
|
"password": "supersecretpassword",
|
|
"room_id": "!myroomid:matrix.example.tld",
|
|
"state": "member",
|
|
"user_ids": [
|
|
"@user1:matrix.example.tld",
|
|
"@user2:matrix.example.tld",
|
|
"@user3:matrix.example.tld",
|
|
],
|
|
}
|
|
)
|
|
with pytest.raises(AnsibleFailJson) as result:
|
|
matrix_member.main()
|
|
|
|
def test_kick_user_fail(self, monkeypatch):
|
|
class TestClass(failure.RoomKick, MatrixNioSuccess):
|
|
pass
|
|
|
|
self.patchAnsibleNioModule(monkeypatch, TestClass)
|
|
set_module_args(
|
|
{
|
|
"hs_url": "matrix.example.tld",
|
|
"user_id": "myuser",
|
|
"password": "supersecretpassword",
|
|
"room_id": "!myroomid:matrix.example.tld",
|
|
"state": "kicked",
|
|
"user_ids": ["@user1:matrix.example.tld"],
|
|
}
|
|
)
|
|
with pytest.raises(AnsibleFailJson) as result:
|
|
matrix_member.main()
|
|
|
|
def test_ban_user_fail(self, monkeypatch):
|
|
class TestClass(failure.RoomBan, MatrixNioSuccess):
|
|
pass
|
|
|
|
self.patchAnsibleNioModule(monkeypatch, TestClass)
|
|
set_module_args(
|
|
{
|
|
"hs_url": "matrix.example.tld",
|
|
"user_id": "myuser",
|
|
"password": "supersecretpassword",
|
|
"room_id": "!myroomid:matrix.example.tld",
|
|
"state": "banned",
|
|
"user_ids": ["@user1:matrix.example.tld"],
|
|
}
|
|
)
|
|
with pytest.raises(AnsibleFailJson) as result:
|
|
matrix_member.main()
|
|
|
|
def test_unban_user_fail(self, monkeypatch):
|
|
class TestClass(failure.RoomUnban, MatrixNioSuccess):
|
|
pass
|
|
|
|
self.patchAnsibleNioModule(monkeypatch, TestClass)
|
|
set_module_args(
|
|
{
|
|
"hs_url": "matrix.example.tld",
|
|
"user_id": "myuser",
|
|
"password": "supersecretpassword",
|
|
"room_id": "!myroomid:matrix.example.tld",
|
|
"state": "member",
|
|
"user_ids": ["@user0:matrix.example.tld"],
|
|
}
|
|
)
|
|
with pytest.raises(AnsibleFailJson) as result:
|
|
matrix_member.main()
|
|
|
|
def test_get_state_fail(self, monkeypatch):
|
|
class TestClass(failure.RoomGetState, MatrixNioSuccess):
|
|
pass
|
|
|
|
self.patchAnsibleNioModule(monkeypatch, TestClass)
|
|
set_module_args(
|
|
{
|
|
"hs_url": "matrix.example.tld",
|
|
"user_id": "myuser",
|
|
"password": "supersecretpassword",
|
|
"room_id": "!myroomid:matrix.example.tld",
|
|
"state": "member",
|
|
"user_ids": ["@user0:matrix.example.tld"],
|
|
}
|
|
)
|
|
with pytest.raises(AnsibleFailJson) as result:
|
|
matrix_member.main()
|