def setUp(self):
if self.type == "mongo":
reset_db()
create_users()
self.configuration = self.mongodb_config
elif self.type == "memory":
self.memory_config['backend']['filename'] = self.DATA_FILE
self.configuration = self.memory_config
elif self.type == "directory":
self.configuration = self.directory_config
elif self.type == "memory_no_config":
self.configuration = self.no_config
elif self.type == "no_taxii":
self.configuration = self.config_no_taxii
elif self.type == "no_auth":
self.configuration = self.config_no_auth
elif self.type == "no_backend":
self.configuration = self.config_no_backend
else:
raise RuntimeError("Unknown backend!")
self.app = create_app(self.configuration)
self.app_context = self.app.app_context()
self.app_context.push()
# TODO: It might be better to not reuse the test client.
self.client = self.app.test_client()
self.auth = {'Authorization': 'Token abc123'}
def setUp(self):
self.app = application_instance
self.app_context = application_instance.app_context()
self.app_context.push()
self.app.testing = True
register_blueprints(self.app)
if self.type == "mongo":
reset_db(self.mongodb_config["backend"]["uri"])
self.configuration = self.mongodb_config
elif self.type == "memory":
self.configuration = self.memory_config
elif self.type == "memory_no_config":
self.configuration = self.no_config
elif self.type == "no_taxii":
self.configuration = self.config_no_taxii
elif self.type == "no_auth":
self.configuration = self.config_no_auth
elif self.type == "no_backend":
self.configuration = self.config_no_backend
else:
raise RuntimeError("Unknown backend!")
set_config(self.app, "backend", self.configuration)
set_config(self.app, "users", self.configuration)
set_config(self.app, "taxii", self.configuration)
self.client = application_instance.test_client()
if self.type == "memory_no_config" or self.type == "no_auth":
encoded_auth = "Basic " + \
base64.b64encode(b"user:pass").decode("ascii")
else:
encoded_auth = "Basic " + \
base64.b64encode(b"admin:Password0").decode("ascii")
self.auth = {"Authorization": encoded_auth}
def backend():
reset_db()
init({
"backend": {
"type": "mongodb",
"url": "mongodb://localhost:27017/"
},
"users": []
})
return get_backend()
def setUp(self):
reset_db()
self.configuration = config.mongodb_config()
self.configuration['backend']['default_page_size'] = 20
self.app = create_app(self.configuration)
self.app_context = self.app.app_context()
self.app_context.push()
self.client = self.app.test_client()
self.auth = {'Authorization': 'Token abc123'}
def setUp(self):
self.app = application_instance
self.app_context = application_instance.app_context()
self.app_context.push()
self.app.testing = True
register_blueprints(self.app)
if self.type == "mongo":
reset_db()
self.configuration = self.mongodb_config
else:
self.memory_config['backend']['filename'] = self.DATA_FILE
self.configuration = self.memory_config
init_backend(self.app, self.configuration["backend"])
set_users_config(self.app, self.configuration["users"])
set_taxii_config(self.app, self.configuration["taxii"])
self.client = application_instance.test_client()
encoded_auth = 'Basic ' + \
base64.b64encode(b"admin:Password0").decode("ascii")
self.auth = {'Authorization': encoded_auth}
def setUp(self):
self.app = application_instance
self.application_context = self.app.app_context()
self.application_context.push()
self.app.testing = True
register_blueprints(self.app)
reset_db()
self.configuration = {
"backend": {
"module": "medallion.backends.mongodb_backend",
"module_class": "MongoBackend",
"url": "mongodb://localhost:27017/"
},
"users": {
"admin": "Password0"
}
}
init_backend(self.app, self.configuration["backend"])
set_config(self.app, self.configuration["users"])
self.client = application_instance.test_client()
encoded_auth = 'Basic ' + base64.b64encode(b"admin:Password0").decode(
"ascii")
self.auth = {'Authorization': encoded_auth}
def backend():
reset_db()
init_backend({"type": "mongodb", "url": "mongodb://localhost:27017/"})
return get_backend()
