4138: Fix noop Node Manager EC2 driver tests.

[arvados.git] / services / nodemanager / tests / testutil.py

diff --git a/services/nodemanager/tests/testutil.py b/services/nodemanager/tests/testutil.py
index ff525f0177ea7fc7be7d5d22a7e02bf0c3e28e06..c5b6539ace77d45152493ed14df3c6b4acf9cad2 100644 (file)
--- a/services/nodemanager/tests/testutil.py
+++ b/services/nodemanager/tests/testutil.py
@@ -30,6 +30,15 @@ def arvados_node_mock(node_num=99, job_uuid=None, age=0, **kwargs):
     node.update(kwargs)
     return node
 
+def cloud_object_mock(name_id):
+    # A very generic mock, useful for stubbing libcloud objects we
+    # only search for and pass around, like locations, subnets, etc.
+    cloud_object = mock.NonCallableMagicMock(['id', 'name'],
+                                             name='cloud_object')
+    cloud_object.id = str(name_id)
+    cloud_object.name = cloud_object.id.upper()
+    return cloud_object
+
 def cloud_node_mock(node_num=99):
     node = mock.NonCallableMagicMock(
         ['id', 'name', 'state', 'public_ips', 'private_ips', 'driver', 'size',