class FlaskArangoGraphTests(FlaskRequestTests):
def setUp(self):
# pylint: disable=no-init,too-few-public-methods,unused-variable
super(FlaskArangoGraphTests, self).setUp()
self.app.config['ARANGO_URL'] = 'http://localhost:8529'
self.app.config['ARANGO_DB'] = 'flask_arango_test'
self.app.config['ARANGO_GRAPH'] = 'GraphTest'
self.arango = Arango()
self.arango.init_app(self.app)
if not self.arango.connection.hasDatabase('flask_arango_test'):
self.arango.connection.createDatabase('flask_arango_test')
class EdgeTest(Edges):
pass
class FromTest(Collection):
pass
class ToTest(Collection):
pass
class GraphTest(Graph):
_edgeDefinitions = (EdgeDefinition(
'EdgeTest',
fromCollections=['FromTest'],
toCollections=['ToTest']),)
_orphanedCollections = ()
if not self.arango.db.hasGraph('GraphTest'):
self.arango.db.createGraph('GraphTest')
def test_database_access(self):
self.assertIsNotNone(self.arango.gdb)
def setUp(self):
# pylint: disable=no-init,too-few-public-methods,unused-variable
super(FlaskArangoGraphTests, self).setUp()
self.app.config['ARANGO_URL'] = 'http://localhost:8529'
self.app.config['ARANGO_DB'] = 'flask_arango_test'
self.app.config['ARANGO_GRAPH'] = 'GraphTest'
self.arango = Arango()
self.arango.init_app(self.app)
if not self.arango.connection.hasDatabase('flask_arango_test'):
self.arango.connection.createDatabase('flask_arango_test')
class EdgeTest(Edges):
pass
class FromTest(Collection):
pass
class ToTest(Collection):
pass
class GraphTest(Graph):
_edgeDefinitions = (EdgeDefinition('EdgeTest',
fromCollections=['FromTest'],
toCollections=['ToTest']), )
_orphanedCollections = ()
if not self.arango.db.hasGraph('GraphTest'):
self.arango.db.createGraph('GraphTest')
def setUp(self):
super(FlaskArangoDatabaseTests, self).setUp()
self.app.config['ARANGO_URL'] = 'http://localhost:8529'
self.app.config['ARANGO_DB'] = 'flask_arango_test'
self.app.config['ARANGO_GRAPH'] = 'GraphTest'
self.arango = Arango()
self.arango.init_app(self.app)
if not self.arango.connection.hasDatabase('flask_arango_test'):
self.arango.connection.createDatabase('flask_arango_test')
class FlaskArangoDatabaseTests(FlaskRequestTests):
def setUp(self):
super(FlaskArangoDatabaseTests, self).setUp()
self.app.config['ARANGO_URL'] = 'http://localhost:8529'
self.app.config['ARANGO_DB'] = 'flask_arango_test'
self.app.config['ARANGO_GRAPH'] = 'GraphTest'
self.arango = Arango()
self.arango.init_app(self.app)
if not self.arango.connection.hasDatabase('flask_arango_test'):
self.arango.connection.createDatabase('flask_arango_test')
def test_database_access(self):
self.assertIsNotNone(self.arango.db)
class FlaskArangoDatabaseTests(FlaskRequestTests):
def setUp(self):
super(FlaskArangoDatabaseTests, self).setUp()
self.app.config['ARANGO_URL'] = 'http://localhost:8529'
self.app.config['ARANGO_DB'] = 'flask_arango_test'
self.app.config['ARANGO_GRAPH'] = 'GraphTest'
self.arango = Arango()
self.arango.init_app(self.app)
if not self.arango.connection.hasDatabase('flask_arango_test'):
self.arango.connection.createDatabase('flask_arango_test')
def test_database_access(self):
self.assertIsNotNone(self.arango.db)
def setUp(self):
super(FlaskArangoDatabaseTests, self).setUp()
self.app.config['ARANGO_URL'] = 'http://localhost:8529'
self.app.config['ARANGO_DB'] = 'flask_arango_test'
self.app.config['ARANGO_GRAPH'] = 'GraphTest'
self.arango = Arango()
self.arango.init_app(self.app)
if not self.arango.connection.hasDatabase('flask_arango_test'):
self.arango.connection.createDatabase('flask_arango_test')
def test_init_app(self):
arango = Arango()
arango.init_app(self.app)
self.assertIsNotNone(arango.connection)
from assignableresources import *
from context import Context
from app_exceptions import *
from flask import Flask, request, Response, abort
from flask_arango import Arango
import ujson
app = Flask(__name__)
app.config.from_object('config.default_settings')
app.config.from_pyfile('eris.cfg', silent=True)
arango = Arango(app)
context = Context(app=app, arango=arango)
def inject_from_dict(event):
if "type" not in event:
return False
event_type = event["type"]
if event_type == "ip":
IPResource.inject_from_dict(event)
elif event_type == "subnet":
raise NotImplementedException
elif event_type == "nsrec":
NSRecResource.inject_from_dict(event)
elif event_type == "email":
EmailResource.inject_from_dict(event)
elif event_type == "domain":
DomainResource.inject_from_dict(event)
elif event_type == "vhost":
VhostResource.inject_from_dict(event)
elif event_type == "uri":
def test_init_app(self):
arango = Arango()
arango.init_app(self.app)
self.assertIsNotNone(arango.connection)
def test_direct_initialization(self):
arango = Arango(self.app)
self.assertIsNotNone(arango.connection)
