Revision 77b657a3 test/ganeti.hooks_unittest.py

b/test/ganeti.hooks_unittest.py
36 36
from ganeti import cmdlib
37 37
from ganeti.constants import HKR_SUCCESS, HKR_FAIL, HKR_SKIP
38 38

  
39
from mocks import FakeConfig, FakeSStore, FakeProc
39
from mocks import FakeConfig, FakeSStore, FakeProc, FakeContext
40 40

  
41 41
class FakeLU(cmdlib.LogicalUnit):
42 42
  HPATH = "test"
......
226 226
                 for node_name in node_list])
227 227

  
228 228
  def setUp(self):
229
    self.cfg = FakeConfig()
230 229
    self.sstore = FakeSStore()
231 230
    self.op = opcodes.OpCode()
232
    self.lu = FakeLU(None, self.op, self.cfg, self.sstore)
231
    self.context = FakeContext()
232
    self.lu = FakeLU(None, self.op, self.context, self.sstore)
233 233

  
234 234
  def testTotalFalse(self):
235 235
    """Test complete rpc failure"""

Also available in: Unified diff