กำลังพยายามจำลอง datetime.date.today () แต่ไม่ทำงาน

มีใครบอกฉันได้ไหมว่าทำไมสิ่งนี้ถึงไม่ทำงาน

>>> import mock
>>> @mock.patch('datetime.date.today')
... def today(cls):
...  return date(2010, 1, 1)
...
>>> from datetime import date
>>> date.today()
datetime.date(2010, 12, 19)

บางทีบางคนอาจแนะนำวิธีที่ดีกว่า

มีปัญหาเล็กน้อย

ก่อนอื่นวิธีที่คุณใช้mock.patchไม่ถูกต้อง เมื่อใช้เป็นมัณฑนากรจะแทนที่ได้รับฟังก์ชั่น / คลาส (ในกรณีนี้datetime.date.today) โดยมีMockวัตถุเฉพาะภายในฟังก์ชั่นการตกแต่ง ดังนั้นเฉพาะภายในคุณเท่านั้นที่today()จะdatetime.date.todayมีฟังก์ชั่นที่แตกต่างกันซึ่งไม่ปรากฏว่าเป็นสิ่งที่คุณต้องการ

สิ่งที่คุณต้องการดูเหมือนจะเป็นแบบนี้มากขึ้น:

@mock.patch('datetime.date.today')
def test():
    datetime.date.today.return_value = date(2010, 1, 1)
    print datetime.date.today()

น่าเสียดายที่วิธีนี้ใช้ไม่ได้:

>>> test()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "build/bdist.macosx-10.6-universal/egg/mock.py", line 557, in patched
  File "build/bdist.macosx-10.6-universal/egg/mock.py", line 620, in __enter__
TypeError: can't set attributes of built-in/extension type 'datetime.date'

สิ่งนี้ล้มเหลวเนื่องจากชนิดในตัวของ Python ไม่เปลี่ยนรูป - ดูคำตอบนี้สำหรับรายละเอียดเพิ่มเติม

ในกรณีนี้ฉันจะ subclass datetime.date ตัวเองและสร้างฟังก์ชั่นที่เหมาะสม:

import datetime
class NewDate(datetime.date):
    @classmethod
    def today(cls):
        return cls(2010, 1, 1)
datetime.date = NewDate

และตอนนี้คุณสามารถทำได้:

>>> datetime.date.today()
NewDate(2010, 1, 1)

ตัวเลือกอื่นคือใช้ https://github.com/spulec/freezegun/

ติดตั้ง:

pip install freezegun

และใช้มัน:

from freezegun import freeze_time

@freeze_time("2012-01-01")
def test_something():

    from datetime import datetime
    print(datetime.now()) #  2012-01-01 00:00:00

    from datetime import date
    print(date.today()) #  2012-01-01

นอกจากนี้ยังมีผลต่อการเรียกใช้วันที่และเวลาอื่น ๆ ในวิธีการโทรจากโมดูลอื่น ๆ :

other_module.py:

from datetime import datetime

def other_method():
    print(datetime.now())    

main.py:

from freezegun import freeze_time

@freeze_time("2012-01-01")
def test_something():

    import other_module
    other_module.other_method()

และในที่สุดก็:

$ python main.py
# 2012-01-01

สำหรับสิ่งที่คุ้มค่า Mock docs พูดถึง datetime.date.today โดยเฉพาะและเป็นไปได้ที่จะทำสิ่งนี้โดยไม่ต้องสร้างคลาสจำลอง

https://docs.python.org/3/library/unittest.mock-examples.html#partial-mocking

>>> from datetime import date
>>> with patch('mymodule.date') as mock_date:
...     mock_date.today.return_value = date(2010, 10, 8)
...     mock_date.side_effect = lambda *args, **kw: date(*args, **kw)
...
...     assert mymodule.date.today() == date(2010, 10, 8)
...     assert mymodule.date(2009, 6, 8) == date(2009, 6, 8)
...

ฉันคิดว่าฉันมาสายเล็กน้อยสำหรับเรื่องนี้ แต่ฉันคิดว่าปัญหาหลักของที่นี่คือคุณกำลังแก้ไข datetime.date.today โดยตรงและตามเอกสารอธิบายสิ่งนี้ผิด

คุณควรแก้ไขข้อมูลอ้างอิงที่นำเข้ามาในไฟล์โดยใช้ฟังก์ชันที่ทดสอบ

สมมติว่าคุณมีไฟล์ function.py ที่คุณมีต่อไปนี้:

import datetime

def get_today():
    return datetime.date.today()

จากนั้นในการทดสอบของคุณคุณควรมีสิ่งนี้

import datetime
import unittest

from functions import get_today
from mock import patch, Mock

class GetTodayTest(unittest.TestCase):

    @patch('functions.datetime')
    def test_get_today(self, datetime_mock):
        datetime_mock.date.today = Mock(return_value=datetime.strptime('Jun 1 2005', '%b %d %Y'))
        value = get_today()
        # then assert your thing...

หวังว่านี่จะช่วยได้เล็กน้อย

วิธีเพิ่มโซลูชันของ Daniel G :

from datetime import date

class FakeDate(date):
    "A manipulable date replacement"
    def __new__(cls, *args, **kwargs):
        return date.__new__(date, *args, **kwargs)

สิ่งนี้จะสร้างคลาสที่เมื่ออินสแตนซ์จะส่งคืนออบเจกต์ datetime.date ปกติ แต่ยังสามารถเปลี่ยนได้

@mock.patch('datetime.date', FakeDate)
def test():
    from datetime import date
    FakeDate.today = classmethod(lambda cls: date(2010, 1, 1))
    return date.today()

test() # datetime.date(2010, 1, 1)

ฉันเผชิญกับสถานการณ์เดียวกันเมื่อสองสามวันก่อนและโซลูชันของฉันคือการกำหนดฟังก์ชั่นในโมดูลเพื่อทดสอบและเพียงแค่ล้อเลียนว่า:

def get_date_now():
    return datetime.datetime.now()

วันนี้ฉันค้นพบFreezeGunและดูเหมือนว่าจะครอบคลุมกรณีนี้อย่างสวยงาม

from freezegun import freeze_time
import datetime
import unittest


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

วิธีที่ง่ายที่สุดสำหรับฉันคือทำสิ่งนี้:

import datetime
from unittest.mock import Mock, patch

def test():
    datetime_mock = Mock(wraps=datetime.datetime)
    datetime_mock.now.return_value = datetime.datetime(1999, 1, 1)
    with patch('datetime.datetime', new=datetime_mock):
        assert datetime.datetime.now() == datetime.datetime(1999, 1, 1)

ข้อควรระวังสำหรับโซลูชันนี้: การทำงานทั้งหมดdatetime moduleจากtarget_moduleจะหยุดทำงาน

คุณสามารถใช้วิธีการต่อไปนี้โดยยึดตามโซลูชัน Daniel G isinstance(d, datetime.date)หนึ่งนี้มีประโยชน์จากการไม่ทำลายการตรวจสอบชนิดด้วย

import mock

def fixed_today(today):
    from datetime import date

    class FakeDateType(type):
        def __instancecheck__(self, instance):
            return isinstance(instance, date)

    class FakeDate(date):
        __metaclass__ = FakeDateType

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

        @staticmethod
        def today():
            return today

    return mock.patch("datetime.date", FakeDate)

โดยทั่วไปเราแทนที่ C-ตามdatetime.dateชั้นเรียนกับประเภทรองหลามของเราเองที่ผลิตเดิมdatetime.dateกรณีและตอบสนองต่อคำสั่งตรงตามพื้นเมืองisinstance()datetime.date

ใช้เป็นผู้จัดการบริบทในการทดสอบของคุณ:

with fixed_today(datetime.date(2013, 11, 22)):
    # run the code under test
    # note, that these type checks will not break when patch is active:
    assert isinstance(datetime.date.today(), datetime.date)

วิธีการที่คล้ายกันสามารถใช้ในการจำลองdatetime.datetime.now()ฟังก์ชั่น

โดยทั่วไปคุณจะมีdatetimeหรืออาจจะdatetime.dateนำเข้ามาในบางโมดูล วิธีที่มีประสิทธิภาพมากขึ้นในการเย้ยหยันวิธีนี้คือการแก้ไขบนโมดูลที่กำลังนำเข้า ตัวอย่าง:

a.py

from datetime import date

def my_method():
    return date.today()

จากนั้นสำหรับการทดสอบของคุณวัตถุจำลองจะถูกส่งผ่านเป็นอาร์กิวเมนต์สำหรับวิธีการทดสอบ คุณจะตั้งค่าจำลองด้วยค่าผลลัพธ์ที่คุณต้องการแล้วเรียกวิธีการของคุณภายใต้การทดสอบ จากนั้นคุณจะยืนยันว่าวิธีการของคุณทำในสิ่งที่คุณต้องการ

>>> import mock
>>> import a
>>> @mock.patch('a.date')
... def test_my_method(date_mock):
...     date_mock.today.return_value = mock.sentinel.today
...     result = a.my_method()
...     print result
...     date_mock.today.assert_called_once_with()
...     assert mock.sentinel.today == result
...
>>> test_my_method()
sentinel.today

คำเตือน เป็นไปได้อย่างแน่นอนที่จะลงเรือด้วยการเยาะเย้ย เมื่อคุณทำมันจะทำให้การทดสอบของคุณนานขึ้นยากที่จะเข้าใจและเป็นไปไม่ได้ที่จะรักษา ก่อนที่คุณจะจำลองวิธีการง่ายๆdatetime.date.todayให้ถามตัวเองว่าคุณจำเป็นต้องจำลองหรือไม่ หากการทดสอบของคุณสั้นและตรงประเด็นและทำงานได้ดีโดยไม่มีการเยาะเย้ยฟังก์ชั่นคุณอาจมองไปที่รายละเอียดภายในของรหัสที่คุณกำลังทดสอบแทนที่จะเป็นวัตถุที่คุณต้องจำลอง

นี่เป็นอีกวิธีในการเยาะเย้ยdatetime.date.today()ด้วยโบนัสเพิ่มเติมที่ส่วนที่เหลือของdatetimeฟังก์ชั่นยังคงทำงานตามที่วัตถุจำลองถูกกำหนดค่าให้ห่อdatetimeโมดูลเดิม:

from unittest import mock, TestCase

import foo_module

class FooTest(TestCase):

    @mock.patch(f'{foo_module.__name__}.datetime', wraps=datetime)
    def test_something(self, mock_datetime):
        # mock only datetime.date.today()
        mock_datetime.date.today.return_value = datetime.date(2019, 3, 15)
        # other calls to datetime functions will be forwarded to original datetime

หมายเหตุwraps=datetimeอาร์กิวเมนต์ถึงmock.patch()- เมื่อการfoo_moduleใช้งานdatetimeฟังก์ชั่นอื่น ๆนอกเหนือจากdate.today()นั้นจะถูกส่งต่อไปยังdatetimeโมดูลห่อเดิม

การแก้ปัญหาหลายอย่างที่กล่าวไว้ในhttp://blog.xelnor.net/python-mocking-datetime/ สรุป:

วัตถุจำลอง - ง่ายและมีประสิทธิภาพ แต่แบ่ง isinstance () ตรวจสอบ:

target = datetime.datetime(2009, 1, 1)
with mock.patch.object(datetime, 'datetime', mock.Mock(wraps=datetime.datetime)) as patched:
    patched.now.return_value = target
    print(datetime.datetime.now())

ชั้นเลียนแบบ

import datetime
import mock

real_datetime_class = datetime.datetime

def mock_datetime_now(target, dt):
    class DatetimeSubclassMeta(type):
        @classmethod
        def __instancecheck__(mcs, obj):
            return isinstance(obj, real_datetime_class)

    class BaseMockedDatetime(real_datetime_class):
        @classmethod
        def now(cls, tz=None):
            return target.replace(tzinfo=tz)

        @classmethod
        def utcnow(cls):
            return target

    # Python2 & Python3 compatible metaclass
    MockedDatetime = DatetimeSubclassMeta('datetime', (BaseMockedDatetime,), {})

    return mock.patch.object(dt, 'datetime', MockedDatetime)

ใช้เป็น:

with mock_datetime_now(target, datetime):
   ....

บางทีคุณอาจใช้วิธี "today ()" ของคุณเองที่คุณจะแก้ไขเมื่อจำเป็น ตัวอย่างที่มีการเยาะเย้ย utcnow () สามารถดูได้ที่นี่: https://bitbucket.org/k_bx/blog/src/tip/source/en_posts/2012-07-13-double-call-hack.rst?at=default

ฉันใช้วิธีการ @ user3016183 โดยใช้มัณฑนากรที่กำหนดเอง:

def changeNow(func, newNow = datetime(2015, 11, 23, 12, 00, 00)):
    """decorator used to change datetime.datetime.now() in the tested function."""
    def retfunc(self):                             
        with mock.patch('mymodule.datetime') as mock_date:                         
            mock_date.now.return_value = newNow
            mock_date.side_effect = lambda *args, **kw: datetime(*args, **kw)
            func(self)
    return retfunc

ฉันคิดว่าอาจช่วยใครซักคนในหนึ่งวัน ...

เป็นไปได้ที่จะเลียนแบบฟังก์ชั่นจากdatetimeโมดูลโดยไม่ต้องเพิ่มside_effects

import mock
from datetime import datetime
from where_datetime_used import do

initial_date = datetime.strptime('2018-09-27', "%Y-%m-%d")
with mock.patch('where_datetime_used.datetime') as mocked_dt:
    mocked_dt.now.return_value = initial_date
    do()

สำหรับผู้ที่ใช้ pytest กับผู้เยาะเย้ยนี่คือวิธีที่ฉันเยาะเย้ยdatetime.datetime.now()ซึ่งคล้ายกับคำถามเริ่มแรก

test_get_now(mocker):
    datetime_mock = mocker.patch("blackline_accounts_import.datetime",)
    datetime_mock.datetime.now.return_value=datetime.datetime(2019,3,11,6,2,0,0)

    now == function_being_tested()  # run function

    assert now == datetime.datetime(2019,3,11,6,2,0,0)

โดยพื้นฐานแล้วการจำลองจะต้องตั้งค่าให้ส่งกลับวันที่ที่ระบุ คุณไม่สามารถแก้ไขวัตถุของ datetime ได้โดยตรง

ผมทำผลงานชิ้นนี้โดยการนำเข้าdatetimeเป็นrealdatetimeและการเปลี่ยนวิธีการที่ผมจำเป็นต้องมีในการเยาะเย้ยด้วยวิธีจริง:

import datetime as realdatetime

@mock.patch('datetime')
def test_method(self, mock_datetime):
    mock_datetime.today = realdatetime.today
    mock_datetime.now.return_value = realdatetime.datetime(2019, 8, 23, 14, 34, 8, 0)

คุณสามารถเยาะเย้ยdatetimeโดยใช้สิ่งนี้:

ในโมดูลsources.py:

import datetime


class ShowTime:
    def current_date():
        return datetime.date.today().strftime('%Y-%m-%d')

ในtests.py:

from unittest import TestCase, mock
import datetime


class TestShowTime(TestCase):
    def setUp(self) -> None:
        self.st = sources.ShowTime()
        super().setUp()

    @mock.patch('sources.datetime.date')
    def test_current_date(self, date_mock):
        date_mock.today.return_value = datetime.datetime(year=2019, month=10, day=1)
        current_date = self.st.current_date()
        self.assertEqual(current_date, '2019-10-01')

CPython จริงใช้โมดูล datetime โดยใช้ทั้งบริสุทธิ์หลามLib / datetime.pyและ C-เพิ่มประสิทธิภาพโมดูล / _datetimemodule.c เวอร์ชันที่ปรับให้เหมาะกับ C ไม่สามารถแก้ไขได้ แต่เวอร์ชัน Python บริสุทธิ์สามารถ

ที่ด้านล่างของการใช้งาน pure-Python ในLib / datetime.pyคือรหัสนี้:

try:
    from _datetime import *  # <-- Import from C-optimized module.
except ImportError:
    pass

รหัสนี้นำเข้าคำจำกัดความที่ปรับให้เหมาะสมทั้งหมดและแทนที่คำจำกัดความ pure-Python ทั้งหมดอย่างมีประสิทธิภาพ เราสามารถบังคับให้ CPython ใช้การใช้งาน Python แบบบริสุทธิ์ของโมดูล datetime โดยทำ:

import datetime
import importlib
import sys

sys.modules["_datetime"] = None
importlib.reload(datetime)

โดยการตั้งค่าsys.modules["_datetime"] = Noneเราบอก Python ให้ละเว้นโมดูล C-optimization จากนั้นเราโหลดโมดูลซึ่งทำให้การนำเข้า_datetimeล้มเหลว ตอนนี้คำจำกัดความ pure-Python ยังคงอยู่และสามารถแก้ไขได้ตามปกติ

หากคุณใช้Pytest ให้รวมข้อมูลโค้ดด้านบนในconftest.pyและคุณสามารถแก้ไขdatetimeวัตถุได้ตามปกติ