1# Copyright (C) 2011 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 json
30import sys
31import webkitpy.thirdparty.unittest2 as unittest
32
33from webkitpy.common.system.outputcapture import OutputCapture
34from webkitpy.layout_tests.models.test_configuration import *
35from webkitpy.layout_tests.port import builders
36from webkitpy.thirdparty.mock import Mock
37from webkitpy.tool.mocktool import MockTool
38from webkitpy.common.system.executive_mock import MockExecutive
39from webkitpy.common.host_mock import MockHost
40from webkitpy.tool.servers.gardeningserver import *
41
42
43class TestPortFactory(object):
44    # FIXME: Why is this a class method?
45    @classmethod
46    def create(cls):
47        host = MockHost()
48        return host.port_factory.get("test-win-xp")
49
50    @classmethod
51    def path_to_generic_test_expectations_file(cls):
52        return cls.create().path_to_generic_test_expectations_file()
53
54
55class MockServer(object):
56    def __init__(self):
57        self.tool = MockTool()
58        self.tool.executive = MockExecutive(should_log=True)
59        self.tool.filesystem.files[TestPortFactory.path_to_generic_test_expectations_file()] = ""
60
61
62# The real GardeningHTTPRequestHandler has a constructor that's too hard to
63# call in a unit test, so we create a subclass that's easier to construct.
64class TestGardeningHTTPRequestHandler(GardeningHTTPRequestHandler):
65    def __init__(self, server):
66        self.server = server
67        self.body = None
68
69    def _expectations_updater(self):
70        return GardeningExpectationsUpdater(self.server.tool, TestPortFactory.create())
71
72    def read_entity_body(self):
73        return self.body if self.body else ''
74
75    def _serve_text(self, text):
76        print "== Begin Response =="
77        print text
78        print "== End Response =="
79
80    def _serve_json(self, json_object):
81        print "== Begin JSON Response =="
82        print json.dumps(json_object)
83        print "== End JSON Response =="
84
85    def _serve_xml(self, xml):
86        print "== Begin XML Response =="
87        print xml
88        print "== End XML Response =="
89
90class GardeningServerTest(unittest.TestCase):
91    def _post_to_path(self, path, body=None, expected_stderr=None, expected_stdout=None, server=None):
92        handler = TestGardeningHTTPRequestHandler(server or MockServer())
93        handler.path = path
94        handler.body = body
95        OutputCapture().assert_outputs(self, handler.do_POST, expected_stderr=expected_stderr, expected_stdout=expected_stdout)
96
97    def test_svnlog(self):
98        expected_stderr = ''
99        expected_stdout = '== Begin XML Response ==\nMOCK output of child process\n== End XML Response ==\n'
100        self._post_to_path('/svnlog', expected_stderr=expected_stderr, expected_stdout=expected_stdout)
101
102    def test_lastroll(self):
103        expected_stderr = 'MOCK run_command: [\'svn\', \'cat\', \'http://src.chromium.org/chrome/trunk/src/DEPS\'], cwd=None, input=None\n'
104        expected_stdout = '== Begin Response ==\n1\n== End Response ==\n'
105        server = MockServer()
106
107        self.output = ['  "webkit_revision": "3",', 'lol']
108
109        def run_command(args, cwd=None, input=None, **kwargs):
110            print >> sys.stderr, "MOCK run_command: %s, cwd=%s, input=%s" % (args, cwd, input)
111            return self.output.pop(0)
112
113        server.tool.executive.run_command = run_command
114        self._post_to_path('/lastroll', expected_stderr=expected_stderr, expected_stdout='== Begin Response ==\n3\n== End Response ==\n', server=server)
115        self._post_to_path('/lastroll', expected_stderr=expected_stderr, expected_stdout='== Begin Response ==\n0\n== End Response ==\n', server=server)
116
117    def disabled_test_rollout(self):
118        expected_stderr = "MOCK run_command: ['echo', 'rollout', '--force-clean', '--non-interactive', '2314', 'MOCK rollout reason'], cwd=/mock-checkout\n"
119        expected_stdout = "== Begin Response ==\nsuccess\n== End Response ==\n"
120        self._post_to_path("/rollout?revision=2314&reason=MOCK+rollout+reason", expected_stderr=expected_stderr, expected_stdout=expected_stdout)
121
122    def disabled_test_rebaselineall(self):
123        expected_stderr = "MOCK run_command: ['echo', 'rebaseline-json'], cwd=/mock-checkout, input={\"user-scripts/another-test.html\":{\"%s\": [%s]}}\n"
124        expected_stdout = "== Begin Response ==\n{result_code: 0}\n== End Response ==\n"
125        server = MockServer()
126
127        self.output = ['{"add": [], "delete": []}', '']
128
129        def run_command(args, cwd=None, input=None, **kwargs):
130            print >> sys.stderr, "MOCK run_command: %s, cwd=%s, input=%s" % (args, cwd, input)
131            return self.output.pop(0)
132
133        server.tool.executive.run_command = run_command
134        self._post_to_path("/rebaselineall", body='{"user-scripts/another-test.html":{"MOCK builder": ["txt","png"]}}', expected_stderr=expected_stderr % ('MOCK builder', '"txt","png"'), expected_stdout=expected_stdout, server=server)
135
136        self._post_to_path("/rebaselineall", body='{"user-scripts/another-test.html":{"MOCK builder (Debug)": ["txt","png"]}}', expected_stderr=expected_stderr % ('MOCK builder (Debug)', '"txt","png"'), expected_stdout=expected_stdout)
137