import/export: Validate remote host/port

[ganeti-local] / test / mocks.py

diff --git a/test/mocks.py b/test/mocks.py
index bea8dd3..4c0ae07 100644 (file)
--- a/test/mocks.py
+++ b/test/mocks.py
@@ -42,9 +42,6 @@ class FakeConfig:
     def GetNodeList(self):
         return ["a", "b", "c"]
 
-    def GetMaster(self):
-        return utils.HostInfo().name
-
     def GetHostKey(self):
         return FAKE_CLUSTER_KEY
 
@@ -58,12 +55,16 @@ class FakeConfig:
 class FakeProc:
     """Fake processor object"""
 
-    def LogWarning(self, msg):
+    def LogWarning(self, msg, *args, **kwargs):
+        pass
+
+    def LogInfo(self, msg, *args, **kwargs):
         pass
 
-    def LogInfo(self, msg):
+    def LogStep(self, current, total, message):
         pass
 
+
 class FakeContext:
     """Fake context object"""
 
@@ -71,4 +72,3 @@ class FakeContext:
         self.cfg = FakeConfig()
         # TODO: decide what features a mock Ganeti Lock Manager must have
         self.GLM = None
-