Run black formatting on tests

Author: remingtonc

hygiene.sh

@@ -1,7 +1,7 @@
 #!/usr/bin/env bash
 # Script which autoformats (black) and lints (pylint) code
 echo "Running black ..."
-FORMAT_COMMAND="black --safe --verbose --exclude proto src/cisco_gnmi"
+FORMAT_COMMAND="black --safe --verbose --exclude proto src/cisco_gnmi tests/"
 if black &> /dev/null; then
     eval $FORMAT_COMMAND
 elif pipenv run black &> /dev/null; then

tests/test_auth.py

@@ -10,5 +10,7 @@ def test_call():
     mock_call = mock.MagicMock(spec=CiscoAuthPlugin.__call__)
 
     instance = CiscoAuthPlugin(username, password)
-    result = instance.__call__([(username, "testUsr"), (password, "testPass")], CiscoAuthPlugin)
+    result = instance.__call__(
+        [(username, "testUsr"), (password, "testPass")], CiscoAuthPlugin
+    )
     mock_call.assert_not_called()

tests/test_util.py

@@ -9,92 +9,103 @@ def test_gen_target_netloc_valueerror():
     with pytest.raises(ValueError):
         util.gen_target_netloc("http://www.test.com", "test_prefix://", 8000)
 
+
 def test_gen_target_netloc_parsed_none():
 
-    mock_netloc = '//www.testing.com:8080'
+    mock_netloc = "//www.testing.com:8080"
     mock_parsed_netloc = urlparse(mock_netloc)
 
     result = util.gen_target_netloc("www.testing.com", "http://", 8080)
     assert mock_parsed_netloc == result
 
+
 def test_validate_proto_enum_exception_one():
 
     enum = gnmi_pb2.SubscriptionMode
 
     with pytest.raises(Exception):
-        util.validate_proto_enum('test', 'INVALID_VALUE', 'test', enum)
+        util.validate_proto_enum("test", "INVALID_VALUE", "test", enum)
+
 
 def test_validate_proto_enum_exception_two():
 
     enum = gnmi_pb2.SubscriptionMode
     fake_subset = [3]
 
     with pytest.raises(Exception):
-        util.validate_proto_enum('test', 2, 'test', enum, subset=fake_subset)
+        util.validate_proto_enum("test", 2, "test", enum, subset=fake_subset)
+
 
 def test_validate_proto_enum_exception_three():
-    
+
     enum = gnmi_pb2.SubscriptionMode
     fake_subset = ["ON_CHANGE", "SAMPLE"]
 
     with pytest.raises(Exception):
-        util.validate_proto_enum('test', 'TARGET_DEFINED', 'test', enum, subset=fake_subset)
+        util.validate_proto_enum(
+            "test", "TARGET_DEFINED", "test", enum, subset=fake_subset
+        )
+
 
 def test_validate_proto_enum_element_in_subset_one():
 
     enum = gnmi_pb2.SubscriptionMode
     fake_subset = ["ON_CHANGE", "SAMPLE"]
 
-    result = util.validate_proto_enum('test', 2, 'test', enum, subset=fake_subset)
+    result = util.validate_proto_enum("test", 2, "test", enum, subset=fake_subset)
     assert 2 == result
 
+
 def test_validate_proto_enum_element_in_subset_two():
-    
+
     enum = gnmi_pb2.SubscriptionMode
     fake_subset = [2, 0]
 
-    result = util.validate_proto_enum('test', 'TARGET_DEFINED', 'test', enum, subset=fake_subset)
+    result = util.validate_proto_enum(
+        "test", "TARGET_DEFINED", "test", enum, subset=fake_subset
+    )
     assert 0 == result
 
+
 def test_validate_proto_enum_value_returned_one():
 
     enum = gnmi_pb2.SubscriptionMode
 
-    result = util.validate_proto_enum('test', 'ON_CHANGE', 'test', enum)
+    result = util.validate_proto_enum("test", "ON_CHANGE", "test", enum)
     assert 1 == result
 
+
 def test_validate_proto_enum_value_returned_two():
-    
+
     enum = gnmi_pb2.SubscriptionMode
 
-    result = util.validate_proto_enum('test', 1, 'test', enum)
+    result = util.validate_proto_enum("test", 1, "test", enum)
     assert 1 == result
 
+
 def test_get_cert_from_target():
 
-    target_netloc = {
-        "hostname": "cisco.com",
-        "port": 443
-    }
-    
-    expected_ssl_cert = ssl.get_server_certificate((
-        target_netloc.get("hostname"),
-        target_netloc.get("port")
-    ))
-    
-    expected_ssl_cert.encode('utf-8')
-      
+    target_netloc = {"hostname": "cisco.com", "port": 443}
+
+    expected_ssl_cert = ssl.get_server_certificate(
+        (target_netloc.get("hostname"), target_netloc.get("port"))
+    )
+
+    expected_ssl_cert.encode("utf-8")
+
     target = util.gen_target_netloc("cisco.com:443")
-    result = util.get_cert_from_target((target)).decode('utf-8')
+    result = util.get_cert_from_target((target)).decode("utf-8")
 
     assert expected_ssl_cert == result
 
+
 def test_get_cn_from_cert_returned_value_invalid_entry(mocker):
 
-    mock_cert_parsed = mocker.patch.object(x509, 'load_pem_x509_certificate')
-    result = util.get_cn_from_cert('INVALID_ENTRY')
+    mock_cert_parsed = mocker.patch.object(x509, "load_pem_x509_certificate")
+    result = util.get_cn_from_cert("INVALID_ENTRY")
 
     assert None == result
 
+
 def test_get_cn_from_cert_returned_value(mocker):
     pass