Add patch to flask_app/app_test

Author: suaaa7

python_test_example/flask_app/app.py

@@ -1,6 +1,6 @@
 import json
 
-from flask import Flask, jsonify
+from flask import Flask, jsonify, make_response
 
 from .service import Service
 
@@ -11,12 +11,14 @@
 
 @app.route('/', methods=['GET'])
 def index():
-    message = {'Message': 'ok'}
-    return jsonify(message)
+    message = {'message': 'OK'}
+    return make_response(jsonify(message), 200)
 
 @app.route('/predict', methods=['POST'])
 def predict():
     print('Call predict in app')
     target = 'A'
-    message = {'result': service.predict(target)}
-    return jsonify(message)
+    if not service.check_model:
+        return make_response(jsonify({'message': 'Service Unavailable'}), 503)
+    result = {'result': service.predict(target)}
+    return make_response(jsonify(result), 200)

python_test_example/flask_app/app_test.py

@@ -1,15 +1,23 @@
 import json
 from unittest import TestCase, main
-
-from flask_app.app import app
+from unittest.mock import Mock, patch
 
 print('In app_test')
 
 class AppTestCase(TestCase):
     def setUp(self):
         print('Call setUp in AppTestCase')
+
+        self.load_model_patcher = patch('flask_app.model.load_model')
+        self.load_model_m = self.load_model_patcher.start()
+        self.load_model_m.return_value = 'test'
+
+        from flask_app.app import app
         self.client = app.test_client()
 
+    def tearDown(self):
+        self.load_model_patcher.stop()
+
     def test_index(self):
         response = self.client.get('/')
         self.assertEqual(response.status_code, 200)

python_test_example/flask_app/model.py

@@ -0,0 +1,3 @@
+def load_model():
+    print('Call load_model in model')
+    return 'model'

python_test_example/flask_app/service.py

@@ -1,11 +1,20 @@
 from random import randint
 
+from .model import load_model
+
 print('In service')
 
 class Service:
     def __init__(self):
         print('Init Service')
+        self.model = load_model()
+        print(f'self.model: {self.model}')
 
     def predict(self, target: str):
         print('Call predict in service')
         return randint(0, 1000) / 1000.0
+
+    def check_model(self):
+        if self.model:
+            return True
+        return False