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
|
"""Unit tests for buildscripts/resmokelib/suitesconfig.py."""
import unittest
import mock
from buildscripts.resmokelib import suitesconfig
from buildscripts.resmokelib import parser
parser.set_run_options()
# pylint: disable=missing-docstring
RESMOKELIB = "buildscripts.resmokelib"
class TestSuitesConfig(unittest.TestCase):
@mock.patch(RESMOKELIB + ".testing.suite.Suite")
@mock.patch(RESMOKELIB + ".suitesconfig.get_named_suites")
def test_no_suites_matching_test_kind(self, mock_get_named_suites, mock_suite_class):
all_suites = ["core", "replica_sets_jscore_passthrough"]
mock_get_named_suites.return_value = all_suites
membership_map = suitesconfig.create_test_membership_map(test_kind="nonexistent_test")
self.assertEqual(membership_map, {})
self.assertEqual(mock_suite_class.call_count, 0)
@mock.patch(RESMOKELIB + ".testing.suite.Suite")
@mock.patch(RESMOKELIB + ".suitesconfig.get_named_suites")
def test_multiple_suites_matching_single_test_kind(self, mock_get_named_suites,
mock_suite_class):
all_suites = ["core", "replica_sets_jscore_passthrough"]
mock_get_named_suites.return_value = all_suites
mock_suite_class.return_value.tests = ["test1", "test2"]
membership_map = suitesconfig.create_test_membership_map(test_kind="js_test")
self.assertEqual(membership_map, dict(test1=all_suites, test2=all_suites))
self.assertEqual(mock_suite_class.call_count, 2)
@mock.patch(RESMOKELIB + ".testing.suite.Suite")
@mock.patch(RESMOKELIB + ".suitesconfig.get_named_suites")
def test_multiple_suites_matching_multiple_test_kinds(self, mock_get_named_suites,
mock_suite_class):
all_suites = ["core", "concurrency"]
mock_get_named_suites.return_value = all_suites
mock_suite_class.return_value.tests = ["test1", "test2"]
membership_map = suitesconfig.create_test_membership_map(
test_kind=("fsm_workload_test", "js_test"))
self.assertEqual(membership_map, dict(test1=all_suites, test2=all_suites))
self.assertEqual(mock_suite_class.call_count, 2)
|