From ae6ebcf3c79abaa86607dab1e9f8f04224fb5f57 Mon Sep 17 00:00:00 2001 From: Shreya Malviya Date: Thu, 14 Oct 2021 14:16:16 +0530 Subject: [PATCH] tests: Modify unit test for reverse schema generation --- .../test_config_schema_per_attack_technique.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/monkey/tests/unit_tests/monkey_island/cc/services/config_schema/test_config_schema_per_attack_technique.py b/monkey/tests/unit_tests/monkey_island/cc/services/config_schema/test_config_schema_per_attack_technique.py index 02010ac30..86383366e 100644 --- a/monkey/tests/unit_tests/monkey_island/cc/services/config_schema/test_config_schema_per_attack_technique.py +++ b/monkey/tests/unit_tests/monkey_island/cc/services/config_schema/test_config_schema_per_attack_technique.py @@ -1,5 +1,5 @@ from monkey_island.cc.services.config_schema.config_schema_per_attack_technique import ( - get_config_schema_per_attack_technique, + ConfigSchemaPerAttackTechnique, ) REVERSE_FAKE_SCHEMA = { @@ -19,4 +19,7 @@ REVERSE_FAKE_SCHEMA = { def test_get_config_schema_per_attack_technique(monkeypatch, fake_schema): - assert get_config_schema_per_attack_technique(fake_schema) == REVERSE_FAKE_SCHEMA + assert ( + ConfigSchemaPerAttackTechnique().get_config_schema_per_attack_technique(fake_schema) + == REVERSE_FAKE_SCHEMA + )