def test_get_item_id(self): equipment = Equipment() equipment.id = 1 self.mock_equipment_service.get_all_equipment.return_value = [equipment] equipment_store = EquipmentStore(self.mock_equipment_service) iter = equipment_store.get_iter_first() self.assertEquals(1, equipment_store.get_value(iter, 0))
def test_get_item_description(self): equipment = Equipment() equipment.id = 1 equipment.description = u"item description" self.mock_equipment_service.get_all_equipment.return_value = [equipment] equipment_store = EquipmentStore(self.mock_equipment_service) iter = equipment_store.get_iter_first() self.assertEquals("item description", equipment_store.get_value(iter, 1))
def test_get_item_active(self): equipment = Equipment() equipment.id = 1 equipment.active = False self.mock_equipment_service.get_all_equipment.return_value = [equipment] equipment_store = EquipmentStore(self.mock_equipment_service) iter = equipment_store.get_iter_first() self.assertFalse(equipment_store.get_value(iter, 4))
def test_get_item_usage_text(self): equipment = Equipment() equipment.id = 1 equipment.life_expectancy = 200 self.mock_equipment_service.get_all_equipment.return_value = [equipment] self.mock_equipment_service.get_equipment_usage.return_value = 100 equipment_store = EquipmentStore(self.mock_equipment_service) iter = equipment_store.get_iter_first() self.assertEquals("100 / 200", equipment_store.get_value(iter, 3))
def test_get_item_usage_percent_usage_exceeds_life_expectancy(self): equipment = Equipment() equipment.id = 1 equipment.life_expectancy = 200 self.mock_equipment_service.get_all_equipment.return_value = [equipment] self.mock_equipment_service.get_equipment_usage.return_value = 300 equipment_store = EquipmentStore(self.mock_equipment_service) iter = equipment_store.get_iter_first() self.assertEquals(100, equipment_store.get_value(iter, 2), "Progress bar cannot exceed 100%.")
def test_get_item_usage_percent_zero_usage(self): equipment = Equipment() equipment.id = 1 equipment.life_expectancy = 200 self.mock_equipment_service.get_all_equipment.return_value = [equipment] self.mock_equipment_service.get_equipment_usage.return_value = 0 equipment_store = EquipmentStore(self.mock_equipment_service) iter = equipment_store.get_iter_first() self.assertEquals(0, equipment_store.get_value(iter, 2))
def test_multiple_equipment_items(self): equipment1 = Equipment() equipment1.id = 1 equipment2 = Equipment() equipment2.id = 2 self.mock_equipment_service.get_all_equipment.return_value = [equipment1, equipment2] equipment_store = EquipmentStore(self.mock_equipment_service) iter = equipment_store.get_iter_first() iter = equipment_store.iter_next(iter) self.assertEquals(2, equipment_store.get_value(iter, 0))