Fix up tests w/r/t recent changes

CCI-MOC · Jan 16, 2025 · ab92354 · ab92354
1 parent c3b75e8
commit ab92354
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/esiclient/tests/unit/v1/test_port_forwarding.py b/esiclient/tests/unit/v1/test_port_forwarding.py
@@ -1,3 +1,4 @@
+import argparse
 import testtools
 import ipaddress
 from unittest import mock
@@ -75,7 +76,7 @@ def test_AddressOrNetwork_network(self):
     def test_AddressOrNetwork_invalid(self):
         self.connection.network.find_network.return_value = None
         arg = AddressOrNetworkArg(self.cli)
-        self.assertRaises(ValueError, arg, "mynetwork")
+        self.assertRaises(argparse.ArgumentTypeError, arg, "mynetwork")
 
 
 class TestAddressOrPort(PortForwardTestCase):
@@ -93,7 +94,7 @@ def test_AddressOrPort_port(self):
     def test_AddressOrPort_invalid(self):
         self.connection.network.find_port.return_value = None
         arg = AddressOrPortArg(self.cli)
-        self.assertRaises(ValueError, arg, "myport")
+        self.assertRaises(argparse.ArgumentTypeError, arg, "myport")
 
 
 class TestNetworkArg(PortForwardTestCase):
@@ -106,7 +107,7 @@ def test_Network_valid(self):
     def test_Network_invalid(self):
         self.connection.network.find_network.return_value = None
         arg = NetworkArg(self.cli)
-        self.assertRaises(ValueError, arg, "mynetwork")
+        self.assertRaises(argparse.ArgumentTypeError, arg, "mynetwork")
 
 
 class TestSubnetArg(PortForwardTestCase):
@@ -119,7 +120,7 @@ def test_Subnet_valid(self):
     def test_Subnet_invalid(self):
         self.connection.network.find_subnet.return_value = None
         arg = SubnetArg(self.cli)
-        self.assertRaises(ValueError, arg, "mysubnet")
+        self.assertRaises(argparse.ArgumentTypeError, arg, "mysubnet")
 
 
 class TestNetworkOpsMixin(PortForwardTestCase):