def test_last_session_not_None(self, mock_datetime): yesterday = mock.Mock() today = mock.Mock() days_since = mock.Mock() today.__sub__ = mock.Mock(return_value=days_since) days_since.days = 1 last_session = mock.Mock() last_session.datetime = mock.Mock(date=yesterday) mock_datetime.today = mock.Mock(return_value=mock.Mock(date=mock.Mock(return_value=today))) output = scraping.get_days_since_last_scrape(last_session=last_session) self.assertEqual(output, 1)
def test_no_last_session_returns_None(self): output = scraping.get_days_since_last_scrape(last_session=None) self.assertEqual(output, None)