fix: Security tests

ipfabric/client.py

@@ -3,12 +3,13 @@
 from typing import Optional, Union
 from urllib.parse import urlparse
 
+from pkg_resources import parse_version
+
 from ipfabric import models
 from ipfabric.api import IPFabricAPI
 from ipfabric.intent import Intent
 from ipfabric.pathlookup import Diagram
 from ipfabric.security import Security
-from pkg_resources import parse_version
 
 DEFAULT_ID = "$last"
 
@@ -43,7 +44,8 @@ def __init__(
         self.inventory = models.Inventory(client=self)
         self.graphs = Diagram(self) if parse_version(self.os_version) < parse_version("4.3") else \
             ImportError("v4.3 digrams has been moved to ipfabrc-diagrams python package")
-        self.security = Security(client=self)
+        self.security = Security(client=self) if parse_version(self.os_version) < parse_version("4.3") else \
+            ImportError("Security Policy tables have changed and need updated")
         self.intent = Intent(client=self)
 
     @check_format

tests/functional/test_security.py

@@ -1,6 +1,8 @@
 import os
 import unittest
 
+from pkg_resources import parse_version
+
 import ipfabric.security
 from ipfabric import IPFClient
 
@@ -11,16 +13,22 @@
 class MyTestCase(unittest.TestCase):
     def test_get_acl(self):
         ipf = IPFClient()
+        if parse_version(ipf.os_version) >= parse_version("4.3"):
+            self.skipTest('IP Fabric version under 4.3')
         acls = ipf.security.search_acl_policies()
         self.assertIsInstance(acls, list)
 
     def test_get_zone(self):
         ipf = IPFClient()
+        if parse_version(ipf.os_version) >= parse_version("4.3"):
+            self.skipTest('IP Fabric version under 4.3')
         zones = ipf.security.search_zone_policies()
         self.assertIsInstance(zones, list)
 
     def test_get_policy(self):
         ipf = IPFClient()
+        if parse_version(ipf.os_version) >= parse_version("4.3"):
+            self.skipTest('IP Fabric version under 4.3')
         acl = ipf.security.search_acl_policies()[0]
         policy = ipf.security.get_policy(acl)
         self.assertIsInstance(policy, ipfabric.security.Policy)