fix device transport loading from inventory (#714)

* fix device transport loading

Signed-off-by: LucaNicosia <[email protected]>

* device transport fix pt2

Signed-off-by: LucaNicosia <[email protected]>

* fix base source typo

Signed-off-by: LucaNicosia <[email protected]>

Author: LucaNicosia

suzieq/poller/controller/source/base_source.py

@@ -218,6 +218,9 @@ def set_device(self, inventory: Dict[str, Dict]):
                                            f" at {jump_host_key_file} doesn't"
                                            " exists")
             transport = self._device.get('transport')
+            if transport:
+                # get the string from the enum
+                transport = transport.value
             ignore_known_hosts = self._device.get('ignore-known-hosts', False)
             slow_host = self._device.get('slow-host', False)
             port = self._device.get('port')
@@ -244,7 +247,7 @@ def _validate_device(self):
             inv_fields = [x for x in self._device if x not in dev_fields]
             if inv_fields:
                 raise InventorySourceError(
-                    f'{self._device.get("name")}: Unknow fields called '
+                    f'{self._device.get("name")}: Unknown fields called '
                     f'{inv_fields}')
 
 

tests/unit/poller/controller/sources/test_devices.py

@@ -5,8 +5,8 @@
 
 import pytest
 from suzieq.poller.controller.source.base_source import Source
-from suzieq.shared.exceptions import InventorySourceError
 from suzieq.poller.controller.utils.inventory_utils import DeviceModel
+from suzieq.shared.utils import PollerTransport
 
 # pylint: disable=protected-access
 
@@ -66,7 +66,7 @@ async def test_devices_set(inventory: Dict):
             'jump-host': None,
             'jump-host-key-file': None,
             'devtype': 'panos',
-            'transport': 'ssh',
+            'transport': PollerTransport.ssh,
 
         }
     }