use regex in string matching

googleapis · Mar 14, 2024 · 2d15254 · 2d15254
1 parent 013e69c
commit 2d15254
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/java-service-management/owlbot.py b/java-service-management/owlbot.py
@@ -16,28 +16,27 @@
 from synthtool.languages import java
 
 file_name = "owl-bot-staging/v1/google-cloud-service-management/src/test/java/com/google/cloud/api/servicemanagement/v1/ServiceManagerClientHttpJsonTest.java"
-ignore_annotation = '@Ignore("See: https://github.com/googleapis/sdk-platform-java/issues/1839")'
 
 for library in s.get_staging_dirs():
     # put any special-case replacements here
     s.replace(
         file_name,
-        'import org.junit.Test;',
+        r'^import org.junit.Test;',
         'import org.junit.Ignore;\nimport org.junit.Test;'
     )
     s.replace(
         file_name,
-        '@Test\npublic void setIamPolicyTest() throws Exception {',
+        r"\s+@Test\n\s+public void setIamPolicyTest\(\) throws Exception.*",
         '@Ignore("See: https://github.com/googleapis/sdk-platform-java/issues/1839")\n@Test\npublic void setIamPolicyTest() throws Exception {'
     )
     s.replace(
         file_name,
-        '@Test\npublic void getIamPolicyTest() throws Exception {',
+        r'\s+@Test\n\s+public void getIamPolicyTest\(\) throws Exception.*',
         '@Ignore("See: https://github.com/googleapis/sdk-platform-java/issues/1839")\n@Test\npublic void getIamPolicyTest() throws Exception {'
     )
     s.replace(
         file_name,
-        '@Test\npublic void testIamPermissionsTest() throws Exception {',
+        r'\s+@Test\n\s+public void testIamPermissionsTest\(\) throws Exception.*',
         '@Ignore("See: https://github.com/googleapis/sdk-platform-java/issues/1839")@Test\npublic void testIamPermissionsTest() throws Exception {'
     )
     s.move(library)