mirror of https://github.com/jumpserver/jumpserver
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
55 lines
1.3 KiB
55 lines
1.3 KiB
7 years ago
|
# -*- coding: utf-8 -*-
|
||
|
|
#
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
import sys
|
||
|
|
|
||
|
|
sys.path.insert(0, "../..")
|
||
|
|
|
||
|
|
from ops.ansible.runner import AdHocRunner, CommandRunner
|
||
|
|
|
||
|
|
|
||
|
|
class TestAdHocRunner(unittest.TestCase):
|
||
|
|
def setUp(self):
|
||
|
|
host_data = [
|
||
|
|
{
|
||
|
|
"hostname": "testserver",
|
||
|
|
"ip": "192.168.244.168",
|
||
|
|
"port": 22,
|
||
|
|
"username": "root",
|
||
|
|
"password": "redhat",
|
||
|
|
},
|
||
|
|
]
|
||
|
|
self.runner = AdHocRunner(hosts=host_data)
|
||
|
|
|
||
|
|
def test_run(self):
|
||
|
|
tasks = [
|
||
|
|
{"action": {"module": "shell", "args": "ls"}},
|
||
|
|
{"action": {"module": "shell", "args": "whoami"}},
|
||
|
|
]
|
||
|
|
ret = self.runner.run(tasks, "all")
|
||
|
|
print(ret.results_summary)
|
||
|
|
print(ret.results_raw)
|
||
|
|
|
||
|
|
|
||
|
|
class TestCommandRunner(unittest.TestCase):
|
||
|
|
def setUp(self):
|
||
|
|
host_data = [
|
||
|
|
{
|
||
|
|
"hostname": "testserver",
|
||
|
|
"ip": "192.168.244.168",
|
||
|
|
"port": 22,
|
||
|
|
"username": "root",
|
||
|
|
"password": "redhat",
|
||
|
|
},
|
||
|
|
]
|
||
|
|
self.runner = CommandRunner(hosts=host_data)
|
||
|
|
|
||
|
|
def test_execute(self):
|
||
|
|
res = self.runner.execute('ls', 'all')
|
||
|
|
print(res.results_command)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|