############################################################################## # Copyright (c) 2016 Huawei Technologies Co.,Ltd and others. # # All rights reserved. This program and the accompanying materials # are made available under the terms of the Apache License, Version 2.0 # which accompanies this distribution, and is available at # http://www.apache.org/licenses/LICENSE-2.0 ############################################################################## import unittest import mock import json from api.views import Test from api.views import Result from api.views import Env class TestTestCase(unittest.TestCase): @mock.patch('api.views.request') def test_post(self, mock_request): mock_request.json.get.side_effect = ['hello', {}] result = json.loads(Test().post()) self.assertEqual('error', result['status']) class ResultTestCase(unittest.TestCase): @mock.patch('api.views.request') def test_get(self, mock_request): mock_request.args.get.return_value = 'hello' print Result().get() result = json.loads(Result().get()) self.assertEqual('error', result['status']) class EnvTestCase(unittest.TestCase): @mock.patch('api.views.request') def test_post(self, mock_request): mock_request.json.get.side_effect = ['hello', {}] result = json.loads(Env().post()) self.assertEqual('error', result['status']) def main(): unittest.main() if __name__ == '__main__': main()