def test_get_last_generation_id_not_exist(datasource): # noqa: F811
"""Does it error out if we ask for a bogus taskmanager?"""
with pytest.raises((KeyError, NoResultFound)):
datasource.get_last_generation_id(taskmanager_name="no_such_task_manager")
with pytest.raises((KeyError, NoResultFound)):
datasource.get_last_generation_id(
taskmanager_name="no_such_task_manager",
taskmanager_id="11111111-1111-1111-1111-111111111111",
)
def test_get_last_generation_id(datasource): # noqa: F811
"""Can we get the last generation id by name or name and uuid"""
result1 = datasource.get_last_generation_id(taskmanager_name="taskmanager1")
assert result1 == 1
result1 = datasource.get_last_generation_id(
taskmanager_name="taskmanager1",
taskmanager_id="11111111-1111-1111-1111-111111111111",
)
assert result1 == 1
result2 = datasource.get_last_generation_id(
taskmanager_name="taskmanager2",
taskmanager_id="22222222-2222-2222-2222-222222222222",
)
assert result2 == 2
def test_duplicate_datablock(datasource): # noqa: F811
"""Can we duplicate taskmanager1 and all its entries"""
result1 = datasource.get_last_generation_id(
taskmanager_name="taskmanager1",
taskmanager_id="11111111-1111-1111-1111-111111111111",
)
assert result1 == 1
result1 = datasource.get_dataproducts(taskmanager_id=1, )
assert len(result1) == 2
datasource.duplicate_datablock(1, 1, 90)
result1 = datasource.get_last_generation_id(
taskmanager_name="taskmanager1",
taskmanager_id="11111111-1111-1111-1111-111111111111",
)
assert result1 == 90
def test_get_datablock(datasource): # noqa: F811
tm = datasource.get_taskmanager(taskmanager_name="taskmanager1")
gen_id = datasource.get_last_generation_id(taskmanager_name="taskmanager1")
result1 = datasource.get_datablock(tm["sequence_id"], gen_id)
assert result1 == {
"my_test_key": b"my_test_value",
"a_test_key": b"a_test_value"
}
