def test_UpdateSession(self):
"""
Update a session
Assert:
An arbitrarily large amount of credit should be granted.
"""
msg = UpdateSessionRequest()
credit_update = CreditUsageUpdate()
credit_update.sid = 'abc'
msg.updates.extend([credit_update])
resp = self.servicer.UpdateSession(msg, None)
session_response = self._rm_whitespace(str(resp))
expected = self._rm_whitespace(USR)
self.assertEqual(session_response, expected, 'There should be a large '
'amount of additional credit granted')
def test_UpdateSession(self):
"""
Update a session
Assert:
An arbitrarily large amount of credit should be granted.
"""
msg = UpdateSessionRequest()
credit_update = CreditUsageUpdate()
credit_update.common_context.sid.id = 'abc'
msg.updates.extend([credit_update])
resp = self.servicer.UpdateSession(msg, None)
session_response = self._rm_whitespace(str(resp))
expected = self._rm_whitespace(USR)
self.assertEqual(
session_response, expected, 'There should be an '
'infinite, unmetered credit grant and an infinite, '
'metered credit grant')
def update_session(client, args):
message = UpdateSessionRequest()
resp = client.UpdateSession(message)
print('Successfully got a response from UpdateSession:\n', resp)