diff --git a/brewtils/schema_parser.py b/brewtils/schema_parser.py index ded758a6..86e7a22c 100644 --- a/brewtils/schema_parser.py +++ b/brewtils/schema_parser.py @@ -504,6 +504,7 @@ def parse_status_history(cls, status_history, from_string=False, **kwargs): from_string=from_string, **kwargs ) + @classmethod def parse_replication(cls, replication, from_string=False, **kwargs): """Convert raw JSON string or dictionary to a replication model object @@ -1099,7 +1100,7 @@ def serialize_status_history(cls, status_history, to_string=True, **kwargs): schema_name=brewtils.models.StatusHistory.schema, **kwargs ) - + @classmethod def serialize_replication(cls, replication, to_string=True, **kwargs): """Convert a replication model into serialized form diff --git a/brewtils/test/comparable.py b/brewtils/test/comparable.py index fb56d399..bcf4f5de 100644 --- a/brewtils/test/comparable.py +++ b/brewtils/test/comparable.py @@ -247,7 +247,6 @@ def assert_connection_equal(obj1, obj2, do_raise=False): assert_replication_equal = partial(_assert_wrapper, expected_type=Replication) - def assert_command_equal(obj1, obj2, do_raise=False): return _assert_wrapper( obj1, diff --git a/test/schema_parser_test.py b/test/schema_parser_test.py index ebe5920f..d49652ee 100644 --- a/test/schema_parser_test.py +++ b/test/schema_parser_test.py @@ -587,7 +587,7 @@ def test_single_specific_from_string(self): lazy_fixture("status_history_dict"), assert_status_history_equal, lazy_fixture("bg_status_history"), - ), + ), ( brewtils.models.Replication, lazy_fixture("replication_dict"), @@ -958,7 +958,7 @@ def test_single(self, model, expected): "serialize_status_history", lazy_fixture("bg_status_history"), lazy_fixture("status_history_dict"), - ), + ), ( "serialize_replication", lazy_fixture("bg_replication"),