def testGetPayload_simple(self):
string_buffer = StringIO.StringIO(_RESPONSE_ONLY)
header = string_buffer.readline()
command_sequence = client_replay.CommandSequence()
command_sequence._parser = client_replay._Parser(string_buffer)
payload_string = command_sequence._parser._GetPayloadString(header)
self.assertEqual(payload_string, '{"param2": 42\n}\n')
def testGetPayload_badscript(self):
string_buffer = StringIO.StringIO(_BAD_SCRIPT)
header = string_buffer.readline()
command_sequence = client_replay.CommandSequence()
command_sequence._parser = client_replay._Parser(string_buffer)
payload_string = command_sequence._parser._GetPayloadString(header)
self.assertEqual(payload_string, '{"param2": "))}\\})}/{)}({(})}"\n}\n')
def testGetPayload_script(self):
string_buffer = StringIO.StringIO(_PAYLOAD_SCRIPT)
header = string_buffer.readline()
command_sequence = client_replay.CommandSequence()
command_sequence._parser = client_replay._Parser(string_buffer)
payload_string = command_sequence._parser._GetPayloadString(header)
self.assertEqual(payload_string, '{"param2": "function(){func()}"\n}\n')
def testGetPayload_simple(self):
string_buffer = StringIO.StringIO(_RESPONSE_ONLY)
header = string_buffer.readline()
command_sequence = client_replay.CommandSequence()
command_sequence._parser = client_replay._Parser(string_buffer)
payload_string = command_sequence._parser._GetPayloadString(header)
self.assertEqual(payload_string, '{"param2": 42\n}\n')
def testGetPayload_badscript(self):
string_buffer = StringIO.StringIO(_BAD_SCRIPT)
header = string_buffer.readline()
command_sequence = client_replay.CommandSequence()
command_sequence._parser = client_replay._Parser(string_buffer)
payload_string = command_sequence._parser._GetPayloadString(header)
self.assertEqual(payload_string, '{"param2": "))}\\})}/{)}({(})}"\n}\n')
def testGetPayload_script(self):
string_buffer = StringIO.StringIO(_PAYLOAD_SCRIPT)
header = string_buffer.readline()
command_sequence = client_replay.CommandSequence()
command_sequence._parser = client_replay._Parser(string_buffer)
payload_string = command_sequence._parser._GetPayloadString(header)
self.assertEqual(payload_string, '{"param2": "function(){func()}"\n}\n')
def testGetNextClientHeaderLine(self):
string_buffer = StringIO.StringIO(_PAYLOAD_SCRIPT)
command_sequence = client_replay.CommandSequence()
command_sequence._parser = client_replay._Parser(string_buffer)
self.assertEquals(command_sequence._parser._GetNextClientHeaderLine(),
("[1531428670.535][INFO]: [b15232d5497ec0d8300a5a1ea56f33ce]"
" RESPONSE GetTitle {\n"))
def testGetNextClientHeaderLine_readableTime(self):
string_buffer = StringIO.StringIO(_PAYLOAD_READABLE_TIME)
command_sequence = client_replay.CommandSequence()
command_sequence._parser = client_replay._Parser(string_buffer)
self.assertEquals(command_sequence._parser._GetNextClientHeaderLine(),
("[08-12-2019_15:45:34.824002][INFO]:"
" [b15232d5497ec0d8300a5a1ea56f33ce] RESPONSE GetTitle {\n"))
def testParserGetNext(self):
string_buffer = StringIO.StringIO(_WITH_PARAMS)
command_sequence = client_replay.CommandSequence()
command_sequence._parser = client_replay._Parser(string_buffer)
command = command_sequence._parser.GetNext()
self.assertEqual(command.name, "GetTitle")
self.assertEqual(command.GetPayloadPrimitive(), {"param1": 7})
self.assertEqual(command.session_id, _SESSION_ID)
def testParserGetNext(self):
string_buffer = StringIO.StringIO(_WITH_PARAMS)
command_sequence = client_replay.CommandSequence()
command_sequence._parser = client_replay._Parser(string_buffer)
command = command_sequence._parser.GetNext()
self.assertEqual(command.name, "GetTitle")
self.assertEqual(command.GetPayloadPrimitive(), {"param1": 7})
self.assertEqual(command.session_id, _SESSION_ID)
def testIngestLoggedResponse(self):
string_buffer = StringIO.StringIO(_RESPONSE_ONLY)
command_sequence = client_replay.CommandSequence()
command_sequence._parser = client_replay._Parser(string_buffer)
response = command_sequence._parser.GetNext()
self.assertEqual(response.name, "GetTitle")
self.assertEqual(response.GetPayloadPrimitive(), {"param2": 42})
self.assertEqual(response.session_id, _SESSION_ID)
def testIngestLoggedResponse(self):
string_buffer = StringIO.StringIO(_RESPONSE_ONLY)
command_sequence = client_replay.CommandSequence()
command_sequence._parser = client_replay._Parser(string_buffer)
response = command_sequence._parser.GetNext()
self.assertEqual(response.name, "GetTitle")
self.assertEqual(response.GetPayloadPrimitive(), {"param2": 42})
self.assertEqual(response.session_id, _SESSION_ID)
def testNextCommandEmptyParams(self):
string_buffer = StringIO.StringIO(_NO_PARAMS)
command_sequence = client_replay.CommandSequence()
command_sequence._parser = client_replay._Parser(string_buffer)
command = command_sequence.NextCommand(None)
response = command_sequence._last_response
self.assertEqual(command.name, "GetTitle")
self.assertEqual(command.GetPayloadPrimitive(), {"sessionId": _SESSION_ID})
self.assertEqual(command.session_id, _SESSION_ID)
self.assertEqual(response.name, "GetTitle")
self.assertEqual(response.GetPayloadPrimitive(), None)
self.assertEqual(response.session_id, _SESSION_ID)
def testNextCommandEmptyParams(self):
string_buffer = StringIO.StringIO(_NO_PARAMS)
command_sequence = client_replay.CommandSequence()
command_sequence._parser = client_replay._Parser(string_buffer)
command = command_sequence.NextCommand(None)
response = command_sequence._last_response
self.assertEqual(command.name, "GetTitle")
self.assertEqual(command.GetPayloadPrimitive(), {"sessionId": _SESSION_ID})
self.assertEqual(command.session_id, _SESSION_ID)
self.assertEqual(response.name, "GetTitle")
self.assertEqual(response.GetPayloadPrimitive(), None)
self.assertEqual(response.session_id, _SESSION_ID)
def testParseResponse_false(self):
string_buffer = StringIO.StringIO(_COMMAND_ONLY)
command_sequence = client_replay.CommandSequence()
command_sequence._parser = client_replay._Parser(string_buffer)
self.assertFalse(command_sequence._parser.GetNext().IsResponse())
def testParseResponse_true(self):
string_buffer = StringIO.StringIO(_RESPONSE_ONLY)
command_sequence = client_replay.CommandSequence()
command_sequence._parser = client_replay._Parser(string_buffer)
self.assertTrue(command_sequence._parser.GetNext().IsResponse())
def testParseCommand_true(self):
string_buffer = StringIO.StringIO(_COMMAND_ONLY)
command_sequence = client_replay.CommandSequence()
command_sequence._parser = client_replay._Parser(string_buffer)
self.assertTrue(command_sequence._parser.GetNext().IsCommand())
def testParseCommand_true(self): string_buffer = StringIO.StringIO(_COMMAND_ONLY) command_sequence = client_replay.CommandSequence() command_sequence._parser = client_replay._Parser(string_buffer) self.assertTrue(command_sequence._parser.GetNext().IsCommand())
def testParseResponse_true(self): string_buffer = StringIO.StringIO(_RESPONSE_ONLY) command_sequence = client_replay.CommandSequence() command_sequence._parser = client_replay._Parser(string_buffer) self.assertTrue(command_sequence._parser.GetNext().IsResponse())
def testParseResponse_false(self): string_buffer = StringIO.StringIO(_COMMAND_ONLY) command_sequence = client_replay.CommandSequence() command_sequence._parser = client_replay._Parser(string_buffer) self.assertFalse(command_sequence._parser.GetNext().IsResponse())
def testParseCommand_false(self):
string_buffer = io.StringIO(_RESPONSE_ONLY)
command_sequence = client_replay.CommandSequence()
command_sequence._parser = client_replay._Parser(string_buffer)
self.assertFalse(command_sequence._parser.GetNext().IsCommand())