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"])
