Skip to content

Commit 7c75f50

Browse files
Fix integration tests (#217)
The duration type has not been updated in delivery periods for integration tests, causing type errors.
2 parents 15d87b3 + f883cbf commit 7c75f50

File tree

2 files changed

+20
-7
lines changed

2 files changed

+20
-7
lines changed

RELEASE_NOTES.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,4 +14,4 @@
1414

1515
## Bug Fixes
1616

17-
<!-- Here goes notable bug fixes that are worth a special mention or explanation -->
17+
* Fixed duration type in delivery period for integration tests.

integration_tests/test_api.py

Lines changed: 19 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -15,8 +15,11 @@
1515
Client,
1616
Currency,
1717
DeliveryArea,
18+
DeliveryDuration,
1819
DeliveryPeriod,
20+
DeliveryTimeFilter,
1921
EnergyMarketCodeType,
22+
Interval,
2023
MarketSide,
2124
OrderDetail,
2225
OrderState,
@@ -57,7 +60,7 @@ def __init__(self) -> None:
5760
)
5861
self.delivery_period = DeliveryPeriod(
5962
start=self.delivery_start,
60-
duration=timedelta(minutes=15),
63+
duration=DeliveryDuration.from_timedelta(timedelta(minutes=15)),
6164
)
6265
self.price = Price(amount=Decimal("56"), currency=Currency.EUR)
6366
self.quantity = Power(mw=Decimal("0.1"))
@@ -134,7 +137,7 @@ async def create_test_trade(
134137
)
135138
delivery_period = DeliveryPeriod(
136139
start=delivery_start,
137-
duration=timedelta(minutes=15),
140+
duration=DeliveryDuration.from_timedelta(timedelta(minutes=15)),
138141
)
139142
buy_order = await create_test_order(
140143
trader=trader,
@@ -236,7 +239,11 @@ async def test_list_gridpool_orders(trader: _TestTrader) -> None:
236239
orders = [
237240
order
238241
async for order in trader.client.list_gridpool_orders(
239-
gridpool_id=GRIDPOOL_ID, delivery_period=trader.delivery_period
242+
gridpool_id=GRIDPOOL_ID,
243+
delivery_time_filter=DeliveryTimeFilter(
244+
time_interval=Interval(trader.delivery_period.start),
245+
duration_filters=[trader.delivery_period.duration],
246+
),
240247
)
241248
]
242249
listed_orders_id = [order.order_id for order in orders]
@@ -350,12 +357,15 @@ async def test_cancel_all_orders(trader: _TestTrader) -> None:
350357

351358
async def test_list_gridpool_trades(trader: _TestTrader) -> None:
352359
"""Test listing gridpool trades."""
353-
buy_order, sell_order = await create_test_trade(trader)
360+
await create_test_trade(trader)
354361
trades = [
355362
trade
356363
async for trade in trader.client.list_gridpool_trades(
357364
GRIDPOOL_ID,
358-
delivery_period=buy_order.order.delivery_period,
365+
delivery_time_filter=DeliveryTimeFilter(
366+
time_interval=Interval(trader.delivery_period.start),
367+
duration_filters=[trader.delivery_period.duration],
368+
),
359369
)
360370
]
361371
assert len(trades) >= 1
@@ -397,7 +407,10 @@ async def test_receive_public_trades_filter(trader: _TestTrader) -> None:
397407
start_time = datetime.now(timezone.utc).replace(second=0, microsecond=0)
398408
start_time += timedelta(minutes=30 - start_time.minute % 15) # next 15-minute mark
399409

400-
delivery_period = DeliveryPeriod(start=start_time, duration=timedelta(minutes=15))
410+
delivery_period = DeliveryPeriod(
411+
start=start_time,
412+
duration=DeliveryDuration.from_timedelta(timedelta(minutes=15)),
413+
)
401414

402415
price = Price(amount=Decimal("808"), currency=Currency.EUR)
403416

0 commit comments

Comments
 (0)