21 lines
628 B
Python
21 lines
628 B
Python
import arrow
|
|
|
|
from freezegun import freeze_time
|
|
|
|
|
|
def test_convert_validity_years(session):
|
|
from lemur.common.missing import convert_validity_years
|
|
|
|
with freeze_time("2016-01-01"):
|
|
data = convert_validity_years(dict(validity_years=2))
|
|
|
|
assert data["validity_start"] == arrow.utcnow().isoformat()
|
|
assert data["validity_end"] == arrow.utcnow().shift(years=+2).isoformat()
|
|
|
|
with freeze_time("2015-01-10"):
|
|
data = convert_validity_years(dict(validity_years=1))
|
|
assert (
|
|
data["validity_end"]
|
|
== arrow.utcnow().shift(years=+1, days=-2).isoformat()
|
|
)
|