def test_dynamic_bulkadd(self):
tqd = flexmock(taskqueue_distributed.TaskQueueServiceStub)
flexmock(taskqueue_service_pb)
taskqueue_service_pb.should_receive("TaskQueueBulkAddRequest").\
and_return(FakeBulkAddRequest())
taskqueue_service_pb.should_receive("TaskQueueBulkAddResponse").\
and_return(FakeBulkAddResponse())
tqd.should_receive("__GetTQLocation").and_return("some_location")
tqd.should_receive("_RemoteSend").and_return()
tqd = taskqueue_distributed.TaskQueueServiceStub("app_id", "hostname", 12345)
self.assertEquals(None, tqd._Dynamic_BulkAdd(FakeBulkAddRequest(), None))
def test_dynamic_add(self):
flexmock(taskqueue_service_pb)
taskqueue_service_pb.should_receive("TaskQueueBulkAddRequest").\
and_return(FakeBulkAddRequest())
taskqueue_service_pb.should_receive("TaskQueueBulkAddResponse").\
and_return(FakeBulkAddResponse())
tqd = flexmock(taskqueue_distributed.TaskQueueServiceStub)
tqd.should_receive("_GetTQLocations").and_return(["some_location"])
tqd.should_receive("_RemoteSend").and_return()
tqd.should_receive("_Dynamic_BulkAdd").and_return()
tqd = taskqueue_distributed.TaskQueueServiceStub("app_id", "hostname")
self.assertEquals(None, tqd._Dynamic_Add(FakeAddRequest(), FakeAddResponse()))
def test_dynamic_bulkadd(self):
app_id = 'app-id'
tqd = flexmock(taskqueue_distributed.TaskQueueServiceStub)
flexmock(taskqueue_service_pb)
taskqueue_service_pb.should_receive("TaskQueueBulkAddRequest").\
and_return(FakeBulkAddRequest())
taskqueue_service_pb.should_receive("TaskQueueBulkAddResponse").\
and_return(FakeBulkAddResponse())
tqd.should_receive("_GetTQLocations").and_return(["some_location"])
tqd.should_receive("_RemoteSend").and_return()
tqd = taskqueue_distributed.TaskQueueServiceStub(app_id, "hostname")
port_file = os.path.join(
'/', 'etc', 'appscale', 'port-{}.txt'.format('app-id_default_v1'))
flexmock(sys.modules['__builtin__']).should_receive('open').\
with_args(port_file).and_return(flexmock(read=lambda: '80'))
self.assertEquals(None, tqd._Dynamic_BulkAdd(FakeBulkAddRequest(), None))