class TestInspectTaskAdd(unittest.TestCase):
"""点检任务创建测试类"""
def setUp(self):
self.isy = InspectStrategy()
self.it = InspectTask()
self.it_add_data = {
"code": "001",
"inspectTime": "2018-07-24 11:18:35",
"bmEquipmentGid": params.bmEquipmentGid,
"bmEquipInspectItemGid": params.bmEquipInspectItemGid,
"bmEquipInspectPartGid": params.bmEquipInspectPartGid
}
def test_inspecttask_add_1(self):
"""
点检任务创建
:return:
"""
code = create_code()
self.it_add_data['code'] = code
self.it.inspecttask_add = Mock(side_effect=self.it.inspecttask_add)
resp_it_add = self.it.inspecttask_add(self.it_add_data)
self.assertIn('data', resp_it_add.keys())
class TestInspectTaskModify(unittest.TestCase):
"""点检任务更新测试类"""
def setUp(self):
self.isy = InspectStrategy()
self.it = InspectTask()
self.it_add_data = {
"code": "001",
"inspectTime": "2018-07-24 11:18:35",
"bmEquipmentGid": params.bmEquipmentGid,
"bmEquipInspectItemGid": params.bmEquipInspectItemGid,
"bmEquipInspectPartGid": params.bmEquipInspectPartGid
}
def test_inspecttask__1(self):
"""
点检任务更新
:return:
"""
code = create_code()
self.it_add_data['code'] = code
self.it.inspecttask_add = Mock(side_effect=self.it.inspecttask_add)
resp_it_add = self.it.inspecttask_add(self.it_add_data)
self.it_add_data['inspectTime'] = "2018-07-01 00:00:00"
self.it.inspecttask_modify = Mock(
side_effect=self.it.inspecttask_modify)
resp_it_modify = self.it.inspecttask_modify(self.it_add_data)
self.assertIn('success', resp_it_modify.keys())