X-Git-Url: https://git.rkrishnan.org/?p=tahoe-lafs%2Ftahoe-lafs.git;a=blobdiff_plain;f=src%2Fallmydata%2Ftest%2Fcommon_util.py;fp=src%2Fallmydata%2Ftest%2Fcommon_util.py;h=47dbe59a3fb685ad02cb99c54f7425e71cff9b61;hp=e51ab8b001c0da1a8874f1293ff55827b89ac70f;hb=5b33f1fdf3964db3a7475df36d5e83db46f6b002;hpb=c86388b79c34ba0e71bee01e5c739db2ae6e3403

diff --git a/src/allmydata/test/common_util.py b/src/allmydata/test/common_util.py
index e51ab8b0..47dbe59a 100644
--- a/src/allmydata/test/common_util.py
+++ b/src/allmydata/test/common_util.py
@@ -173,6 +173,24 @@ class TestMixin(SignalMixin):
         if required_to_quiesce and active:
             self.fail("Reactor was still active when it was required to be quiescent.")
 
+
+class TimezoneMixin(object):
+
+    def setTimezone(self, timezone):
+        unset = object()
+        originalTimezone = os.environ.get('TZ', unset)
+        def restoreTimezone():
+            if originalTimezone is unset:
+                del os.environ['TZ']
+                time.tzset()
+            else:
+                os.environ['TZ'] = originalTimezone
+                time.tzset()
+        os.environ['TZ'] = timezone
+        time.tzset()
+        self.addCleanup(restoreTimezone)
+
+
 try:
     import win32file
     import win32con