def test_build_exclude_set(): image_tags = [ 'some_image:latest', 'repo/foo:12345', 'duplicate:latest', ] exclude_image_file = StringIO(textwrap.dedent(""" # Exclude this one because duplicate:latest # Also this one repo/bar:abab """)) expected = set([ 'some_image:latest', 'repo/foo:12345', 'duplicate:latest', 'repo/bar:abab', ]) exclude_set = docker_gc.build_exclude_set(image_tags, exclude_image_file) assert exclude_set == expected
def test_build_exclude_set_empty(): exclude_set = docker_gc.build_exclude_set(None, None) assert exclude_set == set()