Use black on the tests folder

Author: ncrocfer

tests/conftest.py

@@ -17,19 +17,19 @@ def create_ports(tmpdir, monkeypatch):
     def _create_ports(ports):
         def get_db():
             tmp_db = tmpdir.join("db.json")
-            db = TinyDB(
-                str(tmp_db),
-                storage=CachingMiddleware(JSONStorage)
-            )
+            db = TinyDB(str(tmp_db), storage=CachingMiddleware(JSONStorage))
 
             for port in ports:
-                db.insert({
-                    'name': port[0],
-                    'port': port[1],
-                    "description": port[2],
-                    "protocol": port[3]
-                })
+                db.insert(
+                    {
+                        "name": port[0],
+                        "port": port[1],
+                        "description": port[2],
+                        "protocol": port[3],
+                    }
+                )
             return db
 
         return monkeypatch.setattr(whatportis.db, "get_database", get_db)
-    return  _create_ports
+
+    return _create_ports

tests/test_cli.py

@@ -6,21 +6,21 @@
 
 def test_search_with_no_arg(runner):
     result = runner.invoke(run, [])
-    regex = re.compile('^Usage:')
+    regex = re.compile("^Usage:")
     assert result.exit_code == 2
     assert regex.match(result.output)
 
 
 def test_search_with_multiple_args(runner):
-    result = runner.invoke(run, ['foo', 'bar'])
-    regex = re.compile('^Usage:')
+    result = runner.invoke(run, ["foo", "bar"])
+    regex = re.compile("^Usage:")
     assert result.exit_code == 2
     assert regex.match(result.output)
 
 
 def test_search_str_port(runner, create_ports):
     create_ports([["foo", "1234", "My description", "tcp"]])
-    result = runner.invoke(run, ['foo'])
+    result = runner.invoke(run, ["foo"])
     assert result.exit_code == 0
     assert "foo" in result.output
     assert "1234" in result.output
@@ -30,7 +30,7 @@ def test_search_str_port(runner, create_ports):
 
 def test_search_int_port(runner, create_ports):
     create_ports([["foo", "1234", "My description", "tcp"]])
-    result = runner.invoke(run, ['1234'])
+    result = runner.invoke(run, ["1234"])
     assert result.exit_code == 0
     assert "foo" in result.output
     assert "1234" in result.output
@@ -39,42 +39,50 @@ def test_search_int_port(runner, create_ports):
 
 
 def test_search_str_like_port(runner, create_ports):
-    create_ports([
-        ["foo", "1", "My description", "tcp"],
-        ["foobar", "2", "My description", "udp"],
-        ["baz", "3", "My description", "tcp"]
-    ])
-
-    result = runner.invoke(run, ['foo', '--like'])
+    create_ports(
+        [
+            ["foo", "1", "My description", "tcp"],
+            ["foobar", "2", "My description", "udp"],
+            ["baz", "3", "My description", "tcp"],
+        ]
+    )
+
+    result = runner.invoke(run, ["foo", "--like"])
     assert result.exit_code == 0
-    assert result.output.count('foo') == 2
+    assert result.output.count("foo") == 2
 
-    result = runner.invoke(run, ['bar', '--like'])
+    result = runner.invoke(run, ["bar", "--like"])
     assert result.exit_code == 0
-    assert result.output.count('bar') == 1
+    assert result.output.count("bar") == 1
 
 
 def test_search_int_like_port(runner, create_ports):
-    create_ports([
-        ["foo", "8", "My description", "tcp"],
-        ["foobar", "9", "My description", "udp"],
-        ["baz", "80", "My description", "tcp"]
-    ])
-
-    result = runner.invoke(run, ['8', '--like'])
+    create_ports(
+        [
+            ["foo", "8", "My description", "tcp"],
+            ["foobar", "9", "My description", "udp"],
+            ["baz", "80", "My description", "tcp"],
+        ]
+    )
+
+    result = runner.invoke(run, ["8", "--like"])
     assert result.exit_code == 0
-    assert result.output.count('8') == 2
+    assert result.output.count("8") == 2
 
 
 def test_search_port_returns_json(runner, create_ports):
     create_ports([["foo", "1234", "My description", "tcp"]])
-    result = runner.invoke(run, ['--json', 'foo'])
+    result = runner.invoke(run, ["--json", "foo"])
 
     result = json.loads(result.output)
-    wanted = [{
-        'description': 'My description',
-        'name': 'foo',
-        'port': '1234',
-        'protocol': 'tcp'
-    }]
-    assert sorted(result, key=lambda p: p["name"]) == sorted(wanted, key=lambda p: p["name"])
+    wanted = [
+        {
+            "description": "My description",
+            "name": "foo",
+            "port": "1234",
+            "protocol": "tcp",
+        }
+    ]
+    assert sorted(result, key=lambda p: p["name"]) == sorted(
+        wanted, key=lambda p: p["name"]
+    )

tests/test_utils.py

@@ -2,37 +2,47 @@
 
 
 def test_merge_protocols_different_ports():
-    ports = [{
-        "description": "My description 1",
-        "name": "MyName 1",
-        "port": "1234",
-        "protocol": "udp"
-    }, {
-        "description": "My description 2",
-        "name": "MyName 2",
-        "port": "5678",
-        "protocol": "tcp"
-    }]
+    ports = [
+        {
+            "description": "My description 1",
+            "name": "MyName 1",
+            "port": "1234",
+            "protocol": "udp",
+        },
+        {
+            "description": "My description 2",
+            "name": "MyName 2",
+            "port": "5678",
+            "protocol": "tcp",
+        },
+    ]
     result = merge_protocols(ports)
-    assert sorted(result, key=lambda p: p["name"]) == sorted(ports, key=lambda p: p["name"])
+    assert sorted(result, key=lambda p: p["name"]) == sorted(
+        ports, key=lambda p: p["name"]
+    )
 
 
 def test_merge_protocols_same_port():
-    ports = [{
-        "description": "My description",
-        "name": "MyName",
-        "port": "1234",
-        "protocol": "udp"
-    }, {
-        "description": "My description",
-        "name": "MyName",
-        "port": "1234",
-        "protocol": "tcp"
-    }]
+    ports = [
+        {
+            "description": "My description",
+            "name": "MyName",
+            "port": "1234",
+            "protocol": "udp",
+        },
+        {
+            "description": "My description",
+            "name": "MyName",
+            "port": "1234",
+            "protocol": "tcp",
+        },
+    ]
     result = merge_protocols(ports)
-    assert result == [{
-        "description": "My description",
-        "name": "MyName",
-        "port": "1234",
-        "protocol": "udp, tcp"
-    }]
+    assert result == [
+        {
+            "description": "My description",
+            "name": "MyName",
+            "port": "1234",
+            "protocol": "udp, tcp",
+        }
+    ]