Jak wyśmiać Pythona datetime.now() w metodzie klasy do testowania jednostkowego?

Question

Próbuję napisać testy dla klasy, która ma metody, takie jak:

import datetime 
import pytz 

class MyClass: 
    def get_now(self, timezone): 
     return datetime.datetime.now(timezone) 

    def do_many_things(self, tz_string='Europe/London'): 
     tz = pytz.timezone(tz_string) 
     localtime_now = self.get_now(tz) 
     ... 
     return things 

chcę go przetestować, a do tego muszę się upewnić, że wywołanie datetime.datetime.now() zwraca coś przewidywalnego.

Czytałem dużo przykładów używania testów w Mock, ale nie znalazłem niczego tak bardzo, jak potrzebuję i nie mogę się dowiedzieć, jak z nich korzystać w moich testach.

Wyodrębniłem metodę get_now() na wypadek, gdyby łatwiej było to udawać, zamiast datetime.datetime.now(), ale nadal jestem zakłopotany. Wszelkie przemyślenia na temat pisania UnitTests do tego przy użyciu Mock? (To wszystko w Django, FWIW; nie jestem pewien, czy to robi różnicę w tym przypadku).

Answer 1

Można by utworzyć funkcję, która zwraca konkretny datetime, zlokalizowaną w strefie czasowej przyjętej w:

import mock 

def mocked_get_now(timezone): 
    dt = datetime.datetime(2012, 1, 1, 10, 10, 10) 
    return timezone.localize(dt) 

@mock.patch('path.to.your.models.MyClass.get_now', side_effect=mocked_get_now) 
def your_test(self, mock_obj): 
    # Within this test, `MyClass.get_now()` is a mock that'll return a predictable 
    # timezone-aware datetime object, set to 2012-01-01 10:10:10. 

W ten sposób można przetestować, czy wynikowa datetime świadoma czasu jest poprawnie obsługiwana; Wyniki w innym miejscu powinny pokazywać właściwą strefę czasową, ale będą miały przewidywalną datę i godzinę.

Używana jest funkcja mocked_get_now jako efekt uboczny przy kpieniu z get_now; zawsze, gdy kod wywołuje get_now, połączenie jest rejestrowane przez mock, imocked_get_now jest wywoływane, a jego wartość zwracana jest używana jako wartość zwracana do wywołującego z get_now.

Answer 2

Używam date, ale sama idea powinna pracować dla datetime:

class SpoofDate(date): 
    def __new__(cls, *args, **kwargs): 
     return date.__new__(date, *args, **kwargs) 

...

from mock import patch 

@patch('some.module.date', SpoofDate) 
def testSomething(self): 
    SpoofDate.today = classmethod(lambda cls : date(2012, 9, 24)) 

Gdzie some.module import date. Patch zastępuje zaimportowaną wersję date za pomocą SpoofDate, którą możesz następnie zmienić na nowo, aby robić, co chcesz.

Answer 3

użyłbym pomocników z pakietu do „testfixtures” wyśmiewać się klasę datetime jesteś wzywającą teraz() na:

http://packages.python.org/testfixtures/datetime.html#datetimes

ten sposób można przetestować wszystkie przypadki masz , cały czas.

Answer 4

Można użyć freezegun:

from freezegun import freeze_time 

def test(): 
    assert datetime.datetime.now() != datetime.datetime(2012, 1, 14) 
    with freeze_time("2012-01-14"): 
     assert datetime.datetime.now() == datetime.datetime(2012, 1, 14) 
    assert datetime.datetime.now() != datetime.datetime(2012, 1, 14) 

Zasadniczo drwi datetime połączeń modułu.