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)