1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
"""Unit tests for buildscripts.patch_builds.change_data.py."""
from __future__ import absolute_import
import os
import unittest
from mock import patch, MagicMock
import buildscripts.patch_builds.change_data as under_test
NS = "buildscripts.patch_builds.change_data"
def ns(relative_name): # pylint: disable=invalid-name
"""Return a full name from a name relative to the test module"s name space."""
return NS + "." + relative_name
def create_mock_repo(working_dir=""):
return MagicMock(working_dir=working_dir)
class TestFindChangedFilesInRepos(unittest.TestCase):
@patch(ns("find_changed_files"))
def test_nothing_found(self, changed_files_mock):
repos_mock = [MagicMock()]
changed_files_mock.return_value = set()
self.assertEqual(0, len(under_test.find_changed_files_in_repos(repos_mock)))
@patch(ns("find_changed_files"))
def test_changed_files_in_multiple_repos(self, changed_files_mock):
repos_mock = [MagicMock(), MagicMock()]
first_repo_file_changes = [
os.path.join("jstests", "test1.js"),
os.path.join("jstests", "test1.cpp"),
]
second_repo_file_changes = [
os.path.join("jstests", "test2.js"),
]
changed_files_mock.side_effect = [first_repo_file_changes, second_repo_file_changes]
self.assertEqual(3, len(under_test.find_changed_files_in_repos(repos_mock)))
class TestGenerateRevisionMap(unittest.TestCase):
def test_mongo_revisions_is_mapped_correctly(self):
mock_repo_list = [create_mock_repo(os.getcwd()), create_mock_repo("/path/to/enterprise")]
revision_data = {"mongo": "revision1234", "enterprise": "revision5678"}
revision_map = under_test.generate_revision_map(mock_repo_list, revision_data)
self.assertEqual(revision_map[mock_repo_list[0].git_dir], revision_data["mongo"])
self.assertEqual(revision_map[mock_repo_list[1].git_dir], revision_data["enterprise"])
def test_missing_revisions_are_not_returned(self):
mock_repo_list = [create_mock_repo(os.getcwd()), create_mock_repo("/path/to/enterprise")]
revision_data = {"mongo": "revision1234"}
revision_map = under_test.generate_revision_map(mock_repo_list, revision_data)
self.assertEqual(revision_map[mock_repo_list[0].git_dir], revision_data["mongo"])
self.assertEqual(len(revision_map), 1)
def test_missing_repos_are_not_returned(self):
mock_repo_list = [create_mock_repo(os.getcwd())]
revision_data = {"mongo": "revision1234", "enterprise": "revision56768"}
revision_map = under_test.generate_revision_map(mock_repo_list, revision_data)
self.assertEqual(revision_map[mock_repo_list[0].git_dir], revision_data["mongo"])
self.assertEqual(len(revision_map), 1)
|
