Fix: default field generation
This commit is contained in:
parent
2dc1cf6fb8
commit
de9a4bc4be
@ -1,10 +1,9 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import uuid
|
||||
from dataclasses import dataclass
|
||||
from dataclasses import dataclass, field
|
||||
from datetime import datetime
|
||||
from typing import TYPE_CHECKING, Optional
|
||||
|
||||
from pydantic import Field
|
||||
from uuid import UUID, uuid4
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from backend.model.tribe import Tribe
|
||||
@ -14,7 +13,7 @@ if TYPE_CHECKING:
|
||||
class Student:
|
||||
name: str
|
||||
tribe: Tribe
|
||||
id: Optional[str] = Field(default_factory=lambda: str(uuid.uuid1()))
|
||||
id: str = field(default_factory=lambda: str(uuid4()))
|
||||
|
||||
def __post_init__(self) -> None:
|
||||
self.tribe.register_student(self)
|
||||
@ -26,3 +25,9 @@ class Student:
|
||||
|
||||
def __hash__(self) -> int:
|
||||
return hash(self.id)
|
||||
|
||||
def to_tuple(self) -> tuple:
|
||||
return (self.id, self.name, self.tribe.name)
|
||||
|
||||
def to_dict(self) -> dict:
|
||||
return {"id": self.id, "name": self.name, "tribe_name": self.tribe.name}
|
||||
|
@ -1,10 +1,24 @@
|
||||
from random import randint
|
||||
from uuid import UUID
|
||||
|
||||
from backend.model.student import Student
|
||||
from backend.model.tribe import Tribe
|
||||
from tests.model.fakes import build_assessments, build_tribes
|
||||
|
||||
|
||||
def test_init_student():
|
||||
tribe = build_tribes(1)[0]
|
||||
student = Student(name="Bob", tribe=tribe)
|
||||
print(student)
|
||||
|
||||
assert type(student.name) == str
|
||||
assert type(student.tribe) == Tribe
|
||||
assert type(student.id) == str
|
||||
|
||||
student2 = Student(name="Hop", tribe=tribe)
|
||||
assert student.id != student2.id
|
||||
|
||||
|
||||
def test_tribe_register_assessment():
|
||||
assessments_qty = randint(1, 10)
|
||||
tribes = build_tribes(1)
|
||||
|
Loading…
Reference in New Issue
Block a user