1 files changed, 70 insertions, 0 deletions
diff --git a/test/netopeer_controler.py b/test/netopeer_controler.py
new file mode 100644
index 0000000..74c7dc6
--- /dev/null
+++ b/test/netopeer_controler.py
@@ -0,0 +1,70 @@
+#
+# Copyright (c) 2019 PANTHEON.tech.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at:
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+import pexpect
+
+class Netopeer_controler:
+    def __init__(self):
+        self.name = "netopeer2-cli"
+        self.password = "0000"
+
+    def __del__(self):
+        self.kill()
+
+    def kill(self):
+        if self.child is None:
+            return
+
+        self.child.sendline("exit\r")
+        self.child.logfile.close()
+        #self.child.kill()
+        self.child = None
+
+    def terminate(self):
+        self.kill()
+
+    def set_password(self, password):
+        self.password = password;
+
+    def spawn(self):
+        self.child = pexpect.spawn(self.name)
+        self.child.logfile = open('/var/log/Netopeer_controler.log', 'wb')
+        self.pid = self.child.pid
+        self.child.expect(">")
+        self.child.sendline("connect\r")
+        i = self.child.expect(["Password:", "Are you sure you want to continue connecting (yes/no)?"])
+        if 0 == i:
+            self.child.sendline(self.password + '\r')
+        elif 1 == i:
+            self.child.sendline("yes\r")
+            self.child.expect("Password:")
+            self.child.sendline(self.password + '\r')
+
+        self.child.expect(">")
+
+    def get(self, msg):
+        self.child.sendline("get --filter-xpath {}\r".format(msg))
+        self.child.expect("> ")
+        print(self.child.before.decode('ascii'))
+
+    def edit_config(self, msg):
+        f = open("/tmp/tmp_example.xml", "w")
+        f.write(msg)
+        f.close()
+
+        self.child.sendline("edit-config --target running --config=/tmp/tmp_example.xml\r")
+        self.child.expect("> ")
+        print(self.child.before.decode('ascii'))