def testHBaseConnection_constructor_3(): dummy_thriftModule = exp.DummyObjectWithExpectations('dummy_thriftModule') dummy_tsocketModule = exp.DummyObjectWithExpectations('dummy_tsocketModule') dummy_transportModule = exp.DummyObjectWithExpectations('dummy_transportModule') dummy_protocolModule = exp.DummyObjectWithExpectations('dummy_protocolModule') dummy_ttypesModule = exp.DummyObjectWithExpectations('dummy_ttypesModule') dummy_clientClass = exp.DummyObjectWithExpectations('dummy_clientClass') dummy_columnClass = exp.DummyObjectWithExpectations('dummy_columnClass') dummy_mutationClass = exp.DummyObjectWithExpectations('dummy_mutationClass') class FakeIOError(Exception): pass class FakeIllegalArgument(Exception): pass class FakeAlreadyExists(Exception): pass class FakeTException(Exception): pass dummy_ttypesModule.expect('IOError', None, None, FakeIOError, None) dummy_ttypesModule.expect('IllegalArgument', None, None, FakeIllegalArgument, None) dummy_ttypesModule.expect('AlreadyExists', None, None, FakeAlreadyExists, None) dummy_thriftModule.expect('TException', None, None, FakeTException, None) dummy_transportObject = exp.DummyObjectWithExpectations('dummy_transportObject') dummy_protocolObject = exp.DummyObjectWithExpectations('dummy_protocolObject') dummy_clientObject = exp.DummyObjectWithExpectations('dummy_clientObject') dummy_tsocketModule.expect('TSocket', ('somehostname', 666), {}, dummy_transportObject) dummy_transportModule.expect('TBufferedTransport', (dummy_transportObject,), {}, dummy_transportObject) dummy_protocolModule.expect('TBinaryProtocol', (dummy_transportObject,), {}, dummy_protocolObject) dummy_clientClass.expect('__call__', (dummy_protocolObject,), {}, dummy_clientObject) dummy_transportObject.expect('setTimeout', (9000,), {}) dummy_transportObject.expect('open', (), {}, None, FakeTException('bad news')) #transport fails 1st time dummy_tsocketModule.expect('TSocket', ('somehostname', 666), {}, dummy_transportObject) dummy_transportModule.expect('TBufferedTransport', (dummy_transportObject,), {}, dummy_transportObject) dummy_protocolModule.expect('TBinaryProtocol', (dummy_transportObject,), {}, dummy_protocolObject) dummy_clientClass.expect('__call__', (dummy_protocolObject,), {}, dummy_clientObject) theCauseException = FakeTException('bad news') theExpectedExceptionAsString = "the connection is not viable. retries fail: No connection was made to HBase (2 tries): <class info deleted>-bad news" dummy_transportObject.expect('setTimeout', (9000,), {}) dummy_transportObject.expect('open', (), {}, None, theCauseException) #transport fails 2nd time dummy_transportObject.expect('close', (), {}) try: conn = hbc.HBaseConnection('somehostname', 666, 9000, thrift=dummy_thriftModule, tsocket=dummy_tsocketModule, ttrans=dummy_transportModule, protocol=dummy_protocolModule, ttp=dummy_ttypesModule, client=dummy_clientClass, column=dummy_columnClass, mutation=dummy_mutationClass) except Exception, x: assert False, 'expecited no exception, but got %s' % str(x)
def testHBaseConnection_constructor_2(): dummy_thriftModule = exp.DummyObjectWithExpectations('dummy_thriftModule') dummy_tsocketModule = exp.DummyObjectWithExpectations('dummy_tsocketModule') dummy_transportModule = exp.DummyObjectWithExpectations('dummy_transportModule') dummy_protocolModule = exp.DummyObjectWithExpectations('dummy_protocolModule') dummy_ttypesModule = exp.DummyObjectWithExpectations('dummy_ttypesModule') dummy_clientClass = exp.DummyObjectWithExpectations('dummy_clientClass') dummy_columnClass = exp.DummyObjectWithExpectations('dummy_columnClass') dummy_mutationClass = exp.DummyObjectWithExpectations('dummy_mutationClass') class FakeIOError(Exception): pass class FakeIllegalArgument(Exception): pass class FakeAlreadyExists(Exception): pass class FakeTException(Exception): pass dummy_ttypesModule.expect('IOError', None, None, FakeIOError, None) dummy_ttypesModule.expect('IllegalArgument', None, None, FakeIllegalArgument, None) dummy_ttypesModule.expect('AlreadyExists', None, None, FakeAlreadyExists, None) dummy_thriftModule.expect('TException', None, None, FakeTException, None) dummy_transportObject = exp.DummyObjectWithExpectations('dummy_transportObject') dummy_protocolObject = exp.DummyObjectWithExpectations('dummy_protocolObject') dummy_clientObject = exp.DummyObjectWithExpectations('dummy_clientObject') dummy_tsocketModule.expect('TSocket', ('somehostname', 666), {}, dummy_transportObject) dummy_transportModule.expect('TBufferedTransport', (dummy_transportObject,), {}, dummy_transportObject) dummy_protocolModule.expect('TBinaryProtocol', (dummy_transportObject,), {}, dummy_protocolObject) dummy_clientClass.expect('__call__', (dummy_protocolObject,), {}, dummy_clientObject) dummy_transportObject.expect('setTimeout', (9000,), {}) dummy_transportObject.expect('open', (), {}, None, FakeTException('bad news')) #transport fails 1st time dummy_tsocketModule.expect('TSocket', ('somehostname', 666), {}, dummy_transportObject) dummy_transportModule.expect('TBufferedTransport', (dummy_transportObject,), {}, dummy_transportObject) dummy_protocolModule.expect('TBinaryProtocol', (dummy_transportObject,), {}, dummy_protocolObject) dummy_clientClass.expect('__call__', (dummy_protocolObject,), {}, dummy_clientObject) dummy_transportObject.expect('setTimeout', (9000,), {}) dummy_transportObject.expect('open', (), {}) #transport succeeds dummy_transportObject.expect('close', (), {}) conn = hbc.HBaseConnection('somehostname', 666, 9000, thrift=dummy_thriftModule, tsocket=dummy_tsocketModule, ttrans=dummy_transportModule, protocol=dummy_protocolModule, ttp=dummy_ttypesModule, client=dummy_clientClass, column=dummy_columnClass, mutation=dummy_mutationClass) conn.close()
def __init__(self): self.dummy_thriftModule = exp.DummyObjectWithExpectations('dummy_thriftModule') self.dummy_tsocketModule = exp.DummyObjectWithExpectations('dummy_tsocketModule') self.dummy_transportModule = exp.DummyObjectWithExpectations('dummy_transportModule') self.dummy_protocolModule = exp.DummyObjectWithExpectations('dummy_protocolModule') self.dummy_ttypesModule = exp.DummyObjectWithExpectations('dummy_ttypesModule') self.dummy_clientClass = exp.DummyObjectWithExpectations('dummy_clientClass') self.dummy_columnClass = exp.DummyObjectWithExpectations('dummy_columnClass') self.dummy_mutationClass = exp.DummyObjectWithExpectations('dummy_mutationClass') class FakeIOError(Exception): pass self.FakeIOError = FakeIOError class FakeIllegalArgument(Exception): pass self.FakeIllegalArgument = FakeIllegalArgument class FakeAlreadyExists(Exception): pass self.FakeAlreadyExists = FakeAlreadyExists class FakeTException(Exception): pass self.FakeTException = FakeTException self.dummy_ttypesModule.expect('IOError', None, None, self.FakeIOError, None) self.dummy_ttypesModule.expect('IllegalArgument', None, None, self.FakeIllegalArgument, None) self.dummy_ttypesModule.expect('AlreadyExists', None, None, self.FakeAlreadyExists, None) self.dummy_thriftModule.expect('TException', None, None, self.FakeTException, None) self.dummy_transportObject = exp.DummyObjectWithExpectations('dummy_transportObject') self.dummy_protocolObject = exp.DummyObjectWithExpectations('dummy_protocolObject') self.dummy_clientObject = exp.DummyObjectWithExpectations('dummy_clientObject') self.dummy_tsocketModule.expect('TSocket', ('somehostname', 666), {}, self.dummy_transportObject) self.dummy_transportModule.expect('TBufferedTransport', (self.dummy_transportObject,), {}, self.dummy_transportObject) self.dummy_protocolModule.expect('TBinaryProtocol', (self.dummy_transportObject,), {}, self.dummy_protocolObject) self.dummy_clientClass.expect('__call__', (self.dummy_protocolObject,), {}, self.dummy_clientObject) self.dummy_transportObject.expect('setTimeout', (9000,), {}) self.dummy_transportObject.expect('open', (), {}) self.conn = hbc.HBaseConnection('somehostname', 666, 9000, thrift=self.dummy_thriftModule, tsocket=self.dummy_tsocketModule, ttrans=self.dummy_transportModule, protocol=self.dummy_protocolModule, ttp=self.dummy_ttypesModule, client=self.dummy_clientClass, column=self.dummy_columnClass, mutation=self.dummy_mutationClass)