tests: Modify unit test for reverse schema generation

This commit is contained in:
Shreya Malviya 2021-10-14 14:16:16 +05:30
parent faa4c18cab
commit ae6ebcf3c7
1 changed files with 5 additions and 2 deletions

View File

@ -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
)