Signed-off-by: Michael Hanselmann <han...@google.com>
---
test/ganeti.rapi.client_unittest.py | 13 +++++++++++++
1 files changed, 13 insertions(+), 0 deletions(-)
diff --git a/test/ganeti.rapi.client_unittest.py
b/test/ganeti.rapi.client_unittest.py
index 546d1c4..46262c4 100755
--- a/test/ganeti.rapi.client_unittest.py
+++ b/test/ganeti.rapi.client_unittest.py
@@ -409,6 +409,19 @@ class GanetiRapiClientTests(unittest.TestCase):
self.client.ModifyNodeStorageUnits,
"node-n", "floppy-disk", "hdc")
+ def testRepairNodeStorageUnits(self):
+ self.rapi.AddResponse("99")
+ self.assertEqual(99, self.client.RepairNodeStorageUnits("node-z", "lvm-pv",
+ "hda"))
+ self.assertHandler(rlib2.R_2_nodes_name_storage_repair)
+ self.assertItems(["node-z"])
+ self.assertQuery("storage_type", ["lvm-pv"])
+ self.assertQuery("name", ["hda"])
+
+ self.assertRaises(client.InvalidStorageType,
+ self.client.RepairNodeStorageUnits,
+ "node-n", "floppy-disk", "hdc")
+
def testGetNodeTags(self):
self.rapi.AddResponse("[\"fry\", \"bender\"]")
self.assertEqual(["fry", "bender"], self.client.GetNodeTags("node-k"))
--
1.7.0.4
