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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
"""Unit tests for fuzzer_tasks.py"""
import unittest
import buildscripts.task_generation.task_types.fuzzer_tasks as under_test
# pylint: disable=missing-docstring,invalid-name,unused-argument,no-self-use,protected-access
def build_mock_fuzzer_params(jstestfuzz_vars="vars for jstestfuzz"):
return under_test.FuzzerGenTaskParams(
task_name="task name",
variant="build variant",
suite="resmoke suite",
num_files=10,
num_tasks=5,
resmoke_args="args for resmoke",
npm_command="jstestfuzz",
jstestfuzz_vars=jstestfuzz_vars,
continue_on_failure=True,
resmoke_jobs_max=5,
should_shuffle=True,
timeout_secs=100,
require_multiversion_setup=False,
use_large_distro=None,
large_distro_name="large distro",
config_location="config_location",
)
class TestFuzzerGenTaskParams(unittest.TestCase):
def test_jstestfuzz_params_should_be_generated(self):
params = build_mock_fuzzer_params()
jstestfuzz_vars = params.jstestfuzz_params()
self.assertEqual(params.npm_command, jstestfuzz_vars["npm_command"])
self.assertIn(f"--numGeneratedFiles {params.num_files}", jstestfuzz_vars["jstestfuzz_vars"])
self.assertIn(params.jstestfuzz_vars, jstestfuzz_vars["jstestfuzz_vars"])
def test_jstestfuzz_params_should_handle_no_vars(self):
params = build_mock_fuzzer_params(jstestfuzz_vars=None)
self.assertNotIn("None", params.jstestfuzz_params()["jstestfuzz_vars"])
def test_get_task_name_with_no_version(self):
params = build_mock_fuzzer_params()
self.assertEqual(params.task_name, params.task_name)
def test_get_non_sharded_resmoke_args(self):
params = build_mock_fuzzer_params()
self.assertEqual(params.resmoke_args, params.get_resmoke_args())
class TestGenerateTasks(unittest.TestCase):
def test_fuzzer_tasks_are_generated(self):
mock_params = build_mock_fuzzer_params()
fuzzer_service = under_test.FuzzerGenTaskService()
fuzzer_task = fuzzer_service.generate_tasks(mock_params)
self.assertEqual(fuzzer_task.task_name, mock_params.task_name)
self.assertEqual(len(fuzzer_task.sub_tasks), mock_params.num_tasks)
class TestBuildFuzzerSubTask(unittest.TestCase):
def test_sub_task_should_be_built_correct(self):
mock_params = build_mock_fuzzer_params()
fuzzer_service = under_test.FuzzerGenTaskService()
sub_task = fuzzer_service.build_fuzzer_sub_task(3, mock_params)
self.assertEqual(sub_task.name, f"{mock_params.task_name}_3_{mock_params.variant}")
self.assertEqual(len(sub_task.commands), 6)
def test_sub_task_should_include_timeout_info(self):
mock_params = build_mock_fuzzer_params()
fuzzer_service = under_test.FuzzerGenTaskService()
sub_task = fuzzer_service.build_fuzzer_sub_task(3, mock_params)
cmd = sub_task.commands[0].as_dict()
self.assertEqual(cmd["command"], "timeout.update")
self.assertEqual(cmd["params"]["timeout_secs"], mock_params.timeout_secs)
|