Skip to content

Commit

Permalink
start testing
Browse files Browse the repository at this point in the history
  • Loading branch information
ashimali committed Apr 26, 2024
1 parent ca821ff commit ea6892f
Show file tree
Hide file tree
Showing 2 changed files with 25 additions and 13 deletions.
24 changes: 11 additions & 13 deletions application/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
from enum import Enum
from typing import List, Optional

from pydantic import UUID4, BaseModel, Field
from pydantic import UUID4, BaseModel, ConfigDict, Field
from slugify import slugify
from sqlalchemy import UUID, Boolean, Date, DateTime, ForeignKey, Integer, Text, event
from sqlalchemy.dialects.postgresql import ARRAY, ENUM, JSONB
Expand Down Expand Up @@ -225,22 +225,23 @@ def __repr__(self):


class StageModel(BaseModel):
model_config = ConfigDict(from_attributes=True)

name: str
value: str

class Config:
from_attributes = True


class FrequencyOfUpdatesModel(BaseModel):
model_config = ConfigDict(from_attributes=True)

name: str
value: str

class Config:
from_attributes = True


class ConsiderationModel(BaseModel):

model_config = ConfigDict(from_attributes=True)

name: str
description: Optional[str]
synonyms: Optional[List[str]]
Expand All @@ -256,14 +257,11 @@ class ConsiderationModel(BaseModel):
legislation: Optional[dict]
slug: Optional[str]

class Config:
from_attributes = True


class AnswerModel(BaseModel):

model_config = ConfigDict(from_attributes=True)

text: str
consideration_id: UUID4 = Field(exclude=True)
question_slug: str

class Config:
from_attributes = True
14 changes: 14 additions & 0 deletions tests/test_questions.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
from application.commands import extract_values
from application.question_sets import questions


def test_prev_next_slugs_are_valid():

for stage in questions.keys():
qs = questions[stage]
slugs = set([next(iter(q.keys())) for q in qs])
next_prev_slugs = set([])
for q in qs:
extract_values(q, next_prev_slugs)

assert next_prev_slugs.issubset(slugs)

0 comments on commit ea6892f

Please sign in to comment.