def testUpdate(self): lsid = "some LSID" existing_job = ALAJob(lsid) session_maker = MagicMock() session = MagicMock(spec=scoped_session) toTest = ALAJobDAO(session_maker) new_lsid = "a new lsid" dataset_id = 123 status = "new status" submitted_time = datetime.datetime.now() + datetime.timedelta(seconds=1000) start_time = datetime.datetime.now() + datetime.timedelta(seconds=5000) end_time = datetime.datetime.now() + datetime.timedelta(seconds=10000) attempts = 120 session_maker.generate_session.return_value = session out = toTest.update(existing_job, lsid=new_lsid, dataset_id=dataset_id, status=status, submitted_time=submitted_time, start_time=start_time, end_time=end_time, attempts=attempts) self.assertIsNotNone(out) self.assertEqual(new_lsid, out.lsid) self.assertEqual(dataset_id, out.dataset_id) self.assertEqual(status, out.status) self.assertEqual(submitted_time, out.submitted_time) self.assertEqual(start_time, out.start_time) self.assertEqual(end_time, out.end_time) self.assertEqual(attempts, out.attempts) session.add.assert_called_with(existing_job) session.flush.assert_called session.expunge.assert_called_with(existing_job)
def testCreateNewJob(self): lsid = 'urn:lsid:biodiversity.org.au:afd.taxon:31a9b8b8-4e8f-4343-a15f-2ed24e0bf1ae' toTest = ALAJobDAO(None) toTest._session_maker = MagicMock() toTest._session_maker.generate_session.return_value = MagicMock(spec=scoped_session) new_job = toTest.create_new(lsid) self.assertIsNotNone(new_job) self.assertEqual(lsid, new_job.lsid) self.assertEqual(None, new_job.dataset_id) toTest._session_maker.generate_session.return_value.add.assert_called_with(ANY)
def testFindById(self): lsid = 'urn:lsid:biodiversity.org.au:afd.taxon:31a9b8b8-4e8f-4343-a15f-2ed24e0bf1ae' session_maker = MagicMock() session = MagicMock(spec=scoped_session) toTest = ALAJobDAO(session_maker) session_maker.generate_session.return_value = session session.query.return_value.get.return_value = ALAJob(lsid) out_job = toTest.findById(1) self.assertIsNotNone(out_job) self.assertEqual(lsid, out_job.lsid) self.assertEqual(None, out_job.dataset_id) session.query.return_value.get.assert_called_with(1)