1 files changed, 34 insertions, 0 deletions
diff --git a/buildscripts/tests/resmokelib/discovery/test_discovery.py b/buildscripts/tests/resmokelib/discovery/test_discovery.py
new file mode 100644
index 00000000000..e3a45e03e63
--- /dev/null
+++ b/buildscripts/tests/resmokelib/discovery/test_discovery.py
@@ -0,0 +1,34 @@
+"""Unit tests for discovery subcommand."""
+
+import unittest
+from unittest.mock import MagicMock
+
+import buildscripts.resmokelib.discovery as under_test
+from buildscripts.resmokelib.testing.suite import Suite
+
+# pylint: disable=missing-docstring
+
+
+class TestTestDiscoverySubCommand(unittest.TestCase):
+    def test_gather_tests_should_return_discovered_tests(self):
+        suite_name = "my suite"
+        mock_suite = MagicMock(spec_set=Suite)
+        mock_suite.get_display_name.return_value = suite_name
+        mock_suite.tests = [
+            "test_0.js",
+            "test_1.js",
+            [
+                "test_2.js",
+                "test_3.js",
+                "test_4.js",
+            ],
+            "test_5.js",
+        ]
+        test_discovery_subcommand = under_test.TestDiscoverySubcommand(suite_name)
+
+        tests = test_discovery_subcommand.gather_tests(mock_suite)
+
+        self.assertEqual(suite_name, tests.suite_name)
+
+        for i in range(6):
+            self.assertIn(f"test_{i}.js", tests.tests)