Skip to content

Commit 4751bab

Browse files
Fix integration tests
The duration type has not been updated in delivery periods for integration tests, causing type errors.
1 parent 15d87b3 commit 4751bab

File tree

2 files changed

+11
-15
lines changed

2 files changed

+11
-15
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: 10 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
Client,
1616
Currency,
1717
DeliveryArea,
18+
DeliveryDuration,
1819
DeliveryPeriod,
1920
EnergyMarketCodeType,
2021
MarketSide,
@@ -57,7 +58,7 @@ def __init__(self) -> None:
5758
)
5859
self.delivery_period = DeliveryPeriod(
5960
start=self.delivery_start,
60-
duration=timedelta(minutes=15),
61+
duration=DeliveryDuration.from_timedelta(timedelta(minutes=15)),
6162
)
6263
self.price = Price(amount=Decimal("56"), currency=Currency.EUR)
6364
self.quantity = Power(mw=Decimal("0.1"))
@@ -134,7 +135,7 @@ async def create_test_trade(
134135
)
135136
delivery_period = DeliveryPeriod(
136137
start=delivery_start,
137-
duration=timedelta(minutes=15),
138+
duration=DeliveryDuration.from_timedelta(timedelta(minutes=15)),
138139
)
139140
buy_order = await create_test_order(
140141
trader=trader,
@@ -235,9 +236,7 @@ async def test_list_gridpool_orders(trader: _TestTrader) -> None:
235236
# filter by delivery period to avoid fetching too many orders
236237
orders = [
237238
order
238-
async for order in trader.client.list_gridpool_orders(
239-
gridpool_id=GRIDPOOL_ID, delivery_period=trader.delivery_period
240-
)
239+
async for order in trader.client.list_gridpool_orders(gridpool_id=GRIDPOOL_ID)
241240
]
242241
listed_orders_id = [order.order_id for order in orders]
243242
for order_id in created_orders_id:
@@ -350,14 +349,8 @@ async def test_cancel_all_orders(trader: _TestTrader) -> None:
350349

351350
async def test_list_gridpool_trades(trader: _TestTrader) -> None:
352351
"""Test listing gridpool trades."""
353-
buy_order, sell_order = await create_test_trade(trader)
354-
trades = [
355-
trade
356-
async for trade in trader.client.list_gridpool_trades(
357-
GRIDPOOL_ID,
358-
delivery_period=buy_order.order.delivery_period,
359-
)
360-
]
352+
await create_test_trade(trader)
353+
trades = [trade async for trade in trader.client.list_gridpool_trades(GRIDPOOL_ID)]
361354
assert len(trades) >= 1
362355

363356

@@ -397,7 +390,10 @@ async def test_receive_public_trades_filter(trader: _TestTrader) -> None:
397390
start_time = datetime.now(timezone.utc).replace(second=0, microsecond=0)
398391
start_time += timedelta(minutes=30 - start_time.minute % 15) # next 15-minute mark
399392

400-
delivery_period = DeliveryPeriod(start=start_time, duration=timedelta(minutes=15))
393+
delivery_period = DeliveryPeriod(
394+
start=start_time,
395+
duration=DeliveryDuration.from_timedelta(timedelta(minutes=15)),
396+
)
401397

402398
price = Price(amount=Decimal("808"), currency=Currency.EUR)
403399

0 commit comments

Comments
 (0)