from invoke import Collection from mongoengine import ConnectionError from openelex.settings import init_db from fetch import fetch from shell import shell import archive, cache, datasource, load, load_metadata, transform, validate, bake # Build tasks namespace ns = Collection() ns.add_task(fetch) ns.add_collection(archive) ns.add_collection(cache) ns.add_collection(datasource) ns.add_collection(load) ns.add_collection(load_metadata) ns.add_collection(transform) ns.add_collection(validate) ns.add_collection(bake) ns.add_task(shell) # Initialize prod Mongo connection try: init_db() except ConnectionError: pass
def setUp(self): try: self.db = init_db('openelex_test') except ConnectionError: raise SkipTest('Could not connect to Mongo on localhost')