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'
