add shell test

googleapis · May 9, 2024 · 509c792 · 509c792
1 parent 608c155
commit 509c792
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 0 deletions.
diff --git a/...ion/java_gapic_srcjar/samples/snippets/generated/src/main/java/com/example_com_sample.txt b/...ion/java_gapic_srcjar/samples/snippets/generated/src/main/java/com/example_com_sample.txt
diff --git a/...ation/java_gapic_srcjar/samples/snippets/generated/src/main/java/io/example_io_sample.txt b/...ation/java_gapic_srcjar/samples/snippets/generated/src/main/java/io/example_io_sample.txt
diff --git a/library_generation/test/utilities_unit_tests.py b/library_generation/test/utilities_unit_tests.py
@@ -113,6 +113,30 @@ def test_sh_util_nonexistent_function_fails(self):
         with self.assertRaises(RuntimeError):
             result = util.sh_util("nonexistent_function")
 
+    def test_mv_src_files_sample_suffix_io_succeeds(self):
+        previous_dir = os.getcwd()
+        os.chdir(f"{resources_dir}/test_mv_src")
+        util.sh_util("mv_src_files samples main destination io")
+        self.assertTrue(
+            os.path.isfile(
+                "destination/samples/snippets/generated/io/example_io_sample.txt"
+            )
+        )
+        shutil.rmtree("destination/samples")
+        os.chdir(previous_dir)
+
+    def test_mv_src_files_sample_suffix_com_succeeds(self):
+        previous_dir = os.getcwd()
+        os.chdir(f"{resources_dir}/test_mv_src")
+        util.sh_util("mv_src_files samples main destination")
+        self.assertTrue(
+            os.path.isfile(
+                "destination/samples/snippets/generated/com/example_com_sample.txt"
+            )
+        )
+        shutil.rmtree("destination/samples")
+        os.chdir(previous_dir)
+
     def test_eprint_valid_input_succeeds(self):
         test_input = "This is some test input"
         # create a stdio capture object