def test_feedback(self): counter = FeedbackCounter() for index in range(0, 10): counter.create() counter.reject() with captured_output() as (out, err): counter.feedback(filename='test', records=20) # captured print output res = out.getvalue().strip() self.assertIn('10 created', res) self.assertIn('20 records processed', res)
def test_feedbackcounter(self): counter = FeedbackCounter() self.assertEqual(counter.counter, 0) counter.increment() self.assertEqual(counter.counter, 1) counter.increment() self.assertEqual(counter.counter, 2) counter.reject() self.assertEqual(counter.counter, 3) self.assertEqual(counter.rejected, 1) counter.update() self.assertEqual(counter.counter, 4) self.assertEqual(counter.rejected, 1) self.assertEqual(counter.updated, 1) counter.create() self.assertEqual(counter.counter, 5) self.assertEqual(counter.updated, 1) self.assertEqual(counter.created, 1)