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)
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_direct_initialization(self): arango = Arango(self.app) self.assertIsNotNone(arango.connection)