"""Test for Recipient model.""" from datetime import datetime import pytest from collaborator.models.rds.recipient import Recipient @pytest.mark.parametrize( "fields, expected", [ ( { "recipient_id": 1, "transferwise_id": 1, "name": "SomeGuy", "currency": "USD", "created_by": "alw:111", "created_date": datetime(2019, 9, 8, 7, 6, 5), "deleted_by": None, "deleted_date": None, }, { "id": 1, "transferwise_id": 1, "name": "SomeGuy", "currency": "USD", "created_by": "alw:111", "created_date": datetime(2019, 9, 8, 7, 6, 5).isoformat(), "deleted_by": None, "deleted_date": None, }, ), ( { "recipient_id": 1, "transferwise_id": 1, "name": "SomeGuy", "currency": "USD", "created_by": "alw:111", "created_date": datetime(2019, 9, 8, 7, 6, 5), "deleted_date": datetime(2019, 9, 8, 7, 6, 5), "deleted_by": "alw:111", }, { "id": 1, "transferwise_id": 1, "name": "SomeGuy", "currency": "USD", "created_by": "alw:111", "created_date": datetime(2019, 9, 8, 7, 6, 5).isoformat(), "deleted_date": datetime(2019, 9, 8, 7, 6, 5).isoformat(), "deleted_by": "alw:111", }, ), ], ) def test_to_dict(fields, expected): """Test getting dictionary of model fields.""" item = Recipient(**fields) assert item.to_dict() == expected