Fix LUGrowDisk

[ganeti-local] / test / mocks.py

diff --git a/test/mocks.py b/test/mocks.py
index 4f569a6..442709c 100644 (file)
--- a/test/mocks.py
+++ b/test/mocks.py
@@ -1,4 +1,4 @@
-#!/usr/bin/python
+#
 #
 
 # Copyright (C) 2006, 2007 Google Inc.
@@ -21,9 +21,18 @@
 
 """Module implementing a fake ConfigWriter"""
 
-import socket
 from ganeti import utils
 
+
+FAKE_CLUSTER_KEY = ("AAAAB3NzaC1yc2EAAAABIwAAAQEAsuGLw70et3eApJ/ZEJkAVZogIrm"
+                    "EYPQJvb1ll52Ti0nr80Wztxibaa8bYGzY22rQIAloIlePeTGcJceAYK"
+                    "PZgm0I/Mp2EUGg2NVsQZIzasz6cW0vYuiUbF9GkVlROmvOAykT58RfM"
+                    "L8RhPrjrQxZc+NXgZtgDugYSZcXHDLUyWM1xKUoYy0MqYG6ZXCC/Zno"
+                    "RThhmjOJgEmvwrMcTWQjmzH3NeJAxaBsEHR8tiVZ/Y23C/ULWLyNT6R"
+                    "fB+DE7IovsMQaS+83AK1Teg7RWNyQczachatf/JT8VjUqFYjJepPjMb"
+                    "vYdB2nQds7/+Bf40C/OpbvnAxna1kVtgFHAo18cQ==")
+
+
 class FakeConfig:
     """Fake configuration object"""
 
@@ -36,6 +45,9 @@ class FakeConfig:
     def GetMaster(self):
         return utils.HostInfo().name
 
+    def GetHostKey(self):
+        return FAKE_CLUSTER_KEY
+
 
 class FakeSStore:
     """Fake simplestore object"""
@@ -45,3 +57,22 @@ class FakeSStore:
 
     def GetMasterNode(self):
         return utils.HostInfo().name
+
+
+class FakeProc:
+    """Fake processor object"""
+
+    def LogWarning(self, msg):
+        pass
+
+    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
+