webkitbot should provide full name, email address, and IRC nicknames on whois
[WebKit-https.git] / Tools / Scripts / webkitpy / tool / bot / irc_command_unittest.py
1 # Copyright (c) 2010 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
29 import os
30 import unittest2 as unittest
31
32 from webkitpy.common.system.outputcapture import OutputCapture
33 from webkitpy.tool.bot.irc_command import *
34 from webkitpy.tool.mocktool import MockTool
35 from webkitpy.common.net.web_mock import MockWeb
36 from webkitpy.common.system.executive_mock import MockExecutive
37 from webkitpy.common.system.filesystem_mock import MockFileSystem
38
39
40 class IRCCommandTest(unittest.TestCase):
41     def test_whois(self):
42         whois = Whois()
43         self.assertEqual("tom: Usage: whois SEARCH_STRING",
44                           whois.execute("tom", [], None, None))
45         self.assertEqual('tom: Adam Barth is "Adam Barth" <abarth@webkit.org> (:abarth) (r). Why do you ask?',
46                           whois.execute("tom", ["Adam", "Barth"], None, None))
47         self.assertEqual("tom: Sorry, I don't know any contributors matching 'unknown@example.com'.",
48                           whois.execute("tom", ["unknown@example.com"], None, None))
49         self.assertEqual('tom: tonyg@chromium.org is "Tony Gentilcore" <tonyg@chromium.org> (:tonyg-cr) (r). Why do you ask?',
50                           whois.execute("tom", ["tonyg@chromium.org"], None, None))
51         self.assertEqual('tom: TonyG@Chromium.org is "Tony Gentilcore" <tonyg@chromium.org> (:tonyg-cr) (r). Why do you ask?',
52                           whois.execute("tom", ["TonyG@Chromium.org"], None, None))
53         self.assertEqual('tom: rniwa is "Ryosuke Niwa" <rniwa@webkit.org> (:rniwa) (r). Why do you ask?',
54                           whois.execute("tom", ["rniwa"], None, None))
55         self.assertEqual('tom: lopez is "Xan Lopez" <xan.lopez@gmail.com> (:xan) (r). Why do you ask?',
56                           whois.execute("tom", ["lopez"], None, None))
57         self.assertEqual(u'tom: Osztrogon\u00e1c is "Csaba Osztrogon\u00e1c" <ossy@webkit.org> (:ossy) (r). Why do you ask?',
58                           whois.execute("tom", [u'Osztrogon\u00e1c'], None, None))
59         self.assertEqual('tom: "Vicki Murley" <vicki@apple.com> hasn\'t told me their nick. Boo hoo :-(',
60                           whois.execute("tom", ["vicki@apple.com"], None, None))
61         self.assertEqual('tom: I\'m not sure who you mean?  "Gavin Peters" <gavinp@chromium.org> (:gavinp) (c) or "Gavin Barraclough" <barraclough@apple.com> (:gbarra) (r) could be \'Gavin\'.',
62                           whois.execute("tom", ["Gavin"], None, None))
63         self.assertEqual('tom: More than 5 contributors match \'david\', could you be more specific?',
64                           whois.execute("tom", ["david"], None, None))
65
66     def test_create_bug(self):
67         create_bug = CreateBug()
68         self.assertEqual("tom: Usage: create-bug BUG_TITLE",
69                           create_bug.execute("tom", [], None, None))
70
71         example_args = ["sherrif-bot", "should", "have", "a", "create-bug", "command"]
72         tool = MockTool()
73
74         # MockBugzilla has a create_bug, but it logs to stderr, this avoids any logging.
75         tool.bugs.create_bug = lambda a, b, cc=None, assignee=None: 50004
76         self.assertEqual("tom: Created bug: http://example.com/50004",
77                           create_bug.execute("tom", example_args, tool, None))
78
79         def mock_create_bug(title, description, cc=None, assignee=None):
80             raise Exception("Exception from bugzilla!")
81         tool.bugs.create_bug = mock_create_bug
82         self.assertEqual("tom: Failed to create bug:\nException from bugzilla!",
83                           create_bug.execute("tom", example_args, tool, None))
84
85     def test_rollout_updates_working_copy(self):
86         rollout = Rollout()
87         tool = MockTool()
88         tool.executive = MockExecutive(should_log=True)
89         expected_logs = "MOCK run_and_throw_if_fail: ['mock-update-webkit'], cwd=/mock-checkout\n"
90         OutputCapture().assert_outputs(self, rollout._update_working_copy, [tool], expected_logs=expected_logs)
91
92     def test_rollout(self):
93         rollout = Rollout()
94         self.assertEqual(([1234], "testing foo"),
95                           rollout._parse_args(["1234", "testing", "foo"]))
96
97         self.assertEqual(([554], "testing foo"),
98                           rollout._parse_args(["r554", "testing", "foo"]))
99
100         self.assertEqual(([556, 792], "testing foo"),
101                           rollout._parse_args(["r556", "792", "testing", "foo"]))
102
103         self.assertEqual(([128, 256], "testing foo"),
104                           rollout._parse_args(["r128,r256", "testing", "foo"]))
105
106         self.assertEqual(([512, 1024, 2048], "testing foo"),
107                           rollout._parse_args(["512,", "1024,2048", "testing", "foo"]))
108
109         # Test invalid argument parsing:
110         self.assertEqual((None, None), rollout._parse_args([]))
111         self.assertEqual((None, None), rollout._parse_args(["--bar", "1234"]))
112
113         # Invalid arguments result in the USAGE message.
114         self.assertEqual("tom: Usage: rollout SVN_REVISION [SVN_REVISIONS] REASON",
115                           rollout.execute("tom", [], None, None))
116
117         tool = MockTool()
118         tool.filesystem.files["/mock-checkout/test/file/one"] = ""
119         tool.filesystem.files["/mock-checkout/test/file/two"] = ""
120         self.assertEqual("Failed to apply reverse diff for file(s): test/file/one, test/file/two",
121                           rollout._check_diff_failure("""
122 Preparing rollout for bug 123456.
123 Updating working directory
124 Failed to apply reverse diff for revision 123456 because of the following conflicts:
125 test/file/one
126 test/file/two
127 Failed to apply reverse diff for revision 123456 because of the following conflicts:
128 test/file/one
129 test/file/two
130 Updating OpenSource
131 Current branch master is up to date.
132         """, tool))
133         self.assertEqual(None, rollout._check_diff_failure("""
134 Preparing rollout for bug 123456.
135 Updating working directory
136 Some other error report involving file paths:
137 test/file/one
138 test/file/two
139 Updating OpenSource
140 Current branch master is up to date.
141         """, tool))
142
143         # FIXME: We need a better way to test IRCCommands which call tool.irc().post()