テストを修正

chaemon · chaemon · commit fbf045e6601d · 2022-01-13T17:29:34.000+09:00
diff --git a/tests/resources/test_setter/ans/main.java b/tests/resources/test_setter/ans/main.java
@@ -3,7 +3,7 @@
 
 class Main {
 
-    // Generated by x.x.x https://github.com/kyuridenamida/atcoder-tools  (tips: You use the default template now. You can remove this line by using your custom template)
+    // Generated by x.x.x
     public static void main(String[] args) throws Exception {
         final Scanner sc = new Scanner(System.in);
         long T;
@@ -17,3 +17,4 @@ static void solve(long T, long X){
 
     }
 }
+
diff --git a/tests/test_setter.py b/tests/test_setter.py
@@ -27,6 +27,17 @@ def get_all_rel_file_paths(dir_path: str):
     return sorted(res)
 
 
+def _strip_version_info_java(java_source: str) -> str:
+    result = ""
+    for line in java_source.split("\n"):
+        if line.startswith("    // Generated by"):
+            result += "    // Generated by x.x.x"
+        else:
+            result += line
+        result += "\n"
+    return result
+
+
 class TestSetter(unittest.TestCase):
 
     def setUp(self):
@@ -47,7 +58,7 @@ def test_setter_change_lang(self):
              "--dir", test_dir,
              "--without-login"]
         )
-        self.assertEqual(open(os.path.join(test_dir, "main.java")).read(),
+        self.assertEqual(_strip_version_info_java(open(os.path.join(test_dir, "main.java")).read()),
                          open(os.path.join(RESOURCE_DIR, "ans", "main.java")).read())
 
         metadata = Metadata.load_from(os.path.join(test_dir, "metadata.json"))

Original file line number	Diff line number	Diff line change
`@@ -3,7 +3,7 @@`
`3`	`3`
`4`	`4`	`class Main {`
`5`	`5`
`6`		`- // Generated by x.x.x https://github.com/kyuridenamida/atcoder-tools (tips: You use the default template now. You can remove this line by using your custom template)`
	`6`	`+ // Generated by x.x.x`
`7`	`7`	`public static void main(String[] args) throws Exception {`
`8`	`8`	`final Scanner sc = new Scanner(System.in);`
`9`	`9`	`long T;`
`@@ -17,3 +17,4 @@ static void solve(long T, long X){`
`17`	`17`
`18`	`18`	`}`
`19`	`19`	`}`
	`20`	`+`