From 8d251eca144e4c5c68155331f129fa02f5b4fa48 Mon Sep 17 00:00:00 2001
From: Scott Worley <scottworley@scottworley.com>
Date: Mon, 21 Jun 2021 12:56:06 -0700
Subject: [PATCH] _setenv()

---
 test_git_cache.py | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/test_git_cache.py b/test_git_cache.py
index caac6a3..bbe7118 100644
--- a/test_git_cache.py
+++ b/test_git_cache.py
@@ -4,9 +4,18 @@ import shutil
 import subprocess
 import unittest
 
+from typing import Optional
+
 import git_cache
 
 
+def _setenv(var: str, value: Optional[str]) -> None:
+    if value is None:
+        del os.environ[var]
+    else:
+        os.environ[var] = value
+
+
 def _git(directory: str, *args: str) -> bytes:
     p = subprocess.run(['git', '-C', directory] + list(args),
                        stdout=subprocess.PIPE, check=True)
@@ -30,7 +39,7 @@ class TestGitCache(unittest.TestCase):
     def setUp(self) -> None:
         self.xdgcache = tempfile.TemporaryDirectory(prefix='git_cache_test-')
         self.old_XDG_CACHE_HOME = os.environ.get('XDG_CACHE_HOME')
-        os.environ['XDG_CACHE_HOME'] = self.xdgcache.name
+        _setenv('XDG_CACHE_HOME', self.xdgcache.name)
 
         os.environ['GIT_AUTHOR_NAME'] = 'test_git_cache'
         os.environ['GIT_COMMITTER_NAME'] = 'test_git_cache'
@@ -45,10 +54,7 @@ class TestGitCache(unittest.TestCase):
         _commit_file(self.upstream, 'file', 'Contents', 'First commit')
 
     def tearDown(self) -> None:
-        if self.old_XDG_CACHE_HOME is None:
-            del os.environ['XDG_CACHE_HOME']
-        else:
-            os.environ['XDG_CACHE_HOME'] = self.old_XDG_CACHE_HOME
+        _setenv('XDG_CACHE_HOME', self.old_XDG_CACHE_HOME)
 
         self.tempdir.cleanup()
         self.xdgcache.cleanup()
-- 
2.44.1