def test_do_request_not_found(self): mock_do_post = mock.Mock( return_value=Response(self.status_not_found, self.resp_not_found)) cert_store = Client(self.apikey, self.cert_path, self.sig_param, self.uri, True) request_func = cert_store.do_post with mock.patch('requests.post', mock_do_post): _, result = cert_store.do_request( { "url": self.uri, "headers": self.header, "body": self.request, }, request_func) self.assertEqual(MSG_DECRYPT_FAILED, result["ErrMessage"]) self.assertEqual(CODE_DECRYPT_FAILED, result["ErrCode"])
def test_do_request_succ(self): mock_do_post = mock.Mock( return_value=Response(self.status_ok, json.dumps(self.resp))) cert_store = Client(self.apikey, self.cert_path, self.sig_param, self.uri, True) request_func = cert_store.do_post with mock.patch('requests.post', mock_do_post): _, result = cert_store.do_request( { "url": self.uri, "headers": self.header, "body": self.request, }, request_func) self.assertTrue(result)
def test_do_request_with_no_encrypt_succ(self): mock_do_post = mock.Mock( return_value=Response(self.status_ok, json.dumps(self.resp))) cert_store = Client(self.apikey, self.cert_path, self.uri, False) request_func = cert_store.do_post with mock.patch('requests.post', mock_do_post): _, result = cert_store.do_request( { "headers": self.header, "body": self.request, }, request_func, ) self.assertEqual(0, result["ErrCode"])
def test_do_request_empty(self): mock_do_post = mock.Mock( return_value=Response(self.status_not_found, "")) mock_run_cmd = mock.Mock(return_value=self.cipher) cert_store = Client(self.apikey, self.cert_path, self.uri) request_func = cert_store.do_post with mock.patch('cryption.crypto.run_cmd', mock_run_cmd): with mock.patch('requests.post', mock_do_post): _, result = cert_store.do_request( { "headers": self.header, "body": self.request, }, request_func) self.assertEqual(MSG_RESP_BODY_EMPTY, result["ErrMsg"]) self.assertEqual(CODE_RESP_BODY_EMPTY, result["ErrCode"])
def test_do_request_succ(self): mock_do_post = mock.Mock( return_value=Response(self.status_ok, json.dumps(self.resp))) mock_run_cmd = mock.Mock( side_effect=[self.cipher, json.dumps(self.resp)]) cert_store = Client(self.apikey, self.cert_path, self.uri) request_func = cert_store.do_post with mock.patch('cryption.crypto.run_cmd', mock_run_cmd): with mock.patch('requests.post', mock_do_post): _, result = cert_store.do_request( { "headers": self.header, "body": self.request, }, request_func) self.assertEqual(0, result["ErrCode"])
def test_do_request_empty(self): mock_do_post = mock.Mock( return_value=Response(self.status_not_found, "")) mock_run_cmd = mock.Mock(return_value=self.cipher) cert_store = Client(self.apikey, self.cert_path, self.sig_param, self.uri, True) request_func = cert_store.do_post with mock.patch('requests.post', mock_do_post): _, result = cert_store.do_request( { "url": self.uri, "headers": self.header, "body": self.request, }, request_func) self.assertEqual(MSG_SERVER_RESP_INVALID, result["ErrMessage"]) self.assertEqual(CODE_SERVER_RESP_INVALID, result["ErrCode"])
def test_do_request_not_found(self): mock_do_post = mock.Mock( return_value=Response(self.status_not_found, self.resp_not_found)) cert_store = Client( self.apikey, #self.cert_path, self.sig_param, self.uri, True, cert=('./tls/client.pem', './tls/client.key'), cacert='./tls/ca.crt') request_func = cert_store.do_post with mock.patch('requests.post', mock_do_post): _, result = cert_store.do_request( { "url": self.uri, "headers": self.header, "body": self.request, }, request_func) self.assertEqual(MSG_SERVER_RESP_INVALID, result["ErrMessage"]) self.assertEqual(CODE_SERVER_RESP_INVALID, result["ErrCode"])
def test_do_request_not_found(self): mock_do_post = mock.Mock( return_value=Response(self.status_not_found, self.resp_not_found)) mock_run_cmd = mock.Mock(side_effect=[ self.cipher, json.dumps({ "ErrCode": CODE_DECRYPT_FAILED, "ErrMsg": MSG_DECRYPT_FAILED }) ]) cert_store = Client(self.apikey, self.cert_path, self.uri) request_func = cert_store.do_post with mock.patch('cryption.crypto.run_cmd', mock_run_cmd): with mock.patch('requests.post', mock_do_post): _, result = cert_store.do_request( { "headers": self.header, "body": self.request, }, request_func) self.assertEqual(MSG_DECRYPT_FAILED, result["ErrMsg"]) self.assertEqual(CODE_DECRYPT_FAILED, result["ErrCode"])
def test_do_request_with_no_encrypt_succ(self): mock_do_post = mock.Mock( return_value=Response(self.status_ok, json.dumps(self.resp))) cert_store = Client( self.apikey, #self.cert_path, self.sig_param, self.uri, True, cert=('./tls/client.pem', './tls/client.key'), cacert='./tls/ca.crt') request_func = cert_store.do_post with mock.patch('requests.post', mock_do_post): _, result = cert_store.do_request( { "url": self.uri, "headers": self.header, "body": self.request, }, request_func, ) self.assertEqual(0, result["ErrCode"])