def test_get_sharing_space_info_existing(self):
servers = ['a', 'b', 'c']
Properties.sharing_space_servers = servers
collection_name = str(uuid.uuid4())
subscriber_list = [self.__subscriber_id, self.__second_subscriber_id]
sharing_secret, subscribers_collections =\
self.__create_sharing_record(subscriber_list, collection_name)
load_balancer = SharingLoadBalancer.get_instance()
load_balancer.get_sharing_space_info(sharing_secret, callback=self.stop)
server_info = self.wait()
server_adrs = server_info['server']
self.assertTrue(server_adrs in servers)
load_balancer.get_sharing_space_info(sharing_secret, callback=self.stop)
second_server_info = self.wait()
second_server_adrs = second_server_info['server']
self.assertEqual(server_adrs, second_server_adrs)
#cleanup
SharingController.remove_sharing_record_by_secret(sharing_secret, callback =self.stop)
self.wait()
StorageServer.remove_collection(self.__account_id, collection_name,
callback=self.stop)
for subscriber_id in subscribers_collections:
subscriber_collection = subscribers_collections[subscriber_id]
StorageServer.remove_collection(subscriber_id, subscriber_collection,
callback=self.stop)
self.wait()
def test_remove_sharing_space_info_invalid_sharing_secret(self):
load_balancer = SharingLoadBalancer.get_instance()
sharing_secret = 'seeeecret'
load_balancer.remove_sharing_space_info(sharing_secret,
callback=self.stop)
self.wait()
def delete(self, sharing_secret):
self.__log.info('SharingLoadBalancer - removing sharing space %s from load balancer info' % sharing_secret)
load_balancer = SharingLoadBalancer.get_instance()
yield gen.Task(load_balancer.remove_sharing_space_info, sharing_secret)
self.set_status(200)
self.finish()
def test_get_sharing_space_info_invalid_secret(self):
load_balancer = SharingLoadBalancer.get_instance()
sharing_secret = 'doodoo'
load_balancer.get_sharing_space_info(sharing_secret, callback=self.stop)
second_server_info = self.wait()
self.assertTrue(second_server_info is None)
def get(self, sharing_secret):
load_balancer = SharingLoadBalancer.get_instance()
sharing_server_info_dict = yield gen.Task(load_balancer.get_sharing_space_info,
sharing_secret)
if sharing_server_info_dict is None:
self.set_status(404)
self.finish()
else :
self.__log.info('SharingLoadBalancer - Querying sharing space %s from load balancer' % sharing_secret)
self.set_status(200)
json_str = json.dumps(sharing_server_info_dict)
self.write(json_str)
self.finish()
