Revision c121d42f test/py/ganeti.client.gnt_cluster_unittest.py
| b/test/py/ganeti.client.gnt_cluster_unittest.py | ||
|---|---|---|
| 24 | 24 |
import unittest |
| 25 | 25 |
import optparse |
| 26 | 26 |
|
| 27 |
from ganeti import errors |
|
| 27 | 28 |
from ganeti.client import gnt_cluster |
| 28 | 29 |
from ganeti import utils |
| 29 | 30 |
from ganeti import compat |
| 30 | 31 |
from ganeti import constants |
| 31 | 32 |
|
| 33 |
import mock |
|
| 32 | 34 |
import testutils |
| 33 | 35 |
|
| 34 | 36 |
|
| ... | ... | |
| 258 | 260 |
self.assertEqual(result, constants.EXIT_FAILURE) |
| 259 | 261 |
|
| 260 | 262 |
|
| 263 |
class InitDrbdHelper(unittest.TestCase): |
|
| 264 |
|
|
| 265 |
def testNoDrbdNoHelper(self): |
|
| 266 |
opts = mock.Mock() |
|
| 267 |
opts.drbd_storage = False |
|
| 268 |
opts.drbd_helper = None |
|
| 269 |
helper = gnt_cluster._InitDrbdHelper(opts) |
|
| 270 |
self.assertEquals(None, helper) |
|
| 271 |
|
|
| 272 |
def testNoDrbdHelper(self): |
|
| 273 |
opts = mock.Mock() |
|
| 274 |
opts.drbd_storage = None |
|
| 275 |
opts.drbd_helper = "/bin/true" |
|
| 276 |
self.assertRaises(errors.OpPrereqError, gnt_cluster._InitDrbdHelper, opts) |
|
| 277 |
|
|
| 278 |
def testDrbdNoHelper(self): |
|
| 279 |
opts = mock.Mock() |
|
| 280 |
opts.drbd_storage = True |
|
| 281 |
opts.drbd_helper = None |
|
| 282 |
helper = gnt_cluster._InitDrbdHelper(opts) |
|
| 283 |
self.assertEquals(constants.DEFAULT_DRBD_HELPER, helper) |
|
| 284 |
|
|
| 285 |
def testDrbdHelper(self): |
|
| 286 |
opts = mock.Mock() |
|
| 287 |
opts.drbd_storage = True |
|
| 288 |
opts.drbd_helper = "/bin/true" |
|
| 289 |
helper = gnt_cluster._InitDrbdHelper(opts) |
|
| 290 |
self.assertEquals(opts.drbd_helper, helper) |
|
| 291 |
|
|
| 292 |
|
|
| 293 |
class GetDrbdHelper(unittest.TestCase): |
|
| 294 |
|
|
| 295 |
def testNoDrbdNoHelper(self): |
|
| 296 |
opts = mock.Mock() |
|
| 297 |
opts.drbd_storage = False |
|
| 298 |
opts.drbd_helper = None |
|
| 299 |
helper = gnt_cluster._GetDrbdHelper(opts) |
|
| 300 |
self.assertEquals("", helper)
|
|
| 301 |
|
|
| 302 |
def testNoDrbdHelper(self): |
|
| 303 |
opts = mock.Mock() |
|
| 304 |
opts.drbd_storage = None |
|
| 305 |
opts.drbd_helper = "/bin/true" |
|
| 306 |
self.assertRaises(errors.OpPrereqError, gnt_cluster._GetDrbdHelper, opts) |
|
| 307 |
|
|
| 308 |
def testDrbdNoHelper(self): |
|
| 309 |
opts = mock.Mock() |
|
| 310 |
opts.drbd_storage = True |
|
| 311 |
opts.drbd_helper = None |
|
| 312 |
helper = gnt_cluster._GetDrbdHelper(opts) |
|
| 313 |
self.assertEquals(None, helper) |
|
| 314 |
|
|
| 315 |
def testDrbdHelper(self): |
|
| 316 |
opts = mock.Mock() |
|
| 317 |
opts.drbd_storage = True |
|
| 318 |
opts.drbd_helper = "/bin/true" |
|
| 319 |
helper = gnt_cluster._GetDrbdHelper(opts) |
|
| 320 |
self.assertEquals(opts.drbd_helper, helper) |
|
| 321 |
|
|
| 322 |
|
|
| 261 | 323 |
if __name__ == "__main__": |
| 262 | 324 |
testutils.GanetiTestProgram() |
Also available in: Unified diff