"""Test for Split model.""" from datetime import datetime import pytest from collaborator.models.rds.split import Split from tests.testutils import db @db.test_schema_default_seed @pytest.mark.parametrize( "split_id, identifier, split_rate, split_type_id, " "collaborator_id, rate_type, created_date, expected_result", [ ( 123, "test", 0.8, 1, 2, "NET", datetime(2000, 12, 11, 10, 9, 8), { "id": 123, "identifier": "test", "split_rate": 0.8, "split_type_id": 1, "collaborator_id": 2, "rate_type": "NET", "created_date": "2000-12-11T10:09:08", "updated_date": None, "source": None, }, ), ( None, "nothing", 0.5, 3, 4, "GROSS", None, { "id": None, "identifier": "nothing", "split_rate": 0.5, "split_type_id": 3, "collaborator_id": 4, "rate_type": "GROSS", "created_date": None, "updated_date": None, "source": None, }, ), ], ) def test_to_dict( split_id, identifier, split_rate, split_type_id, collaborator_id, rate_type, created_date, expected_result, ): """Test get collaborators based on an account where entries exist.""" item = Split( split_id=split_id, identifier=identifier, split_rate=split_rate, split_type_id=split_type_id, collaborator_id=collaborator_id, rate_type=rate_type, created_date=created_date, ) assert item.to_dict() == expected_result