Skip to content

Commit e2ee0c5

Browse files
committed
Add ability to omit start date during contract initialization
1 parent fced377 commit e2ee0c5

File tree

2 files changed

+12
-1
lines changed

2 files changed

+12
-1
lines changed

pcontract/data.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,7 @@ def __getitem__(self, item: int) -> Branch:
111111
def init(
112112
cls,
113113
*,
114-
start_at: datetime | None,
114+
start_at: datetime | None = None,
115115
end_at: datetime,
116116
data: dict[str, Any],
117117
meta: dict[str, Any] | None = None,

tests.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -241,3 +241,14 @@ def test_contract_init_spans_nothing(self):
241241
end_at=self.start - datetime.timedelta(days=1),
242242
data={"key": "venus"},
243243
)
244+
245+
def test_contract_init_without_start_date(self):
246+
delta = datetime.timedelta(hours=1)
247+
contract = Contract.init(data={}, end_at=datetime.datetime.now(tz=utc) + delta)
248+
(branch,) = contract.items
249+
# get away from adding freezegun
250+
self.assertAlmostEqual(
251+
delta.total_seconds(),
252+
branch.span.total_seconds(),
253+
places=5,
254+
)

0 commit comments

Comments
 (0)