def test_set_status_launched_not_uploaded(self):
scan = MagicMock()
scan.status = IrmaScanStatus.finished
self.Scan.load_from_ext_id.return_value = scan
module.set_status("whatever", IrmaScanStatus.launched)
self.assertEqual(scan.status, IrmaScanStatus.finished)
def test_set_status_launched_not_uploaded(self):
scan = MagicMock()
scan.status = IrmaScanStatus.finished
self.Scan.load_from_ext_id.return_value = scan
module.set_status("whatever", IrmaScanStatus.launched)
self.assertEqual(scan.status, IrmaScanStatus.finished)
def test_set_status_launched_status_uploaded(self):
scan = MagicMock()
scan.status = IrmaScanStatus.uploaded
self.Scan.load_from_ext_id.return_value = scan
module.set_status("whatever", IrmaScanStatus.launched)
scan.set_status.assert_called_with(IrmaScanStatus.launched)
def test_set_status_launched_status_uploaded(self):
scan = MagicMock()
scan.status = IrmaScanStatus.uploaded
self.Scan.load_from_ext_id.return_value = scan
module.set_status("whatever", IrmaScanStatus.launched)
scan.set_status.assert_called_with(IrmaScanStatus.launched)