1# Copyright (C) 2012 Google Inc. All rights reserved.
2#
3# Redistribution and use in source and binary forms, with or without
4# modification, are permitted provided that the following conditions are
5# met:
6#
7#    * Redistributions of source code must retain the above copyright
8# notice, this list of conditions and the following disclaimer.
9#    * Redistributions in binary form must reproduce the above
10# copyright notice, this list of conditions and the following disclaimer
11# in the documentation and/or other materials provided with the
12# distribution.
13#    * Neither the name of Google Inc. nor the names of its
14# contributors may be used to endorse or promote products derived from
15# this software without specific prior written permission.
16#
17# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
18# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
19# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
20# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
21# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
22# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
23# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
24# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
25# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
27# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28
29import webkitpy.thirdparty.unittest2 as unittest
30
31from webkitpy.tool.servers.reflectionhandler import ReflectionHandler
32
33
34class TestReflectionHandler(ReflectionHandler):
35    STATIC_FILE_DIRECTORY = "/"
36
37    def __init__(self):
38        self.static_files_served = set()
39        self.errors_sent = set()
40        self.functions_run = set()
41
42    def _serve_static_file(self, name):
43        self.static_files_served.add(name)
44
45    def send_error(self, code, description):
46        self.errors_sent.add(code)
47
48    def function_one(self):
49        self.functions_run.add("function_one")
50
51    def some_html(self):
52        self.functions_run.add("some_html")
53
54
55class WriteConvertingLogger(object):
56    def __init__(self):
57        self.data = ''
58
59    def write(self, data):
60        # If data is still in ASCII, this will throw an exception.
61        self.data = str(data)
62
63
64class TestReflectionHandlerServeXML(ReflectionHandler):
65    def __init__(self):
66        self.requestline = False
67        self.client_address = '127.0.0.1'
68        self.request_version = '1'
69        self.wfile = WriteConvertingLogger()
70
71    def serve_xml(self, data):
72        self._serve_xml(data)
73
74    def log_message(self, _format, *_args):
75        pass
76
77
78class ReflectionHandlerTest(unittest.TestCase):
79    def assert_handler_response(self, requests, expected_static_files, expected_errors, expected_functions):
80        handler = TestReflectionHandler()
81        for request in requests:
82            handler.path = request
83            handler._handle_request()
84        self.assertEqual(handler.static_files_served, expected_static_files)
85        self.assertEqual(handler.errors_sent, expected_errors)
86        self.assertEqual(handler.functions_run, expected_functions)
87
88    def test_static_content_or_function_switch(self):
89        self.assert_handler_response(["/test.js"], set(["test.js"]), set(), set())
90        self.assert_handler_response(["/test.js", "/test.css", "/test.html"], set(["test.js", "test.html", "test.css"]), set(), set())
91        self.assert_handler_response(["/test.js", "/test.exe", "/testhtml"], set(["test.js"]), set([404]), set())
92        self.assert_handler_response(["/test.html", "/function.one"], set(["test.html"]), set(), set(['function_one']))
93        self.assert_handler_response(["/some.html"], set(["some.html"]), set(), set())
94
95    def test_svn_log_non_ascii(self):
96        xmlChangelog = u'<?xml version="1.0"?>\n<log>\n<logentry revision="1">\n<msg>Patch from John Do\xe9.</msg>\n</logentry>\n</log>'
97        handler = TestReflectionHandlerServeXML()
98        handler.serve_xml(xmlChangelog)
99        self.assertEqual(handler.wfile.data, xmlChangelog.encode('utf-8'))
100