Add the 2.0-specific node flags to the design doc

[ganeti-local] / test / mocks.py

diff --git a/test/mocks.py b/test/mocks.py
index cfff9c2..249cd72 100644 (file)
--- a/test/mocks.py
+++ b/test/mocks.py
@@ -42,16 +42,9 @@ class FakeConfig:
     def GetNodeList(self):
         return ["a", "b", "c"]
 
-    def GetMaster(self):
-        return utils.HostInfo().name
-
     def GetHostKey(self):
         return FAKE_CLUSTER_KEY
 
-
-class FakeSStore:
-    """Fake simplestore object"""
-
     def GetClusterName(self):
         return "test.cluster"
 
@@ -67,3 +60,11 @@ class FakeProc:
 
     def LogInfo(self, msg):
         pass
+
+class FakeContext:
+    """Fake context object"""
+
+    def __init__(self):
+        self.cfg = FakeConfig()
+        # TODO: decide what features a mock Ganeti Lock Manager must have
+        self.GLM = None