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"]]
