Make tests pass on Windows.

eerimoq · Jan 6, 2020 · 1112d07 · 1112d07
1 parent f9775ee
commit 1112d07
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/tests/test_parser.py b/tests/test_parser.py
@@ -1,3 +1,4 @@
+import os
 import unittest
 
 import pbtools
@@ -9,8 +10,9 @@ def test_int32(self):
         parsed = pbtools.parse_file('tests/files/int32.proto')
 
         self.assertEqual(parsed.package, 'int32')
-        self.assertTrue(parsed.abspath.startswith('/'))
-        self.assertTrue(parsed.abspath.endswith('/tests/files/int32.proto'))
+        self.assertTrue(os.path.isabs(parsed.abspath))
+        self.assertTrue(
+            parsed.abspath.endswith(os.path.join('tests', 'files', 'int32.proto')))
         self.assertEqual(parsed.imports, [])
         self.assertEqual(parsed.options, [])
         self.assertEqual(len(parsed.messages), 2)
@@ -591,7 +593,7 @@ def test_importing(self):
         self.assertEqual(imported.path, 'imported.proto')
         self.assertEqual(imported.package, 'imported')
         self.assertTrue(
-            imported.abspath.endswith('/tests/files/imported.proto'))
+            imported.abspath.endswith(os.path.join('tests', 'files', 'imported.proto')))
         self.assertEqual(imported.enums, ['ImportedEnum'])
         self.assertEqual(imported.messages, ['ImportedMessage'])
 
@@ -600,7 +602,7 @@ def test_importing(self):
         self.assertEqual(imported.package, 'imported')
         self.assertTrue(
             imported.abspath.endswith(
-                '/tests/files/imports/imported_duplicated_package.proto'))
+                os.path.join('tests', 'files', 'imports', 'imported_duplicated_package.proto')))
         self.assertEqual(imported.enums, ['ImportedDuplicatedPackageEnum'])
         self.assertEqual(imported.messages,
                          ['Imported2Message', 'ImportedDuplicatedPackageMessage'])
@@ -609,7 +611,8 @@ def test_importing(self):
         self.assertEqual(imported.path, 'imported2.proto')
         self.assertEqual(imported.package, 'imported2.foo.bar')
         self.assertTrue(
-            imported.abspath.endswith('/tests/files/imports/imported2.proto'))
+            imported.abspath.endswith(
+                os.path.join('tests', 'files', 'imports', 'imported2.proto')))
         self.assertEqual(imported.enums, ['Imported2Enum'])
         self.assertEqual(imported.messages,
                          ['Imported2Message', 'Imported3Message'])