def __init__(self):
self.engine = create_test_engine()
self.model_manager = ModelManager(self.engine)
self.sessionmaker = db.create_scoped_sessionmaker(self.engine)
self.workers = ThreadPool(10)
self.inventory_config = InventoryConfig(gcp_api_mocks.ORGANIZATION_ID,
'', {}, '', {})
self.inventory_config.set_service_config(self)
def __init__(self, engine):
self.engine = engine
self.model_manager = ModelManager(self.engine)
self.sessionmaker = db.create_scoped_sessionmaker(self.engine)
self.workers = ThreadPool(1)
self.inventory_config = InventoryConfig(
None, '', {}, 0, {'enabled': False},
['folders/1032', 'projects/1041'])
self.inventory_config.set_service_config(self)
def __init__(self, engine):
self.engine = engine
self.model_manager = ModelManager(self.engine)
self.sessionmaker = db.create_scoped_sessionmaker(self.engine)
self.workers = ThreadPool(1)
self.inventory_config = InventoryConfig(
gcp_api_mocks.ORGANIZATION_ID, '', {}, 0, {
'enabled': True,
'gcs_path': 'gs://test-bucket'
})
self.inventory_config.set_service_config(self)
def test_concurrent_access(self):
"""Start with no models, create multiple, delete them again, concurrent.
"""
num_threads = 4
thread_pool = ThreadPool(num_threads)
def test_func(x):
"""Create, get, delete models."""
for i in range(32):
handle = self.model_manager.create(name='%s-%s' % (x, i))
self.assertTrue(
handle in [m.handle for m in self.model_manager.models()])
self.model_manager.delete(handle)
self.assertTrue(
handle not in
[m.handle for m in self.model_manager.models()])
return True
for x in range(num_threads):
thread_pool.add_func(test_func, x)
thread_pool.join()
self.assertTrue(len(self.model_manager.models()) == 0,
'Expecting no models to stick around')