def test_main_works(self): main.make_app().AndReturn(self.fake_app) jobs = [self.mox.CreateMock(PeriodicJob), self.mox.CreateMock(PeriodicJob)] main.setup_logging(self.fake_app) main.check_connection(self.fake_app).AndReturn(True) main.instances_jobs.jobs_factory(self.fake_app).AndReturn(jobs) self.fake_app.run(use_reloader=False, host='127.0.0.1', port=42) jobs[0].cancel().AndRaise(RuntimeError('ignore me')) jobs[1].cancel() self.mox.ReplayAll() main.main()
def test_main_works(self): main.make_app().AndReturn(self.fake_app) jobs = [ self.mox.CreateMock(PeriodicJob), self.mox.CreateMock(PeriodicJob) ] main.setup_logging(self.fake_app) main.check_connection(self.fake_app).AndReturn(True) main.instances_jobs.jobs_factory(self.fake_app).AndReturn(jobs) self.fake_app.run(use_reloader=False, host='127.0.0.1', port=42) jobs[0].cancel().AndRaise(RuntimeError('ignore me')) jobs[1].cancel() self.mox.ReplayAll() main.main()
def setUp(self): super(NoAuthEndpointTestCase, self).setUp() self.app = make_app() @self.app.route('/hello') @no_auth_endpoint def hello_(): g.client_set = 42 return 'hello, world!'
def setUp(self): super(TestCase, self).setUp() self.app = make_app(None) self.app.config['AUDIT_VERBOSITY'] = 0 self.client = self.app.test_client() if self.FAKE_AUTH: self.fake_client_set = self._fake_client_set_factory() self.__require_auth = auth.require_auth auth.require_auth = self.install_fake_auth
def main(argv=None): try: argv = argv if argv is not None else sys.argv try: command = _COMMANDS[argv[1]] except (KeyError, IndexError): show_help(argv) with make_app().test_request_context(): command(argv) except Exception, e: die('Failed: %s', e)
def setUp(self): super(UserEndpointTestCase, self).setUp() self.app = make_app() self.app.config['SYSTENANT'] = 'test_default_tenant' @self.app.route('/hello') @user_endpoint def hello_(): assert g.audit_data['user_id'] == 'FAKE_UID' assert g.client_set == 'FAKE_CLIENT_SET' return 'hello, world!' self.mox.StubOutWithMock(auth, '_client_set') self.mox.StubOutWithMock(auth, 'admin_role_id') self.mox.StubOutWithMock(auth, 'current_user_id')
def test_make_app_generally_works(self): self.mox.StubOutClassWithMocks(main, 'ApiApp') self.mox.StubOutWithMock(main.DB, 'init_app') self.mox.StubOutWithMock(main, 'register_entry_points') os.environ[main.CONFIG_ENV] = 'TEST' app = main.ApiApp('altai_api.main') app.config = self.mox.CreateMockAnything() app.config.from_object('altai_api.default_settings') app.config.from_envvar(main.CONFIG_ENV) main.DB.init_app(app) for _ in xrange(18): app.register_blueprint(mox.IsA(flask.Blueprint), url_prefix=mox.IsA(basestring)) main.register_entry_points(app) self.mox.ReplayAll() self.assertEquals(app, main.make_app())
def test_main_check_failed(self): main.make_app().AndReturn(self.fake_app) main.setup_logging(self.fake_app) main.check_connection(self.fake_app).AndReturn(False) self.mox.ReplayAll() self.assertRaises(SystemExit, main.main)
def test_settings_are_read(self): self.mox.ReplayAll() # we promise we will never redefine TEST_STRING self.assertEquals(main.make_app(None).config['TEST_STRING'], 'Test')