def setup_app(config_obj): app = Flask(__name__) app.config.from_object(config_obj) database.init_db(app) app.register_blueprint(frontend, url_prefix='') app.register_blueprint(ipxe, url_prefix='/ipxe') app.register_blueprint(hosts, url_prefix='/api/1.0/hosts') app.register_blueprint(groups, url_prefix='/api/1.0/groups') app.register_blueprint(templates, url_prefix='/api/1.0/templates') app.register_blueprint(power, url_prefix='/api/1.0/power') return app
def setUp(self): app = Flask(__name__) app.config.from_object(settings) app.testing = True app.config['MONGO_DBNAME'] = 'bergenholmtest' database.init_db(app) app.register_blueprint(frontend, url_prefix='') app.register_blueprint(ipxe, url_prefix='/ipxe') app.register_blueprint(hosts, url_prefix='/api/1.0/hosts') app.register_blueprint(groups, url_prefix='/api/1.0/groups') app.register_blueprint(templates, url_prefix='/api/1.0/templates') app.register_blueprint(power, url_prefix='/api/1.0/power') self.app = app self.client = app.test_client() self.mongo = database.mongo with self.app.test_request_context('/'): self.mongo.db.hosts.remove() self.mongo.db.groups.remove() self.mongo.db.templates.remove() params = {u"_id": self.host_id} params.update(self.host_params) self.mongo.db.hosts.insert(params) self.mongo.db.groups.insert({ u"_id": u"default", u"base_url": u"http://127.0.0.1"}) params = {u"_id": self.group_id} params.update(self.group_params) self.mongo.db.groups.insert(params) self.mongo.db.templates.insert({ u"_id": self.template_id, u"content": self.template})