Advertisement
Guest User

Untitled

a guest
May 18th, 2012
574
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.67 KB | None | 0 0
  1. import unittest
  2. import datetime
  3. from mock import patch
  4.  
  5. class SomeTestCase(unittest.TestCase):
  6.     def setUp(self):
  7.         self.time = datetime.datetime(2012, 5, 18)
  8.         class fakedatetime(datetime.datetime):
  9.             @classmethod
  10.             def now(cls):
  11.                 return self.time
  12.         patcher = patch('datetime.datetime', fakedatetime)
  13.         self.addCleanup(patcher.stop)
  14.         patcher.start()
  15.  
  16.     def test_something(self):
  17.         self.assertEqual(datetime.datetime.now(), datetime.datetime(2012, 5, 18))
  18.         self.time = datetime.datetime(2012, 5, 20)
  19.         self.assertEqual(datetime.datetime.now(), datetime.datetime(2012, 5, 20))
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement