def test_get_saved_queries():
client = DummyConnect()
args = {"type": "users"}
result = run_command(client=client,
args=args,
command="axonius-get-saved-queries")
assert len(EXPECTED_USERS_SQS) == len(result.outputs)
def test_remove_tags():
client = DummyConnect()
args = {"type": "users", "ids": DUMMY_USER_IDS, "tag_name": "test"}
result = run_command(client=client,
args=args,
command="axonius-remove-tag")
assert len(DUMMY_USER_IDS) == result.outputs
def test_client():
"""Pass."""
client = DummyConnect()
expected = "ok"
args = {}
result = run_command(client=client, args=args, command="test-module")
assert expected == result
def test_get_device():
client = DummyConnect()
args = {"value": "DESKTOP-Gary-Gaither"}
result = run_command(client=client,
args=args,
command="axonius-get-devices-by-hostname")
assert EXPECTED_DEVICE["internal_axon_id"] == result.outputs[
"internal_axon_id"]
def test_add_tags():
client = DummyConnect()
args = {"type": "devices", "ids": DUMMY_DEVICES_IDS, "tag_name": "test"}
result = run_command(client=client, args=args, command="axonius-add-tag")
assert len(DUMMY_DEVICES_IDS) == result.outputs
def test_get_tags():
client = DummyConnect()
args = {"type": "devices"}
result = run_command(client=client, args=args, command="axonius-get-tags")
assert EXPECTED_DEVICE_TAGS == result.outputs