From eaf30bad1eef8493704ac1d76ce431392babcc27 Mon Sep 17 00:00:00 2001 From: Lukas Holecek Date: Feb 26 2020 15:39:47 +0000 Subject: Fix repr() for GenericSubjectType Signed-off-by: Lukas Holecek --- diff --git a/greenwave/subjects/subject_type.py b/greenwave/subjects/subject_type.py index 916e978..cd0fff5 100644 --- a/greenwave/subjects/subject_type.py +++ b/greenwave/subjects/subject_type.py @@ -72,6 +72,9 @@ class GenericSubjectType: for name, attr in SubjectType.safe_yaml_attributes.items(): self.__setattr__(name, attr.default_value) + def __repr__(self): + return ''.format(self.id) + def load_subject_types(subject_types_dir): """ diff --git a/greenwave/tests/test_subjects.py b/greenwave/tests/test_subjects.py index cbedac1..808e026 100644 --- a/greenwave/tests/test_subjects.py +++ b/greenwave/tests/test_subjects.py @@ -60,3 +60,8 @@ def test_subject_to_str(app): def test_subject_to_repr(app): subject = create_subject('koji_build', 'some_nvr') assert repr(subject) == "Subject(, 'some_nvr')" + + +def test_subject_to_repr_generic(app): + subject = create_subject('some_type', 'some_nvr') + assert repr(subject) == "Subject(, 'some_nvr')"