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(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 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_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')
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')
