class TestDetermineID(unittest.TestCase): def setUp(self): self.context = mock.MagicMock() self.context.config = test_config self.command = UploadOpenstackImageCommand(self.context) def test_return_value(self): ret = self.command.determine_type_id('/a/b/c') self.assertEqual(ret, constants.IMAGE_TYPE_ID)