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 the util/taskname.py script."""
import unittest
import buildscripts.util.taskname as under_test
# pylint: disable=invalid-name
class TestNameTask(unittest.TestCase):
def test_name_task_with_width_one(self):
self.assertEqual("name_3_var", under_test.name_generated_task("name", 3, 10, "var"))
def test_name_task_with_width_four(self):
self.assertEqual("task_3141_var", under_test.name_generated_task("task", 3141, 5000, "var"))
class TestRemoveGenSuffix(unittest.TestCase):
def test_removes_gen_suffix(self):
input_task_name = "sharding_auth_auditg_gen"
self.assertEqual("sharding_auth_auditg", under_test.remove_gen_suffix(input_task_name))
def test_doesnt_remove_non_gen_suffix(self):
input_task_name = "sharded_multi_stmt_txn_jscore_passthroug"
self.assertEqual("sharded_multi_stmt_txn_jscore_passthroug",
under_test.remove_gen_suffix(input_task_name))
class TestDetermineTaskBaseName(unittest.TestCase):
def test_task_name_with_build_variant_should_strip_bv_and_sub_task_index(self):
bv = "enterprise-rhel-80-64-bit-dynamic-all-feature-flags-required"
task_name = f"auth_23_{bv}"
base_task_name = under_test.determine_task_base_name(task_name, bv)
self.assertEqual("auth", base_task_name)
def test_task_name_without_build_variant_should_strip_sub_task_index(self):
bv = "enterprise-rhel-80-64-bit-dynamic-all-feature-flags-required"
task_name = "auth_314"
base_task_name = under_test.determine_task_base_name(task_name, bv)
self.assertEqual("auth", base_task_name)
def test_task_name_without_build_variant_or_subtask_index_should_self(self):
bv = "enterprise-rhel-80-64-bit-dynamic-all-feature-flags-required"
task_name = "auth"
base_task_name = under_test.determine_task_base_name(task_name, bv)
self.assertEqual("auth", base_task_name)
|