class TestParseImageName(unittest.TestCase): def setUp(self): self.docker_client = mock.Mock() self.log = mock.Mock() self.image = "whatever" self.squash = Image(self.log, self.docker_client, self.image, None) def test_should_parse_name_name_with_proper_tag(self): self.assertEqual(self.squash._parse_image_name( 'jboss/wildfly:abc'), ('jboss/wildfly', 'abc')) self.assertEqual( self.squash._parse_image_name('jboss:abc'), ('jboss', 'abc')) def test_should_parse_name_name_without_tag(self): self.assertEqual(self.squash._parse_image_name( 'jboss/wildfly'), ('jboss/wildfly', 'latest')) self.assertEqual( self.squash._parse_image_name('jboss'), ('jboss', 'latest'))