Add test_predict_503 to AppTestCase

suaaa7 · suaaa7 · commit cc439d70df04 · 2020-08-02T20:24:18.000+09:00
diff --git a/python_test_example/flask_app/app.py b/python_test_example/flask_app/app.py
@@ -18,7 +18,7 @@ def index():
 def predict():
     print('Call predict in app')
     target = 'A'
-    if not service.check_model:
+    if not service.check_model():
         return make_response(jsonify({'message': 'Service Unavailable'}), 503)
     result = {'result': service.predict(target)}
     return make_response(jsonify(result), 200)
diff --git a/python_test_example/flask_app/app_test.py b/python_test_example/flask_app/app_test.py
@@ -1,6 +1,6 @@
 import json
 from unittest import TestCase, main
-from unittest.mock import Mock, patch
+from unittest.mock import patch
 
 print('In app_test')
 
@@ -38,5 +38,10 @@ def test_predict_data(self):
         response = self.client.post('/predict')
         self.assertIsInstance(json.loads(response.data)['result'], float)
 
+    @patch('flask_app.service.Service.check_model', return_value=False)
+    def test_predict_503(self, mock):
+        response = self.client.post('/predict')
+        self.assertEqual(response.status_code, 503)
+
 if __name__ == '__main__':
     main()
