aboutsummaryrefslogtreecommitdiffstats
path: root/test/netopeer_controler.py
blob: e95cbb4316dde982ae4669a4adc54b8afadd23c4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
#
# 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.child.setwinsize(1,1024)
        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("> ")
        return 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("> ")
        return self.child.before.decode('ascii')