Update source_unittest to specify the timezone so file checks pass properly 
when scraping rsync output.

Signed-off-by: Scott Zawalski <[email protected]>

--- autotest/mirror/source_unittest.py  2010-03-08 19:03:35.000000000 -0800
+++ autotest/mirror/source_unittest.py  2010-03-08 19:03:35.000000000 -0800
@@ -1,7 +1,7 @@
 #!/usr/bin/python
 # Copyright 2009 Google Inc. Released under the GPL v2
 
-import unittest, cStringIO, httplib
+import unittest, cStringIO, httplib, time, os
 
 import common
 from autotest_lib.mirror import source
@@ -65,10 +65,15 @@
         self.db_mock = self.god.create_mock_class(
             source.database.database, 'database')
         self.god.stub_function(source.utils, 'system_output')
+        self.old_tz = getattr(os.environ, 'TZ', '')
+        os.environ['TZ'] = 'America/Los_Angeles'
+        time.tzset()
 
 
     def tearDown(self):
         self.god.unstub_all()
+        os.environ['TZ'] = self.old_tz
+        time.tzset()
 
 
     def test_simple(self):
@@ -224,10 +229,15 @@
             source.urllib2.addinfourl, 'addinfourl')
         self.mime_mock = self.god.create_mock_class(
             httplib.HTTPMessage, 'HTTPMessage')
+        self.old_tz = getattr(os.environ, 'TZ', '')
+        os.environ['TZ'] = 'America/Los_Angeles'
+        time.tzset()
 
 
     def tearDown(self):
         self.god.unstub_all()
+        os.environ['TZ'] = self.old_tz
+        time.tzset()
 
 
     def test_get_new_files(self):
_______________________________________________
Autotest mailing list
[email protected]
http://test.kernel.org/cgi-bin/mailman/listinfo/autotest

Reply via email to