def test_get_env_db(self):
"""Test getting the environment database"""
db = utils.get_environment_db(self.request)
self.assertEqual(self.DEFAULT_ENV, db)
# change environment
utils.set_environment(self.request, 'qa')
db = utils.get_environment_db(self.request)
self.assertEqual('qa', db)
def test_set_env(self):
"""Test setting the environment"""
utils.set_environment(self.request, 'qa')
self.assertEqual('qa', self.request.session.get('environment', False))
def change_environment(request, env):
utils.set_environment(request, env)
if 'next' in request.GET:
return http.HttpResponseRedirect(request.GET.get('next'))
return http.HttpResponseRedirect(reverse('home'))
