mirror of https://github.com/mongodb/mongo
55 lines
1.9 KiB
Python
55 lines
1.9 KiB
Python
"""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-88-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-88-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-88-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)
|