Revision b7a1c816 test/ganeti.rapi.client_unittest.py

b/test/ganeti.rapi.client_unittest.py
151 151
    self.assertEqual(client._REQ_DATA_VERSION_FIELD, rlib2._REQ_DATA_VERSION)
152 152
    self.assertEqual(client._INST_CREATE_REQV1, rlib2._INST_CREATE_REQV1)
153 153
    self.assertEqual(client._INST_REINSTALL_REQV1, rlib2._INST_REINSTALL_REQV1)
154
    self.assertEqual(client._NODE_MIGRATE_REQV1, rlib2._NODE_MIGRATE_REQV1)
154 155
    self.assertEqual(client._INST_NIC_PARAMS, constants.INIC_PARAMS)
155 156
    self.assertEqual(client.JOB_STATUS_QUEUED, constants.JOB_STATUS_QUEUED)
156 157
    self.assertEqual(client.JOB_STATUS_WAITLOCK, constants.JOB_STATUS_WAITLOCK)
......
835 836
                      "node-4", iallocator="hail", remote_node="node-5")
836 837

  
837 838
  def testMigrateNode(self):
839
    self.rapi.AddResponse(serializer.DumpJson([]))
838 840
    self.rapi.AddResponse("1111")
839 841
    self.assertEqual(1111, self.client.MigrateNode("node-a", dry_run=True))
840 842
    self.assertHandler(rlib2.R_2_nodes_name_migrate)
841 843
    self.assertItems(["node-a"])
842 844
    self.assert_("mode" not in self.rapi.GetLastHandler().queryargs)
843 845
    self.assertDryRun()
846
    self.assertFalse(self.rapi.GetLastRequestData())
844 847

  
848
    self.rapi.AddResponse(serializer.DumpJson([]))
845 849
    self.rapi.AddResponse("1112")
846 850
    self.assertEqual(1112, self.client.MigrateNode("node-a", dry_run=True,
847 851
                                                   mode="live"))
......
849 853
    self.assertItems(["node-a"])
850 854
    self.assertQuery("mode", ["live"])
851 855
    self.assertDryRun()
856
    self.assertFalse(self.rapi.GetLastRequestData())
857

  
858
    self.rapi.AddResponse(serializer.DumpJson([]))
859
    self.assertRaises(client.GanetiApiError, self.client.MigrateNode,
860
                      "node-c", target_node="foonode")
861
    self.assertEqual(self.rapi.CountPending(), 0)
862

  
863
  def testMigrateNodeBodyData(self):
864
    self.rapi.AddResponse(serializer.DumpJson([rlib2._NODE_MIGRATE_REQV1]))
865
    self.rapi.AddResponse("27539")
866
    self.assertEqual(27539, self.client.MigrateNode("node-a", dry_run=False,
867
                                                    mode="live"))
868
    self.assertHandler(rlib2.R_2_nodes_name_migrate)
869
    self.assertItems(["node-a"])
870
    self.assertFalse(self.rapi.GetLastHandler().queryargs)
871
    self.assertEqual(serializer.LoadJson(self.rapi.GetLastRequestData()),
872
                     { "mode": "live", })
873

  
874
    self.rapi.AddResponse(serializer.DumpJson([rlib2._NODE_MIGRATE_REQV1]))
875
    self.rapi.AddResponse("14219")
876
    self.assertEqual(14219, self.client.MigrateNode("node-x", dry_run=True,
877
                                                    target_node="node9",
878
                                                    iallocator="ial"))
879
    self.assertHandler(rlib2.R_2_nodes_name_migrate)
880
    self.assertItems(["node-x"])
881
    self.assertDryRun()
882
    self.assertEqual(serializer.LoadJson(self.rapi.GetLastRequestData()),
883
                     { "target_node": "node9", "iallocator": "ial", })
884

  
885
    self.assertEqual(self.rapi.CountPending(), 0)
852 886

  
853 887
  def testGetNodeRole(self):
854 888
    self.rapi.AddResponse("\"master\"")

Also available in: Unified diff