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)
