Fix unit tests

diff --git a/unittests/test_lib/test_xenserver_sr.py b/unittests/test_lib/test_xenserver_sr.py
index 6380c5db07538aabad202694c0e71aba1f3ff8af..1aea88e0165487c60352d9a0ed9e0fe5493b0253 100644 (file)
--- a/unittests/test_lib/test_xenserver_sr.py
+++ b/unittests/test_lib/test_xenserver_sr.py
@@ -7,6 +7,7 @@ class TestNFSStorageRepository(XenRTUnitTestCase):
     def testCreateStoresDeviceConfiguration(self):
         host = Mock()
         xenrt.TEC = Mock()
+        xenrt.GEC = Mock()
         sr = xenrt.lib.xenserver.NFSStorageRepository(host, 'sr-name')
 
         sr.create('guest-IP', '/nfs-export')
@@ -24,6 +25,7 @@ class TestNFSv4StorageRepository(XenRTUnitTestCase):
     def testCreateUsesVersion4AsAParameter(self):
         host = Mock()
         xenrt.TEC = Mock()
+        xenrt.GEC = Mock()
         sr = xenrt.lib.xenserver.NFSv4StorageRepository(host, 'sr-name')
 
         sr.create('guest-IP', '/nfs-export')