Python: How do I mock datetime.utcnow()?

后端 未结 5 1066
梦如初夏
梦如初夏 2020-12-30 02:24

I have the below:

from datetime import datetime

def get_report_month_key():
    month_for_report = datetime.utcnow()
    return month_for_report.strftime(\"         


        
5条回答
  •  被撕碎了的回忆
    2020-12-30 02:48

    in your test file:

    from yourfile import get_report_month_key
    import mock
    import unittest
    from datetime import datetime
    
    class TestCase(unittest.TestCase):
    
        @mock.patch('yourfile.datetime')
        def test_dt(self, mock_dt):
            mock_dt.utcnow = mock.Mock(return_value=datetime(1901, 12, 21))
            r = get_report_month_key()
            self.assertEqual('190112', r)
    

提交回复
热议问题