def create_config():
return Config(cluster=RandomCluster(urls=helper.TEST_CLUSTER_URLS),
timeout_seconds=10,
back_off=StepBackOffStrategy(steps=[1]),
ttl_seconds=None,
auth=BasicAuth(username=helper.TEST_USERNAME,
password=helper.TEST_PASSWORD),
disable_database_delete=False,
disable_collection_delete=False)
def test_disable_delete(self):
cfg = Config(cluster=RandomCluster(urls=helper.TEST_CLUSTER_URLS),
timeout_seconds=10,
back_off=StepBackOffStrategy(steps=[1]),
ttl_seconds=None,
auth=BasicAuth(username=helper.TEST_USERNAME,
password=helper.TEST_PASSWORD))
db = cfg.database(helper.TEST_ARANGO_DB)
coll = db.collection('test_coll')
self.assertRaises(AssertionError, lambda: coll.force_delete())
def test_code_as_config_basic_auth(self):
config = Config(
cluster=RandomCluster(
urls=['http://127.0.0.1:8529']), # see Cluster Styles
timeout_seconds=3,
back_off=StepBackOffStrategy([0.1, 0.5, 1, 1,
1]), # see Back Off Strategies
auth=BasicAuth(username='******', password=''),
ttl_seconds=31622400)
self.assertIsNotNone(config)
def test_database_exists(self):
cfg = Config(
cluster=RandomCluster(urls=helper.TEST_CLUSTER_URLS),
timeout_seconds=10,
back_off=StepBackOffStrategy(steps=[1]),
ttl_seconds=None,
auth=BasicAuth(
username=helper.TEST_USERNAME,
password=helper.TEST_PASSWORD
)
)
db = cfg.database(helper.TEST_ARANGO_DB)
self.assertFalse(db.check_if_exists())
def test_collection_default_ttl(self):
cfg = Config(cluster=RandomCluster(urls=helper.TEST_CLUSTER_URLS),
timeout_seconds=10,
back_off=StepBackOffStrategy(steps=[1]),
ttl_seconds=2,
auth=BasicAuth(username=helper.TEST_USERNAME,
password=helper.TEST_PASSWORD),
disable_database_delete=False)
db = cfg.database(helper.TEST_ARANGO_DB)
self.assertTrue(db.create_if_not_exists())
coll = db.collection('test_coll')
self.assertTrue(coll.create_if_not_exists())
coll.create_or_overwrite_doc('test_doc', {'foo': 3})
self.assertEqual(coll.read_doc('test_doc'), {'foo': 3})
self.assertTrue(db.force_delete())
def test_protect(self):
cfg = Config(cluster=RandomCluster(urls=helper.TEST_CLUSTER_URLS),
timeout_seconds=10,
back_off=StepBackOffStrategy(steps=[1]),
ttl_seconds=None,
auth=BasicAuth(username=helper.TEST_USERNAME,
password=helper.TEST_PASSWORD),
disable_database_delete=False,
disable_collection_delete=False,
protected_collections=['test_coll'])
db = cfg.database(helper.TEST_ARANGO_DB)
coll = db.collection('test_coll')
self.assertRaises(AssertionError, lambda: coll.force_delete())
self.assertTrue(db.create_if_not_exists())
coll = db.collection('test_coll2')
self.assertFalse(coll.force_delete())
self.assertTrue(db.force_delete())
def test_recover(self):
auth = BasicAuth(
username=helper.TEST_USERNAME,
password=helper.TEST_PASSWORD
)
self.assertFalse(auth.try_recover_auth_failure())