|
@@ 171-192 (lines=22) @@
|
| 168 |
|
assert station_reading.td > 0 |
| 169 |
|
|
| 170 |
|
|
| 171 |
|
async def test_get_station_data_by_date_range_with_channel(self) -> None: |
| 172 |
|
"""Test get_station_data_by_date_range endpoint with channel.""" |
| 173 |
|
today = self.tz.localize(datetime.now()) |
| 174 |
|
today = today.replace(hour=0, minute=0, second=0, microsecond=0) |
| 175 |
|
yesterday = today - timedelta(days=1) |
| 176 |
|
# `hour=1` for DST fix cases |
| 177 |
|
today = today.replace(hour=2, minute=0, second=0, microsecond=0) |
| 178 |
|
station_data = await self.ims.get_station_data_by_date_range( |
| 179 |
|
self.station_id, |
| 180 |
|
from_date=yesterday, |
| 181 |
|
to_date=today, |
| 182 |
|
channel_id=self.channel_id, |
| 183 |
|
) |
| 184 |
|
|
| 185 |
|
assert station_data is not None |
| 186 |
|
assert station_data.station_id == self.station_id |
| 187 |
|
assert station_data.data is not None |
| 188 |
|
assert len(station_data.data) > 0 |
| 189 |
|
for station_reading in station_data.data: |
| 190 |
|
assert station_reading.datetime >= to_date_time(yesterday) |
| 191 |
|
assert station_reading.datetime < today |
| 192 |
|
assert station_reading.td > 0 |
| 193 |
|
|
| 194 |
|
|
| 195 |
|
async def test_get_monthly_station_data(self) -> None: |
|
@@ 150-168 (lines=19) @@
|
| 147 |
|
assert station_reading.datetime.date() == self.tz.localize(datetime.now()).date() |
| 148 |
|
|
| 149 |
|
|
| 150 |
|
async def test_get_station_data_by_date_range(self) -> None: |
| 151 |
|
"""Test get_station_data_by_date_range endpoint.""" |
| 152 |
|
today = self.tz.localize(datetime.now()) |
| 153 |
|
today = today.replace(hour=0, minute=0, second=0, microsecond=0) |
| 154 |
|
yesterday = today - timedelta(days=1) |
| 155 |
|
# `hour=1` for DST fix cases |
| 156 |
|
today = today.replace(hour=2, minute=0, second=0, microsecond=0) |
| 157 |
|
station_data = await self.ims.get_station_data_by_date_range( |
| 158 |
|
self.station_id, from_date=yesterday, to_date=today |
| 159 |
|
) |
| 160 |
|
|
| 161 |
|
assert station_data is not None |
| 162 |
|
assert station_data.station_id == self.station_id |
| 163 |
|
assert station_data.data is not None |
| 164 |
|
assert len(station_data.data) > 0 |
| 165 |
|
for station_reading in station_data.data: |
| 166 |
|
assert station_reading.datetime >= to_date_time(yesterday) |
| 167 |
|
assert station_reading.datetime < today |
| 168 |
|
assert station_reading.td > 0 |
| 169 |
|
|
| 170 |
|
|
| 171 |
|
async def test_get_station_data_by_date_range_with_channel(self) -> None: |
GuyKh /
py-ims-envista
