[TASK] Add tests for domain model - #40