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

后端 未结 5 1059
梦如初夏
梦如初夏 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:50

    You can try using freezetime module.

    from yourfile import get_report_month_key
    from freezegun import freeze_time
    import unittest
    
    class TestCase(unittest.TestCase):
    
        @freeze_time('2017-05-01')
        def get_report_month_key_test():
           get_report_month_key().should.equal('201705')
    

提交回复
热议问题