Skip to content

Commit

Permalink
add properties to modelhelper
Browse files Browse the repository at this point in the history
  • Loading branch information
afuetterer committed Nov 3, 2023
1 parent 1289f28 commit 69aa5db
Showing 1 changed file with 25 additions and 12 deletions.
37 changes: 25 additions & 12 deletions rdmo/management/tests/test_frontend.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,19 +31,32 @@
class ModelHelper:
"""Helper class to bundle information about models for test cases."""

url: str
model: Model
form_field: str = "URI Path"
db_field: str = "uri_path"
has_nested: bool = False

@property
def url(self) -> str:
return f"{self.model._meta.model_name}s"

@property
def url_name(self) -> str:
"""Return the url name for the model."""
url = self.url.rstrip("s")
url = url.replace("set", " set")
return url

@property
def verbose_name(self) -> str:
"""Return the verbose_name for the model."""
return self.model._meta.verbose_name

@property
def verbose_name_plural(self) -> str:
"""Return the verbose_name_plural for the model."""
return self.model._meta.verbose_name_plural


@pytest.fixture(scope="function")
def e2e_tests_django_db_setup(django_db_setup, django_db_blocker, fixtures):
Expand Down Expand Up @@ -100,19 +113,19 @@ def test_management_navigation(logged_in_admin_user: Page) -> None:


model_helpers = (
ModelHelper("catalogs", Catalog, has_nested=True),
ModelHelper("sections", Section, has_nested=True),
ModelHelper("pages", PageModel, has_nested=True),
ModelHelper("questionsets", QuestionSet, has_nested=True),
ModelHelper("questions", Question),
ModelHelper(Catalog, has_nested=True),
ModelHelper(Section, has_nested=True),
ModelHelper(PageModel, has_nested=True),
ModelHelper(QuestionSet, has_nested=True),
ModelHelper(Question),
ModelHelper(
"attributes", Attribute, has_nested=True, form_field="Key", db_field="key"
Attribute, has_nested=True, form_field="Key", db_field="key"
),
ModelHelper("optionsets", OptionSet, has_nested=True),
ModelHelper("options", Option),
ModelHelper("conditions", Condition),
ModelHelper("tasks", Task),
ModelHelper("views", View),
ModelHelper(OptionSet, has_nested=True),
ModelHelper(Option),
ModelHelper(Condition),
ModelHelper(Task),
ModelHelper(View),
)


Expand Down

0 comments on commit 69aa5db

Please sign in to comment.