def test_repo_contains_set(repo): assert not repo.contains_set('foo') artifact = artifact_record(id='123') repo.put(artifact) artifact_set = r.ArtifactSet([artifact.id], 'foo') repo.put_set(artifact_set) assert repo.contains_set(artifact_set.id)
def test_repo_delete_set(repo): artifact = artifact_record(id='123') repo.put(artifact) artifact_set = r.ArtifactSet(['123'], 'foo') repo.put_set(artifact_set) repo.delete_set(artifact_set.id) with pytest.raises(KeyError) as e: repo.get_set_by_id(artifact_set.id)
def test_repo_set_put_and_finding(repo): artifact = artifact_record(id='123') repo.put(artifact) artifact_set = r.ArtifactSet([artifact.id], 'foo') repo.put_set(artifact_set) assert repo.get_set_by_id(artifact_set.id) == artifact_set found_set = repo.get_set_by_labels('foo') assert found_set.name == 'foo' assert found_set.artifact_ids == {'123'}
def test_removing_artifact_from_set(): a = artifact_record(id='foo') b = artifact_record(id='blah') artifact_set = r.ArtifactSet([a.id, b.id], labels='myset') updated_set = artifact_set.remove(b) assert b not in updated_set assert updated_set.name is None updated_named_set = artifact_set.remove(b, labels='myset') assert b not in updated_named_set assert updated_named_set.name == 'myset'