Restyled by black

Author: restyled-commits

minos/api_gateway/discovery/database/client.py

@@ -14,19 +14,12 @@ class OrdersMinosApiRouter(MinosApiRouter):
 
 import json
 import logging
-from typing import (
-    Any,
-    Optional
-)
+from typing import Any, Optional
 
 import aioredis
 
-from minos.api_gateway.common import (
-    MinosConfig,
-)
-from minos.api_gateway.discovery.domain.microservice import (
-    MICROSERVICE_KEY_PREFIX,
-)
+from minos.api_gateway.common import MinosConfig
+from minos.api_gateway.discovery.domain.microservice import MICROSERVICE_KEY_PREFIX
 
 log = logging.getLogger(__name__)
 

tests/test_api_gateway/test_discovery/dataset.py

@@ -11,21 +11,27 @@ def generate_random_microservice_names(quantity: int):
 
 
 def generate_record(name):
-    ip = socket.inet_ntoa(struct.pack('>I', random.randint(1, 0xffffffff)))
+    ip = socket.inet_ntoa(struct.pack(">I", random.randint(1, 0xFFFFFFFF)))
     port = random.randint(1, 9999)
 
-    record = {"address": f"{ip}", "port": port,
-              "endpoints": [["GET", f"test_endpoint_{name}"], ["POST", f"test_endpoint_{name}"]]}
+    record = {
+        "address": f"{ip}",
+        "port": port,
+        "endpoints": [["GET", f"test_endpoint_{name}"], ["POST", f"test_endpoint_{name}"]],
+    }
 
     return record
 
 
 def generate_record_old(x):
-    ip = socket.inet_ntoa(struct.pack('>I', random.randint(1, 0xffffffff)))
+    ip = socket.inet_ntoa(struct.pack(">I", random.randint(1, 0xFFFFFFFF)))
     port = random.randint(1, 9999)
     name = f"microservice_{x}"
 
-    record = {"address": f"{ip}", "port": port,
-              "endpoints": [["GET", f"test_endpoint_{name}"], ["POST", f"test_endpoint_{name}"]]}
+    record = {
+        "address": f"{ip}",
+        "port": port,
+        "endpoints": [["GET", f"test_endpoint_{name}"], ["POST", f"test_endpoint_{name}"]],
+    }
 
     return name, record

tests/test_api_gateway/test_discovery/test_views/test_microservice.py

@@ -4,17 +4,14 @@
     AioHTTPTestCase,
     unittest_run_loop,
 )
-from minos.api_gateway.common import (
-    MinosConfig,
-)
-from minos.api_gateway.discovery import (
-    DiscoveryService,
-)
-from tests.test_api_gateway.test_discovery.dataset import generate_record, generate_random_microservice_names, \
-    generate_record_old
-from tests.utils import (
-    BASE_PATH,
+from minos.api_gateway.common import MinosConfig
+from minos.api_gateway.discovery import DiscoveryService
+from tests.test_api_gateway.test_discovery.dataset import (
+    generate_record,
+    generate_random_microservice_names,
+    generate_record_old,
 )
+from tests.utils import BASE_PATH
 
 
 class TestMicroserviceEndpoints(AioHTTPTestCase):
@@ -70,15 +67,16 @@ async def test_bulk_update(self):
         # Check updated records are correct
         for record in expected:
             response = await self.client.get(
-                f"/microservices?verb={record['body']['endpoints'][0][0]}&path={record['body']['endpoints'][0][1]}")
+                f"/microservices?verb={record['body']['endpoints'][0][0]}&path={record['body']['endpoints'][0][1]}"
+            )
 
             self.assertEqual(200, response.status)
 
             body = await response.json()
 
-            self.assertEqual(record['body']['address'], body["address"])
-            self.assertEqual(record['body']['port'], int(body["port"]))
-            self.assertEqual(record['name'], body["name"])
+            self.assertEqual(record["body"]["address"], body["address"])
+            self.assertEqual(record["body"]["port"], int(body["port"]))
+            self.assertEqual(record["name"], body["name"])
 
     async def test_bulk_update_2(self):
         expected = list()
@@ -106,15 +104,16 @@ async def test_bulk_update_2(self):
 
         for record in expected:
             response = await self.client.get(
-                f"/microservices?verb={record['body']['endpoints'][0][0]}&path={record['body']['endpoints'][0][1]}")
+                f"/microservices?verb={record['body']['endpoints'][0][0]}&path={record['body']['endpoints'][0][1]}"
+            )
 
             self.assertEqual(200, response.status)
 
             body = await response.json()
 
-            self.assertEqual(record['body']['address'], body["address"])
-            self.assertEqual(int(record['body']['port']), int(body["port"]))
-            self.assertEqual(record['name'], body["name"])
+            self.assertEqual(record["body"]["address"], body["address"])
+            self.assertEqual(int(record["body"]["port"]), int(body["port"]))
+            self.assertEqual(record["name"], body["name"])
 
     @unittest_run_loop
     async def test_post_missing_param(self):