Come prendere in giro python's datetime.now() in un metodo di classe per il test delle unità?

Question

sto cercando di scrivere i test per una classe che ha metodi come:

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 

voglio provarlo, e per farlo ho bisogno di fare in modo che la chiamata datetime.datetime.now() restituisce qualcosa di prevedibile.

Ho letto molti esempi di utilizzo di Mock nei test, ma non ho trovato nulla di simile a quello che mi serve, e non riesco a capire come usarlo nei miei test.

Ho separato il metodo get_now() nel caso fosse più facile prenderlo in giro, invece di datetime.datetime.now(), ma sono ancora bloccato. Qualche idea su come scrivere UnitTests per questo usando Mock? (Questo è tutto in Django, FWIW, io non sono sicuro se questo fa la differenza in questo caso.)

Answer 1

Faresti crea una funzione che restituisce un datetime specifica, localizzata al fuso orario in passato:

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. 

In questo modo è possibile verificare se viene gestito correttamente il datetime timezone-aware risultante; i risultati altrove dovrebbero mostrare il fuso orario corretto ma avranno una data e un'ora prevedibili.

Si utilizza la funzione mocked_get_now come effetto collaterale quando si deride get_now; quando il codice chiama get_now la chiamata viene registrata da mock, emocked_get_now viene chiamato e il suo valore di ritorno viene utilizzato come valore restituito al chiamante di get_now.

Answer 2

sto usando date, ma la stessa idea dovrebbe funzionare per 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)) 

Dove some.module importazioni date. Patch sostituisce lo date importato con SpoofDate, che puoi ridefinire per fare quello che vuoi.

Answer 3

vorrei utilizzare gli aiutanti dal pacchetto 'testfixtures' prendere in giro la classe datetime si sta chiamando ora() su:

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

In questo modo, è possibile verificare tutti i casi che avete , tutto il tempo.

Answer 4

Si potrebbe utilizzare 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) 

Si prende gioco fondamentalmente datetime chiamate moduli.