def test_apply_model_only_combobox(self): invalid_types = [1, "One", u'1', {'1': "one"}, VMListModeler( mock_domains_info)] for invalid_type in invalid_types: with self.assertRaises(TypeError): self.apply_model(invalid_type, [])
def _new_vm_list_modeler(self): return VMListModeler(mock_domains_info)
def setUp(self): self.modeler = VMListModeler(mock_domains_info)
class VMListModelerTest(unittest.TestCase): def setUp(self): self.modeler = VMListModeler(mock_domains_info) def test_entries_gets_loaded(self): self.assertIsNotNone(self.modeler._entries) def test_valid_qube_name(self): self.modeler.apply_model(Gtk.ComboBox(), list(mock_domains_info.keys())) for name in [ "test-red1", "test-red2", "test-red3", "test-target", "Disposable VM (test-disp6)", ]: mock = MockComboEntry(name) self.assertEqual( name, self.modeler._get_valid_qube_name(mock, mock, mock_whitelist), ) self.assertEqual( name, self.modeler._get_valid_qube_name(None, mock, mock_whitelist), ) self.assertEqual( name, self.modeler._get_valid_qube_name(mock, None, mock_whitelist), ) self.assertIsNone( self.modeler._get_valid_qube_name(None, None, mock_whitelist)) def test_valid_qube_name_whitelist(self): list_exc = ["@dispvm:test-disp6", "test-red2"] whitelist = [name for name in mock_whitelist if name not in list_exc] self.modeler.apply_model(Gtk.ComboBox(), whitelist) for name in list_exc: mock = MockComboEntry(name) self.assertIsNone( self.modeler._get_valid_qube_name(mock, mock, whitelist)) self.assertIsNone( self.modeler._get_valid_qube_name(None, mock, whitelist)) self.assertIsNone( self.modeler._get_valid_qube_name(mock, None, whitelist)) def test_invalid_qube_name(self): self.modeler.apply_model(Gtk.ComboBox(), mock_whitelist) for name in ["test-nonexistant", None, "", 1]: mock = MockComboEntry(name) self.assertIsNone( self.modeler._get_valid_qube_name(mock, mock, mock_whitelist)) self.assertIsNone( self.modeler._get_valid_qube_name(None, mock, mock_whitelist)) self.assertIsNone( self.modeler._get_valid_qube_name(mock, None, mock_whitelist)) def test_apply_model(self): new_object = Gtk.ComboBox() self.assertIsNone(new_object.get_model()) self.modeler.apply_model(new_object, mock_whitelist) self.assertIsNotNone(new_object.get_model()) def test_apply_model_with_entry(self): new_object = Gtk.ComboBox.new_with_entry() self.assertIsNone(new_object.get_model()) self.modeler.apply_model(new_object, []) self.assertIsNotNone(new_object.get_model()) def test_apply_model_only_combobox(self): invalid_types = [ 1, "One", "1", { "1": "one" }, VMListModeler(mock_domains_info), ] for invalid_type in invalid_types: with self.assertRaises(TypeError): self.modeler.apply_model(invalid_type, []) def test_apply_model_whitelist(self): combo = Gtk.ComboBox() self.modeler.apply_model(combo, list(mock_domains_info.keys())) self.assertEqual(7, len(combo.get_model())) names = [entry["api_name"] for entry in self.modeler._entries.values()] self.modeler.apply_model(combo, [names[0]]) self.assertEqual(1, len(combo.get_model())) self.modeler.apply_model(combo, [names[0], names[1]]) self.assertEqual(2, len(combo.get_model())) def test_apply_icon(self): new_object = Gtk.Entry() self.assertIsNone( new_object.get_icon_pixbuf(Gtk.EntryIconPosition.PRIMARY)) self.modeler.apply_icon(new_object, "Disposable VM (test-disp6)") self.assertIsNotNone( new_object.get_icon_pixbuf(Gtk.EntryIconPosition.PRIMARY)) def test_apply_icon_only_entry(self): invalid_types = [1, "One", "1", {"1": "one"}, Gtk.ComboBox()] for invalid_type in invalid_types: with self.assertRaises(TypeError): self.modeler.apply_icon(invalid_type, "test-disp6") def test_apply_icon_only_existing(self): new_object = Gtk.Entry() for name in [ "test-red1", "test-red2", "test-red3", "test-target", "Disposable VM (test-disp6)", ]: self.modeler.apply_icon(new_object, name) for name in ["test-nonexistant", None, "", 1]: with self.assertRaises(ValueError): self.modeler.apply_icon(new_object, name)
def __init__(self, *args, **kwargs): unittest.TestCase.__init__(self, *args, **kwargs) VMListModeler.__init__(self, mock_domains_info)