def test001_list_ok(self): sample = ["test"] expected = {"total": len(sample), "data": sample} with patch("frontend.api.controllers.probes.celery_brain") as mock: mock.probe_list.return_value = sample result = api_probes.list() self.assertEqual(api_probes.response.content_type, "application/json; charset=UTF-8") self.assertTrue(mock.probe_list.called) self.assertEqual(result, expected)
def test002_list_error(self): expected = Exception("test") celery_brain = "frontend.api.controllers.probes.celery_brain" with patch(celery_brain) as mock_error: mock_error.probe_list.side_effect = expected process_error = "frontend.api.controllers.probes.process_error" with patch(process_error) as mock: result = api_probes.list() self.assertIsNone(result) self.assertTrue(mock.called) self.assertEqual(mock.call_args, ((expected,),))