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