def setUp(self): cherrypy.session = {} self.webroot = GilaMonRoot() self.webroot.dfsr = Mock() self.webroot.server = 'DEFAULT' self.webroot.valid_servers = ['DEFAULT', 'VALID_INPUT', 'GOOD_COOKIE']
class TestWebServer(unittest.TestCase): def setUp(self): cherrypy.session = {} self.webroot = GilaMonRoot() self.webroot.dfsr = Mock() self.webroot.server = 'DEFAULT' self.webroot.valid_servers = ['DEFAULT', 'VALID_INPUT', 'GOOD_COOKIE'] def tearDown(self): cherrypy.session = {} server = None self.webroot.server = 'DEFAULT' def test_change_server_none(self): server = self.webroot.change_server() self.assertEqual(server, 'DEFAULT') def test_change_server_bad(self): server = self.webroot.change_server('INVALID_INPUT') self.assertEqual(server, 'DEFAULT') def test_change_server_good(self): server = self.webroot.change_server('VALID_INPUT') self.assertEqual(server, 'VALID_INPUT') def test_check_session_server_no_input_bad_cookie(self): cherrypy.session['server'] = 'BAD_COOKIE' server = self.webroot.check_session_server() self.assertEqual(server, 'DEFAULT') self.assertEqual(server, cherrypy.session.get('server')) def test_check_session_server_no_input_no_cookie(self): server = self.webroot.check_session_server() self.assertEqual(server, 'DEFAULT') self.assertEqual(server, cherrypy.session.get('server')) def test_check_session_server_no_input_good_cookie(self): cherrypy.session['server'] = 'GOOD_COOKIE' server = self.webroot.check_session_server() self.assertEqual(server, 'GOOD_COOKIE') self.assertEqual(server, cherrypy.session.get('server')) def test_check_session_server_bad_input_bad_cookie(self): cherrypy.session['server'] = 'BAD_COOKIE' server = self.webroot.check_session_server('INVALID_INPUT') self.assertEqual(server, 'DEFAULT') self.assertEqual(server, cherrypy.session.get('server')) def test_check_session_server_good_input_good_cookie(self): cherrypy.session['server'] = 'GOOD_COOKIE' server = self.webroot.check_session_server('INVALID_INPUT') self.assertEqual(server, 'GOOD_COOKIE') self.assertEqual(server, cherrypy.session.get('server')) def test_check_session_server_good_input_bad_cookie(self): cherrypy.session['server'] = 'BAD_COOKIE' server = self.webroot.check_session_server('VALID_INPUT') self.assertEqual(server, 'VALID_INPUT') self.assertEqual(server, cherrypy.session.get('server'))