def test_find_by_id(self): session_maker = MagicMock(spec=SessionMaker) session = MagicMock(spec=scoped_session) to_test = ALAOccurrenceDAO(session_maker) to_return = ALAOccurrence(None, None, None) to_return.id = 1 session_maker.generate_session.return_value = session session.query.return_value.get.return_value = to_return out = to_test.find_by_id(1) self.assertIs(to_return, out) session.query.assert_called_with(ALAOccurrence) session.query.return_value.get.assert_called_with(1)
def test_create_new(self): lsid = 'some lsid' occurrence_path = '/path/to/occurrence/file' metadata_path = 'path/to/metadata/file' session_maker = MagicMock(spec=SessionMaker) session = MagicMock(spec=scoped_session) to_test = ALAOccurrenceDAO(session_maker) session_maker.generate_session.return_value = session out = to_test.create_new(lsid, occurrence_path, metadata_path) matcher = self.ALAOccurrenceMatcher(lsid, occurrence_path, metadata_path) session.add.assert_called_with(matcher) session.flush.assert_called() session.expunge.assert_called_with(matcher) self.assertIsNotNone(out) self.assertEqual(lsid, out.lsid) self.assertEqual(occurrence_path, out.occurrence_path) self.assertEqual(metadata_path, out.metadata_path)