def test_insert(datasource): # noqa: F811
"""Can we insert new elements"""
primary_key = datasource.store_taskmanager("taskmanager3", "33333333-3333-3333-3333-333333333333")
assert primary_key > 1
header = Header(primary_key)
metadata = Metadata(primary_key)
datasource.insert(
primary_key,
1,
"sample_test_key",
b"sample_test_value",
header,
metadata,
)
result1 = datasource.get_dataproducts(taskmanager_id=primary_key)
assert result1 == [
{
"key": "sample_test_key",
"taskmanager_id": primary_key,
"generation_id": 1,
"value": b"sample_test_value",
}
]
result2 = datasource.get_dataproducts(taskmanager_id=primary_key, key="sample_test_key")
assert result1 == result2
def test_get_dataproducts(datasource): # noqa: F811
"""Can we get the dataproducts by uuid and uuid with key"""
result1 = datasource.get_dataproducts(taskmanager_id=1)
assert len(result1) == 2
assert result1[0] == {
"taskmanager_id": 1,
"generation_id": 1,
"key": "my_test_key",
"value": b"my_test_value",
}
assert result1[1] == {
"taskmanager_id": 1,
"generation_id": 1,
"key": "a_test_key",
"value": b"a_test_value",
}
result2 = datasource.get_dataproducts(taskmanager_id="2", key="other_test_key")
assert result2 == [
{
"taskmanager_id": 2,
"generation_id": 2,
"key": "other_test_key",
"value": b"other_test_value",
},
]
def test_get_dataproducts_not_exist(datasource): # noqa: F811
"""Does it error out if we ask for bogus information?"""
result = datasource.get_dataproducts(taskmanager_id=100)
assert result == []
result = datasource.get_dataproducts(taskmanager_id=2, key="no_such_key")
assert result == []
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
