class TestInterComListener(unittest.TestCase):
def setUp(self):
config = get_config_for_testing(temp_dir=TMP_DIR)
self.mongo_server = MongoMgr(config=config)
self.generic_listener = InterComListener(config=config)
def tearDown(self):
for item in self.generic_listener.connections.keys():
self.generic_listener.client.drop_database(
self.generic_listener.connections[item]['name'])
self.generic_listener.shutdown()
self.mongo_server.shutdown()
TMP_DIR.cleanup()
gc.collect()
def check_file(self, binary):
self.generic_listener.connections[
self.generic_listener.CONNECTION_TYPE]['fs'].put(
pickle.dumps(binary))
task = self.generic_listener.get_next_task()
self.assertEqual(task, binary)
another_task = self.generic_listener.get_next_task()
self.assertIsNone(another_task, 'task not deleted')
def test_small_file(self):
self.check_file(b'this is a test')
def test_big_file(self):
large_test_data = generate_random_data(size=BSON_MAX_FILE_SIZE + 1024)
self.check_file(large_test_data)
def _get_delete_tasks(self):
intercom = InterComListener(config=self.config)
intercom.CONNECTION_TYPE = 'file_delete_task'
delete_tasks = []
while True:
tmp = intercom.get_next_task()
if tmp is None:
break
delete_tasks.append(tmp['_id'])
intercom.shutdown()
return delete_tasks
def setUp(self):
config = get_config_for_testing(temp_dir=TMP_DIR)
self.mongo_server = MongoMgr(config=config)
self.generic_listener = InterComListener(config=config)
def setUp(self):
self.generic_listener = InterComListener(config=self.config)