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 )
