From cac2f5a9be79ebafaa5224c860ab09cb068d4af4 Mon Sep 17 00:00:00 2001 From: Alan Hamlett Date: Sun, 30 Sep 2018 20:03:59 -0700 Subject: [PATCH] fix tests on Windows with Python 2 --- tests/utils.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/tests/utils.py b/tests/utils.py index 0086d65..0d5003a 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -2,13 +2,14 @@ import logging import os +import platform import re import sys import tempfile from wakatime.compat import u from wakatime.packages.requests.models import Response -from wakatime.utils import BACKSLASH_REPLACE_PATTERN +from wakatime.utils import BACKSLASH_REPLACE_PATTERN, WINDOWS_DRIVE_PATTERN try: @@ -73,7 +74,10 @@ class TestCase(unittest.TestCase): self.assertEquals(first_list, second_list) def normalize_path(self, path): - return re.sub(BACKSLASH_REPLACE_PATTERN, '/', path) + filepath = re.sub(BACKSLASH_REPLACE_PATTERN, '/', path) + if platform.system() == 'Windows' and WINDOWS_DRIVE_PATTERN.match(filepath): + filepath = filepath.capitalize() + return filepath def assertPathsEqual(self, first_path, second_path, message=None): if message: