Add custom clock fixture

stix2.1
Greg Back 2017-01-18 16:14:57 -08:00
parent 4d9dcafbc6
commit 439211082a
1 changed files with 13 additions and 0 deletions

View File

@ -9,6 +9,19 @@ import stix2
amsterdam = pytz.timezone('Europe/Amsterdam') amsterdam = pytz.timezone('Europe/Amsterdam')
eastern = pytz.timezone('US/Eastern') eastern = pytz.timezone('US/Eastern')
FAKE_TIME = datetime.datetime(2017, 01, 01, 12, 34, 56)
# Inspired by: http://stackoverflow.com/a/24006251
@pytest.fixture
def clock(monkeypatch):
class mydatetime(datetime.datetime):
@classmethod
def now(cls):
return FAKE_TIME
monkeypatch.setattr(datetime, 'datetime', mydatetime)
@pytest.mark.parametrize('dt,timestamp', [ @pytest.mark.parametrize('dt,timestamp', [