def test_get_taskmanager_not_exists(datasource): # noqa: F811 """This should error out""" with pytest.raises((KeyError, NoResultFound)): datasource.get_taskmanager(taskmanager_name="no_such_task_manager") with pytest.raises((KeyError, NoResultFound)): datasource.get_taskmanager( taskmanager_name="no_such_task_manager", taskmanager_id="11111111-1111-1111-1111-111111111111", )
def test_delete_data_older_than_arg(datasource): # noqa: F811 """Can we delete old entries""" with pytest.raises((ValueError, NoResultFound), match="Argument has to be positive, non zero integer"): datasource.delete_data_older_than(0) with pytest.raises((ValueError, NoResultFound), match="Argument has to be positive, non zero integer"): datasource.delete_data_older_than(-5) # verify our old record is loaded result1 = datasource.get_taskmanager(taskmanager_name="taskmanager1") assert result1["name"] == "taskmanager1" # should test it deletes the old one datasource.delete_data_older_than(10) with pytest.raises((KeyError, NoResultFound)): datasource.get_taskmanager(taskmanager_name="taskmanager1")
def test_get_taskmanager_exists(datasource): # noqa: F811 """Can I get a taskmanager by name or name and uuid""" # should return the 'newest' instance result1 = datasource.get_taskmanager(taskmanager_name="taskmanager1") assert result1["name"] == "taskmanager1" assert str(result1["taskmanager_id"]) == "11111111-1111-1111-1111-111111111111" result2 = datasource.get_taskmanager( taskmanager_name="taskmanager1", taskmanager_id="11111111-1111-1111-1111-111111111111", ) assert result2["name"] == "taskmanager1" assert str(result2["taskmanager_id"]) == "11111111-1111-1111-1111-111111111111" assert result1 == result2
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" }