Revision effab4ca test/ganeti.query_unittest.py

b/test/ganeti.query_unittest.py
355 355
                   master_candidate=(name != master_name and idx % 3 == 0),
356 356
                   offline=False,
357 357
                   drained=False,
358
                   vm_capable=False,
358
                   vm_capable=True,
359 359
                   master_capable=False,
360 360
                   ndparams={},
361 361
                   group="default",
......
468 468

  
469 469
  def testGetLiveNodeField(self):
470 470
    nodes = [
471
      objects.Node(name="node1", drained=False, offline=False),
472
      objects.Node(name="node2", drained=True, offline=False),
473
      objects.Node(name="node3", drained=False, offline=False),
474
      objects.Node(name="node4", drained=False, offline=True),
471
      objects.Node(name="node1", drained=False, offline=False,
472
                   vm_capable=True),
473
      objects.Node(name="node2", drained=True, offline=False,
474
                   vm_capable=True),
475
      objects.Node(name="node3", drained=False, offline=False,
476
                   vm_capable=True),
477
      objects.Node(name="node4", drained=False, offline=True,
478
                   vm_capable=True),
479
      objects.Node(name="node5", drained=False, offline=False,
480
                   vm_capable=False),
475 481
      ]
476 482
    live_data = dict.fromkeys([node.name for node in nodes], {})
477 483

  
......
511 517
    self.assertRaises(AssertionError, query._GetLiveNodeField,
512 518
                      "hello", constants.QFT_BOOL, ctx, nodes[0])
513 519

  
520
    # Non-vm_capable node
521
    assert not nodes[4].vm_capable
522
    ctx = _QueryData(None, curlive_data={})
523
    self.assertEqual(query._GetLiveNodeField("hello", constants.QFT_NUMBER,
524
                                             ctx, nodes[4]),
525
                     query._FS_UNAVAIL, None)
526

  
514 527

  
515 528
class TestInstanceQuery(unittest.TestCase):
516 529
  def _Create(self, selected):

Also available in: Unified diff