def test_create_mongostorage_config_no_defaults(self):
config = MongoStorageConfiguration()
config.url = 'mongodb://localhost:666/'
config.database = "testy"
config.drop_all_first = False
self.assertEquals({'database': 'testy', 'drop_all_first': False, 'url': 'mongodb://localhost:666/'},
config.create_mongostorage_config())
def setUp(self):
config = MongoStorageConfiguration()
config.drop_all_first = True
self.storage_engine = MongoStorageEngine(config)
self.storage_engine.initialise()
client = TestClient()
self.context = client.create_client_context("TESTUSER")
self.context.brain._security._account_linker = BasicAccountLinkerService(
self.storage_engine)
def test_getters_setters(self):
config = MongoStorageConfiguration()
config.url = 'mongodb://localhost:666/'
self.assertEquals(config.url, 'mongodb://localhost:666/')
config.database = "testy"
self.assertEquals("testy", config.database)
config.drop_all_first = False
self.assertFalse(config.drop_all_first)
def test_to_yaml_no_defaults(self):
config = MongoStorageConfiguration()
config.url = 'mongodb://localhost:666/'
config.database = "testy"
config.drop_all_first = False
data = {}
config.to_yaml(data, defaults=False)
self.assertEquals({'database': 'testy', 'drop_all_first': False, 'url': 'mongodb://localhost:666/'}, data)
def test_load_no_collection(self):
config = MongoStorageConfiguration()
config.drop_all_first = True
engine = MongoStorageEngine(config)
engine.initialise()
store = TestMongoLookupStore(engine)
collection = TestCollection()
store.load(collection)
self.assertEquals(0, len(collection.pairs.keys()))
def upload(storetype,
url,
database,
filename,
dirname,
subdir,
extension,
verbose=False):
config = MongoStorageConfiguration()
config.url = url
config.database = database
config.drop_all_first = False
engine = MongoStorageEngine(config)
engine.initialise()
store = Uploader._get_store(storetype, engine)
outputLog(None, "Emptying [%s]" % storetype)
store.empty()
outputLog(None, "Loading [%s]" % storetype)
if filename is not None:
count, success = store.upload_from_file(
filename,
fileformat=Store.TEXT_FORMAT,
commit=True,
verbose=verbose)
outputLog(None, "Lines processed: %d" % count)
outputLog(None, "Entities successful: %d" % success)
elif dirname is not None:
count, success = store.upload_from_directory(directory=dirname,
subdir=subdir,
extension=extension,
commit=True,
verbose=verbose)
outputLog(None, "Lines processed: %d" % count)
outputLog(None, "Entities successful: %d" % success)
else:
raise Exception("You must specify either --file or --dir")
return count, success
def setUp(self):
config = MongoStorageConfiguration()
config.drop_all_first = True
self.storage_engine = MongoStorageEngine(config)
self.storage_engine.initialise()
def test_initialise_drop_all_first_false(self):
config = MongoStorageConfiguration()
config.drop_all_first = False
engine = MongoStorageEngine(config)
engine.initialise()
self.user_asserts(storage_engine=engine)
