diff options
Diffstat (limited to 'test/netopeer_controler.py')
-rw-r--r-- | test/netopeer_controler.py | 70 |
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')) |