def test_json(read_database: ReadDatabase, connection: ConnectionHandler):
value = MagicMock()
connection.to_json = tj = MagicMock(return_value=value)
data = MagicMock()
assert read_database.json(data) == value
tj.assert_called_with(data)
def test_create_or_update_models(read_database: ReadDatabase,
connection: ConnectionHandler):
fqid1 = MagicMock()
model1 = MagicMock()
fqid2 = MagicMock()
model2 = MagicMock()
models = {fqid1: model1, fqid2: model2}
connection.execute = e = MagicMock()
read_database.json = lambda x: x
read_database.create_or_update_models(models)
args = e.call_args.args[1]
assert (args == [fqid1, model1, fqid2, model2
]) or (args == [fqid2, model2, fqid1, model1])
