def test__get_status_icon_path(self): model = InstanceTableModel() green = ':/icons/led_green.svg' yellow = ':/icons/led_yellow.svg' red = ':/icons/led_red.svg' self.assertEqual(model._get_status_icon_path(NodeState.RUNNING), green) self.assertEqual(model._get_status_icon_path(NodeState.REBOOTING), yellow) self.assertEqual(model._get_status_icon_path(NodeState.UNKNOWN), yellow) self.assertEqual(model._get_status_icon_path(NodeState.PENDING), yellow) self.assertEqual(model._get_status_icon_path(NodeState.STOPPED), red) self.assertEqual(model._get_status_icon_path(NodeState.TERMINATED), red)
def setUp(self): super(TestInstanceModel, self).setUp() self.model = InstanceTableModel()
class TestInstanceModel(BaseTest): def setUp(self): super(TestInstanceModel, self).setUp() self.model = InstanceTableModel() def test__get_status_icon_path(self): green = ':/icons/led_green.svg' yellow = ':/icons/led_yellow.svg' red = ':/icons/led_red.svg' self.assertEqual(self.model._get_status_icon_path(NodeState.RUNNING), green) self.assertEqual(self.model._get_status_icon_path(NodeState.REBOOTING), yellow) self.assertEqual(self.model._get_status_icon_path(NodeState.UNKNOWN), yellow) self.assertEqual(self.model._get_status_icon_path(NodeState.PENDING), yellow) self.assertEqual(self.model._get_status_icon_path(NodeState.STOPPED), red) self.assertEqual(self.model._get_status_icon_path(NodeState.TERMINATED), red) def test_add_instance(self): node = list(gen_fake_nodes(1))[0] node.name = 'Foo' node.size = NodeSize('', '', 2048, 0, 0, 0.0, '') node.status = NodeState.RUNNING self.model.addInstance(node) index = self.model.createIndex(0, 0) self.assertEqual(self.model.data(index, Qt.DisplayRole), 'Foo') index = self.model.createIndex(0, 1) self.assertIsInstance(self.model.data(index, Qt.DecorationRole), QIcon) index = self.model.createIndex(0, 2) self.assertEqual(self.model.data(index, Qt.DisplayRole), 2048) index = self.model.createIndex(0, 3) self.assertEqual(self.model.data(index, Qt.DisplayRole), 0) index = self.model.createIndex(0, 4) self.assertIsNone(self.model.data(index, Qt.DisplayRole)) def test_get_instance(self): node1, node2 = gen_fake_nodes(2) self.model.addInstance(node1) self.model.addInstance(node2) self.assertEqual(self.model.getInstance(0), node1) self.assertEqual(self.model.getInstance(1), node2) self.assertIsNone(self.model.getInstance(2)) def test_header_data(self): self.assertEqual(self.model.headerData(0, Qt.Vertical, Qt.DisplayRole), 1) self.assertEqual(self.model.headerData(0, Qt.Horizontal, Qt.DisplayRole), 'Instance') self.assertEqual(self.model.headerData(1, Qt.Horizontal, Qt.DisplayRole), '') self.assertEqual(self.model.headerData(2, Qt.Horizontal, Qt.DisplayRole), 'Size') self.assertEqual(self.model.headerData(3, Qt.Horizontal, Qt.DisplayRole), 'Devices') self.assertIsNone(self.model.headerData(4, Qt.Horizontal, Qt.DisplayRole)) def test_update_instance_status(self): node1, node2 = gen_fake_nodes(2) self.model.addInstance(node1) self.model.addInstance(node2) node = self.model.getInstance(0) node.state = NodeState.STOPPED self.model.update_instance_status(node) node = self.model.getInstance(0) self.assertEqual(node.state, NodeState.STOPPED) node = self.model.getInstance(1) self.model.update_instance_status(node) # instance didn't change node = self.model.getInstance(1) self.assertEqual(node.state, NodeState.RUNNING) def test_update(self): pass