X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=system%2FTestNode.py;h=f3f5d21668d6aa8cb9e82da70a707acbccf8a8df;hb=3dfbfb8e94682f8e7b453bb959d38421cc92928b;hp=7461929c3d6a15f937c77f115b26abc5c0d91621;hpb=9385eabe5ac58244848c08805f0bec15c7e11009;p=tests.git

diff --git a/system/TestNode.py b/system/TestNode.py
index 7461929..f3f5d21 100644
--- a/system/TestNode.py
+++ b/system/TestNode.py
@@ -193,6 +193,12 @@ class TestNode:
                                            self.name(),{'boot_state':'reinstall'})
         return True
     
+    def nodestate_upgrade(self):
+        "all nodes: mark PLCAPI boot_state as upgrade"
+        self.test_plc.apiserver.UpdateNode(self.test_plc.auth_root(),
+                                           self.name(),{'boot_state':'upgrade'})
+        return True
+    
     def nodestate_safeboot(self):
         "all nodes: mark PLCAPI boot_state as safeboot"
         self.test_plc.apiserver.UpdateNode(self.test_plc.auth_root(),
@@ -211,9 +217,53 @@ class TestNode:
             print("Dry_run: skipped getting current node state")
             return True
         state = self.test_plc.apiserver.GetNodes(self.test_plc.auth_root(), self.name(), ['boot_state'])[0]['boot_state']
-        print(self.name(),':',state)
+        print("boot_state for {} : {}".format(self.name(), state))
         return True
     
+    def nodedistro_f14(self):
+        return self.nodedistro_set('f14')
+    def nodedistro_f18(self):
+        return self.nodedistro_set('f18')
+    def nodedistro_f20(self):
+        return self.nodedistro_set('f20')
+    def nodedistro_f21(self):
+        return self.nodedistro_set('f21')
+    def nodedistro_f22(self):
+        return self.nodedistro_set('f22')
+    def nodedistro_set(self, distro):
+        "set the fcdistro tag to distro, passed in arg"
+        self.test_plc.apiserver.SetNodeFcdistro(self.test_plc.auth_root(),
+                                                self.name(), distro)
+        return True
+    def nodeflavour_show(self):
+        "display the fcdistro tag - or flavour actually - of node"
+        if self.dry_run():
+            print("Dry_run: would fetch node flavour")
+            return True
+        flavour = self.test_plc.apiserver.GetNodeFlavour(self.test_plc.auth_root(),
+                                                         self.name())
+        print("Flavour for {} : {}".format(self.name(), flavour))
+        return True
+
+    def nodeplain_set(self, plain):
+        " set bootstrapfs-plain tag on nodes"
+        self.test_plc.apiserver.SetNodePlainBootstrapfs(self.test_plc.auth_root(),
+                                                        self.name(), plain)
+        return True
+    def nodeplain_on(self):
+        return self.nodeplain_set("True")
+    def nodeplain_off(self):
+        return self.nodeplain_set("")
+    def nodeplain_show(self):
+        "display bootstrapfs-plain tag"
+        if self.dry_run():
+            print("Dry_run: would fetch node plain-bootstrapfs tag")
+            return True
+        plain = self.test_plc.apiserver.GetNodePlainBootstrapfs(self.test_plc.auth_root(),
+                                                                  self.name())
+        print("Plain bootstrapfs for {} is {}".format(self.name(), plain))
+        return True
+
     def qemu_local_config(self):
         "all nodes: compute qemu config qemu.conf and store it locally"
         if not self.is_qemu():
@@ -275,6 +325,7 @@ class TestNode:
                                          .format(now, self.nodedir()), dry_run=self.dry_run()) == 0
 
     def qemu_nodefamily(self):
+        "write nodefamily stamp in qemu working dir"
         auth = self.test_plc.auth_root()
         hostname = self.node_spec['node_fields']['hostname']
         nodeflavour = self.test_plc.apiserver.GetNodeFlavour(auth, hostname)