def DoSimpleTest( self, inputMessage ): ''' Does a simple test @param: message The message to send ''' mocks = self.GetMockObjects( inputMessage ) g = Gntp( mocks[0] ) g.transport = mocks[ 1 ] g.dataReceived( inputMessage.encode().encode( 'utf-8', 'replace' ) ) for m in mocks: mox.Verify( m )
def DoBufferingTest( self, message ): ''' Does a test of the buffering using the specified message ''' actualData = message.encode().encode( 'utf-8', 'replace' ) for chunkSize in range( 1, len( actualData ) + 1 ): mocks = self.GetMockObjects( message ) g = Gntp( mocks[0] ) g.transport = mocks[ 1 ] # Split the data numChunks = int( math.ceil( len( actualData ) / chunkSize ) ) chunks = [ actualData[ c * chunkSize: c * ( chunkSize + 1 ) ] for c in range( numChunks ) ] # Pass it in for c in chunks: g.dataReceived( c ) import pdb; pdb.set_trace() # Verify for m in mocks: mox.Verify( m )