def test_build_and_send_request(self):
# construct a conn, mocking all the real work
AUTH_TOKEN = "AUTHKEY"
WC_TOKEN = "WCKEY"
with mock.patch.object(HealthVaultConn, '_get_auth_token') as gat:
with mock.patch.object(HealthVaultConn, '_get_record_id') as gri:
gat.return_value = AUTH_TOKEN
gri.return_value = "8"
c = HealthVaultConn(
app_id="1", app_thumbprint="2", public_key=TEST_PUBLIC_KEY, private_key=TEST_PRIVATE_KEY,
wctoken=WC_TOKEN, server="6", shell_server="7"
)
with mock.patch.object(c, '_send_request') as sendRequest:
c._build_and_send_request(method_name="METHOD", info="<info>BOO</info>")
payload = sendRequest.call_args[0][0]
request = ET.fromstring(payload)
# print elt_to_string(request)
header = request.find('header')
method_name = header.find('method').text
self.assertEqual("METHOD", method_name)
self.assertEqual("1", header.find('method-version').text)
self.assertEqual("8", header.find('record-id').text)
self.assertEqual(AUTH_TOKEN, header.find('auth-session/auth-token').text)
self.assertEqual(WC_TOKEN, header.find('auth-session/user-auth-token').text)
self.assertEqual("BOO", request.find('info').text)
def test_build_and_send_request(self):
# construct a conn, mocking all the real work
AUTH_TOKEN = "AUTHKEY"
WC_TOKEN = "WCKEY"
with mock.patch.object(HealthVaultConn, '_get_auth_token') as gat:
with mock.patch.object(HealthVaultConn, '_get_record_id') as gri:
gat.return_value = AUTH_TOKEN
gri.return_value = "8"
c = HealthVaultConn(app_id="1",
app_thumbprint="2",
public_key=TEST_PUBLIC_KEY,
private_key=TEST_PRIVATE_KEY,
wctoken=WC_TOKEN,
server="6",
shell_server="7")
with mock.patch.object(c, '_send_request') as sendRequest:
c._build_and_send_request(method_name="METHOD",
info="<info>BOO</info>")
payload = sendRequest.call_args[0][0]
request = ET.fromstring(payload)
# print elt_to_string(request)
header = request.find('header')
method_name = header.find('method').text
self.assertEqual("METHOD", method_name)
self.assertEqual("1", header.find('method-version').text)
self.assertEqual("8", header.find('record-id').text)
self.assertEqual(AUTH_TOKEN,
header.find('auth-session/auth-token').text)
self.assertEqual(WC_TOKEN,
header.find('auth-session/user-auth-token').text)
self.assertEqual("BOO", request.find('info').text)
