def test_fetch_models_mapped_fields(read_database: ReadDatabase, connection: ConnectionHandler): args = [MagicMock(), MagicMock(), MagicMock()] row = MagicMock() row.copy = lambda: row connection.query = q = MagicMock(return_value=[row]) models = read_database.fetch_models(*args, [1]) q.assert_called_with(*args) assert list(models.values()) == [row]
def test_fetch_models(read_database: ReadDatabase, connection: ConnectionHandler): args = [MagicMock(), MagicMock(), MagicMock()] row = MagicMock() row["data"] = MagicMock() connection.query = q = MagicMock(return_value=[row]) models = read_database.fetch_models(*args, []) q.assert_called_with(*args) assert list(models.values()) == [row["data"]]