def test_report_needs_input_date_and_pk(self): g = Goal() g.input = 'Ok' g.get_pk = lambda: 'pk' r = g.get_report() self.assertIsInstance(r, Record) self.assertEqual(r.date, g.input_date)
def test_connector_preload_calls_execute(self): g = Goal('Kokoko?') g.get_pk = lambda: 'kokoko' with mock.patch('goals.Connector.executemany', mock.Mock()) as p: Connector(goals=[g]) self.assertEqual(len(p.call_args), 2)