Add support for specifying port with EOS (#160)

pyntc/devices/eos_device.py

@@ -41,11 +41,23 @@ class EOSDevice(BaseDevice):
 
     vendor = "arista"
 
-    def __init__(self, host, username, password, transport="http", timeout=60, **kwargs):
+    def __init__(self, host, username, password, transport="http", port=None, timeout=None, **kwargs):
         super().__init__(host, username, password, device_type="arista_eos_eapi")
         self.transport = transport
+        self.port = port
         self.timeout = timeout
-        self.connection = eos_connect(transport, host=host, username=username, password=password, timeout=timeout)
+        eapi_args = {
+            "transport": transport,
+            "host": host,
+            "username": username,
+            "password": password,
+        }
+        optional_args = ("port", "timeout")
+        for arg in optional_args:
+            value = getattr(self, arg)
+            if value is not None:
+                eapi_args[arg] = value
+        self.connection = eos_connect(**eapi_args)
         self.native = EOSNative(self.connection)
         # _connected indicates Netmiko ssh connection
         self._connected = False

test/unit/test_devices/test_eos_device.py

@@ -318,3 +318,39 @@ def test_starting_config(self):
 
 if __name__ == "__main__":
     unittest.main()
+
+
+@mock.patch("pyntc.devices.eos_device.eos_connect")
+def test_init_no_transport(mock_eos_connect):
+    EOSDevice("host", "username", "password")
+    mock_eos_connect.assert_called_with(host="host", username="username", password="password", transport="http")
+
+
+@mock.patch("pyntc.devices.eos_device.eos_connect")
+def test_init_https_transport(mock_eos_connect):
+    EOSDevice("host", "username", "password", transport="https")
+    mock_eos_connect.assert_called_with(host="host", username="username", password="password", transport="https")
+
+
+@mock.patch("pyntc.devices.eos_device.eos_connect")
+def test_init_pass_port(mock_eos_connect):
+    EOSDevice("host", "username", "password", port=8080)
+    mock_eos_connect.assert_called_with(
+        host="host", username="username", password="password", transport="http", port=8080
+    )
+
+
+@mock.patch("pyntc.devices.eos_device.eos_connect")
+def test_init_pass_timeout(mock_eos_connect):
+    EOSDevice("host", "username", "password", timeout=30)
+    mock_eos_connect.assert_called_with(
+        host="host", username="username", password="password", transport="http", timeout=30
+    )
+
+
+@mock.patch("pyntc.devices.eos_device.eos_connect")
+def test_init_pass_port_and_timeout(mock_eos_connect):
+    EOSDevice("host", "username", "password", port=8080, timeout=30)
+    mock_eos_connect.assert_called_with(
+        host="host", username="username", password="password", transport="http", port=8080, timeout=30
+    )