| # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| """Integration tests for bug hunter.""" |
| |
| import csv |
| from optparse import Values |
| import os |
| import unittest |
| |
| from bug_hunter import BugHunter |
| |
| try: |
| import gdata.data |
| import gdata.projecthosting.client |
| except ImportError: |
| logging.error('gdata-client needs to be installed. Please install\n' |
| 'and try again (http://code.google.com/p/gdata-python-client/)') |
| sys.exit(1) |
| |
| |
| class BugHunterTest(unittest.TestCase): |
| """Unit tests for the Bug Hunter class.""" |
| _TEST_FILENAME = 'test.csv' |
| |
| def _CleanTestFile(self): |
| if os.path.exists(self._TEST_FILENAME): |
| os.remove(self._TEST_FILENAME) |
| |
| def setUp(self): |
| self._CleanTestFile() |
| |
| def tearDown(self): |
| self._CleanTestFile() |
| |
| def _GetIssue(self): |
| return [{'issue_id': '0', 'title': 'title', 'author': 'author', |
| 'status': 'status', 'state': 'state', 'content': 'content', |
| 'comments': [], 'labels': [], 'urls': []}] |
| |
| def _GetDefaultOption(self, set_10_days_ago, query='steps'): |
| ops = Values() |
| ops.query = query |
| if set_10_days_ago: |
| ops.interval_value = 10 |
| ops.interval_unit = 'days' |
| else: |
| ops.interval_value = None |
| ops.email_entries = ['comments'] |
| ops.project_name = 'chromium' |
| ops.query_title = 'query title' |
| ops.max_comments = None |
| return ops |
| |
| def testGetIssueReturnedIssue(self): |
| bh = BugHunter( |
| self._GetDefaultOption(False, |
| query=('audio opened-after:2010/10/10' |
| ' opened-before:2010/10/20'))) |
| self.assertEquals(len(bh.GetIssues()), 18) |
| |
| def testGetIssueReturnedIssueWithStatus(self): |
| ops = self._GetDefaultOption(False) |
| ops.query = 'Feature:Media* Status:Unconfirmed' |
| issues = BugHunter(ops).GetIssues() |
| for issue in issues: |
| self.assertEquals(issue['status'], 'Unconfirmed') |
| |
| def testGetIssueReturnNoIssue(self): |
| ops = self._GetDefaultOption(True) |
| ops.query = 'thisshouldnotmatchpleaseignorethis*' |
| self.assertFalse(BugHunter(ops).GetIssues()) |
| |
| def testGetComments(self): |
| comments = BugHunter(self._GetDefaultOption(False)).GetComments(100000, 2) |
| self.assertEquals(len(comments), 2) |
| expected_comments = [(None, 'rby...@chromium.org', |
| '2011-10-31T19:54:40.000Z'), |
| (None, 'backer@chromium.org', |
| '2011-10-14T13:59:37.000Z')] |
| self.assertEquals(comments, expected_comments) |
| |
| def testWriteIssuesToFileInCSV(self): |
| ops = self._GetDefaultOption(False) |
| bh = BugHunter(ops) |
| bh.WriteIssuesToFileInCSV(self._GetIssue(), self._TEST_FILENAME) |
| |
| with open(self._TEST_FILENAME, 'r') as f: |
| reader = csv.reader(f) |
| self.assertEquals(reader.next(), ['status', 'content', 'state', |
| 'issue_id', 'urls', 'title', 'labels', |
| 'author', 'comments']) |
| self.assertEquals(reader.next(), ['status', 'content', 'state', '0', |
| '[]', 'title', '[]', 'author', '[]']) |
| self.assertRaises(StopIteration, reader.next) |
