1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
"""Test the region names request"""
import asyncio
import unittest
from unittest.mock import AsyncMock
from mypermobil import MyPermobil, create_session, MyPermobilAPIException
class TestRegion(unittest.TestCase):
"""Test the region names request"""
def test_region(self):
"""Test the region names request"""
async def region_name_test():
"""Test the region names request"""
session = await create_session()
api = MyPermobil("test", session)
api.set_email("email.that.ends.with@permobil.com")
names = await api.request_region_names()
await api.close_session()
assert len(names) > 0
async def region_with_flags():
"""Test the region with flags request"""
session = await create_session()
api = MyPermobil("test", session)
regions = await api.request_regions(include_icons=True)
await api.close_session()
assert len(regions) > 0
async def region_error():
"""Test the region error request"""
session = AsyncMock()
response = AsyncMock()
response.status = 404
response.text = AsyncMock(return_value="test")
session.get = AsyncMock(return_value=response)
api = MyPermobil("test", session)
with self.assertRaises(MyPermobilAPIException):
await api.request_regions(include_icons=True)
asyncio.run(region_name_test())
asyncio.run(region_with_flags())
asyncio.run(region_error())
if __name__ == "__main__":
unittest.main()
|
