def generate_rpc_reply(self, dev, output_file, hostname, config_data):
"""
Generates rpc-reply based on command/rpc given and stores them in snap_files
:param dev: device handler
:param output_file: filename to store snapshots
:param hostname: hostname of device
:param config_data : data of main config file
"""
val = None
test_files = []
for tfile in config_data.get('tests'):
if not os.path.isfile(tfile):
tfile = os.path.join(get_path('DEFAULT', 'test_file_path'),
tfile)
if os.path.isfile(tfile):
test_file = open(tfile, 'r')
test_files.append(yaml.load(test_file))
else:
self.logger.error(
colorama.Fore.RED +
"ERROR!! File %s is not found for taking snapshots" %
tfile,
extra=self.log_detail)
g = Parser()
for tests in test_files:
val = g.generate_reply(tests, dev, output_file, hostname, self.db)
return val
def test_rpc_1(self, mock_etree, mock_dev):
prs = Parser()
test_file = os.path.join(os.path.dirname(__file__),
'configs', 'test_rpc.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file)
dev = jnpr.junos.device.Device(
host="1.1.1.1",
user="******",
passwd="xyz")
dev.open()
m_op = mock_open()
with patch('jnpr.jsnapy.snap.open', m_op, create=True) as m_open:
prs.generate_reply(
test_file,
dev,
"1.1.1.1_snap_mock",
"1.1.1.1",
self.db)
self.assertEqual(prs.command_list, [])
self.assertEqual(
prs.rpc_list, [
'get-config', 'get-interface-information'])
self.assertEqual(
prs.test_included, [
'test_rpc_version', 'test_interface'])
dev.close()
def test_snap_sqlite_3(self, mock_insert, mock_reply, mock_init,
mock_etree, mock_dev):
mock_init.return_value = None
prs = Parser()
test_file = os.path.join(os.path.dirname(__file__), 'configs',
'delta.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file, Loader=yaml.FullLoader)
dev = jnpr.junos.device.Device(host="1.1.1.1",
user="******",
passwd="xyz")
dev.open()
m_op = mock_open()
self.db['store_in_sqlite'] = True
self.db['db_name'] = "abc.db"
with patch('jnpr.jsnapy.snap.open', m_op, create=True) as m_open:
prs.generate_reply(test_file, dev, "snap_mock", "1.1.1.1", self.db)
db_dict = dict()
db_dict['cli_command'] = 'show_chassis_fpc'
db_dict['snap_name'] = "snap_mock"
db_dict['filename'] = "1.1.1.1" + \
"_" "snap_mock" + "_" + "show_chassis_fpc" + "." + "xml"
db_dict['format'] = "xml"
db_dict['data'] = mock_reply()
mock_insert.assert_called_once_with(db_dict)
dev.close()
def test_snap_sqlite_4(self, mock_insert, mock_reply, mock_init,
mock_etree, mock_dev):
mock_init.return_value = None
prs = Parser()
calls = []
test_file = os.path.join(os.path.dirname(__file__), 'configs',
'test_rpc.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file, Loader=yaml.FullLoader)
dev = jnpr.junos.device.Device(host="1.1.1.1",
user="******",
passwd="xyz")
dev.open()
m_op = mock_open()
self.db['store_in_sqlite'] = True
self.db['db_name'] = "abc.db"
with patch('jnpr.jsnapy.snap.open', m_op, create=True) as m_open:
prs.generate_reply(test_file, dev, "snap_mock", "1.1.1.1", self.db)
db_dict = dict()
db_dict['cli_command'] = 'get-config'
db_dict['snap_name'] = "snap_mock_tcxj9UUkDu6z-Jv5vBTBtA=="
db_dict['filename'] = "1.1.1.1" + "_" + \
"snap_mock_tcxj9UUkDu6z-Jv5vBTBtA==" + "_" + "get-config" + "." + "xml"
db_dict['format'] = 'xml'
db_dict['data'] = mock_reply()
calls.append(call(db_dict))
db_dict2 = db_dict.copy()
db_dict2['snap_name'] = "snap_mock_8E_z-UpxqzKuorLyN66fYA=="
db_dict2['cli_command'] = 'get-interface-information'
db_dict2['filename'] = "1.1.1.1" + "_" + \
"snap_mock_8E_z-UpxqzKuorLyN66fYA==" + "_" + "get-interface-information" + "." + "xml"
calls.append(call(db_dict2))
mock_insert.assert_has_calls(calls)
dev.close()
def test_write_warning(self, mock_store_data):
self.db['store_in_sqlite'] = True
par = Parser()
par._write_warning("mock_reply", self.db, 'mock.xml',
self.hostname, 'mock_cmd', 'text',
'mock_output')
mock_store_data.assert_called()
def test_rpc_2(self, mock_etree, mock_parse, mock_config):
prs = Parser()
test_file = os.path.join(os.path.dirname(__file__),
'configs', 'test_rpc.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file)
dev = jnpr.junos.device.Device(
host="1.1.1.1",
user="******",
passwd="abc")
with (
patch('jnpr.junos.rpcmeta._RpcMetaExec.__getattr__')
) as (mock_rpc):
prs.generate_reply(
test_file,
dev,
"1.1.1.1_snap_mock",
"1.1.1.1",
self.db)
mock_rpc.assert_called_once_with('get_interface_information')
mock_config.assert_called_once_with(
options={
'format': 'xml'},
filter_xml=ANY)
def test_snap_sqlite_3(
self, mock_insert, mock_reply, mock_init, mock_etree, mock_dev):
mock_init.return_value = None
prs = Parser()
test_file = os.path.join(os.path.dirname(__file__),
'configs', 'delta.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file)
dev = jnpr.junos.device.Device(
host="1.1.1.1",
user="******",
passwd="xyz")
dev.open()
m_op = mock_open()
self.db['store_in_sqlite'] = True
self.db['db_name'] = "abc.db"
with patch('jnpr.jsnapy.snap.open', m_op, create=True) as m_open:
prs.generate_reply(
test_file,
dev,
"snap_mock",
"1.1.1.1",
self.db)
db_dict = dict()
db_dict['cli_command'] = 'show_chassis_fpc'
db_dict['snap_name'] = "snap_mock"
db_dict['filename'] = "1.1.1.1" + \
"_" "snap_mock" + "_" + "show_chassis_fpc" + "." + "xml"
db_dict['format'] = "xml"
db_dict['data'] = mock_reply()
mock_insert.assert_called_once_with(db_dict)
dev.close()
def generate_rpc_reply(self, dev, output_file, hostname, config_data):
"""
Generates rpc-reply based on command/rpc given and stores them in snap_files
:param dev: device handler
:param output_file: filename to store snapshots
:param hostname: hostname of device
:param config_data : data of main config file
"""
val = None
test_files = []
for tfile in config_data.get('tests'):
if not os.path.isfile(tfile):
tfile = os.path.join(
expanduser(get_path(
'DEFAULT',
'test_file_path')),
tfile)
if os.path.isfile(tfile):
test_file = open(tfile, 'r')
test_files.append(yaml.load(test_file))
else:
self.logger.error(
colorama.Fore.RED +
"ERROR!! File %s is not found for taking snapshots" %
tfile, extra=self.log_detail)
g = Parser()
for tests in test_files:
val = g.generate_reply(tests, dev, output_file, hostname, self.db)
return val
def test_generate_reply_error_2(self, mock_dev, mock_err):
par = Parser()
test_file = os.path.join(os.path.dirname(__file__),
'configs', 'bogus_testfile_2.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file)
dev = Device(user='******', host='abc', passwd='xyz')
dev.open()
par.generate_reply(test_file, dev, '1.1.1.1_snap_mock', self.hostname, self.db)
mock_err.assert_called()
def test_generate_reply_error_2(self, mock_dev, mock_err):
par = Parser()
test_file = os.path.join(os.path.dirname(__file__),
'configs', 'bogus_testfile_2.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file, Loader=yaml.FullLoader)
dev = Device(user='******', host='abc', passwd='xyz')
dev.open()
par.generate_reply(test_file, dev, '1.1.1.1_snap_mock', self.hostname, self.db)
mock_err.assert_called()
def test_generate_reply_rpc_error_4(self, mock_dev, mock_tostring, mock_err, mock_rpc, mock_write_warn):
from jnpr.junos.exception import RpcError
mock_rpc.side_effect = RpcError
par = Parser()
test_file = os.path.join(os.path.dirname(__file__),
'configs', 'bogus_testfile_4.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file)
dev = Device(host='10.221.136.250', user='******', passwd='xyz')
dev.open()
par.generate_reply(test_file, dev, 'mock.xml', self.hostname, self.db)
mock_err.assert_called()
mock_write_warn.assert_called()
def test_snap_4(self, mock_etree, mock_parse):
prs = Parser()
test_file = os.path.join(os.path.dirname(__file__), 'configs',
'delta_text.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file, Loader=yaml.FullLoader)
dev = jnpr.junos.device.Device(host="1.1.1.1",
user="******",
passwd="abc")
with patch('jnpr.junos.rpcmeta._RpcMetaExec.cli') as mock_cli:
prs.generate_reply(test_file, dev, "1.1.1.1_snap_mock", "1.1.1.1",
self.db)
mock_cli.assert_called_once_with('show chassis fpc', format='text')
def test_generate_reply_rpc_error_4(self, mock_dev, mock_tostring, mock_err, mock_rpc, mock_write_warn):
from jnpr.junos.exception import RpcError
mock_rpc.side_effect = RpcError
par = Parser()
test_file = os.path.join(os.path.dirname(__file__),
'configs', 'bogus_testfile_4.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file, Loader=yaml.FullLoader)
dev = Device(host='10.221.136.250', user='******', passwd='xyz')
dev.open()
par.generate_reply(test_file, dev, 'mock.xml', self.hostname, self.db)
mock_err.assert_called()
mock_write_warn.assert_called()
def test_rpc_6(self, mock_etree, mock_parse):
prs = Parser()
test_file = os.path.join(os.path.dirname(__file__), 'configs',
'test_rpc_2_error.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file, Loader=yaml.FullLoader)
dev = jnpr.junos.device.Device(host="1.1.1.1",
user="******",
passwd="abc")
with patch('logging.Logger.error') as mock_log:
prs.generate_reply(test_file, dev, "1.1.1.1_snap_mock", "1.1.1.1",
self.db)
c = mock_log.call_args_list[0]
self.assertNotEqual(c[0][0].find("ERROR occurred"), -1)
def test_snap_sqlite_1(self, mock_sqlite, mock_etree, mock_dev):
prs = Parser()
test_file = os.path.join(os.path.dirname(__file__), 'configs',
'delta.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file, Loader=yaml.FullLoader)
dev = jnpr.junos.device.Device(host="1.1.1.1",
user="******",
passwd="xyz")
dev.open()
m_op = mock_open()
self.db['store_in_sqlite'] = True
self.db['db_name'] = "abc.db"
with patch('jnpr.jsnapy.snap.open', m_op, create=True) as m_open:
prs.generate_reply(test_file, dev, "1.1.1.1_snap_mock", "1.1.1.1",
self.db)
mock_sqlite.assert_called_once_with('1.1.1.1', 'abc.db')
dev.close()
def test_snap_3(self, mock_etree, mock_parse):
prs = Parser()
test_file = os.path.join(os.path.dirname(__file__),
'configs', 'delta.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file)
dev = jnpr.junos.device.Device(
host="1.1.1.1",
user="******",
passwd="abc")
with patch('jnpr.junos.rpcmeta._RpcMetaExec.cli') as mock_cli:
prs.generate_reply(
test_file,
dev,
"1.1.1.1_snap_mock",
"1.1.1.1",
self.db)
mock_cli.assert_called_once_with('show chassis fpc', format='xml')
def test_snap_sqlite_2(self, mock_insert, mock_init, mock_etree, mock_dev):
mock_init.return_value = None
prs = Parser()
test_file = os.path.join(os.path.dirname(__file__), 'configs',
'delta.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file, Loader=yaml.FullLoader)
dev = jnpr.junos.device.Device(host="1.1.1.1",
user="******",
passwd="xyz")
dev.open()
m_op = mock_open()
with patch('jnpr.jsnapy.snap.open', m_op, create=True) as m_open:
prs.generate_reply(test_file, dev, "1.1.1.1_snap_mock",
"01.216.193.114", self.db)
self.assertFalse(mock_insert.called)
self.assertFalse(mock_init.called)
dev.close()
def test_snap(self, mock_log, mock_etree, mock_dev):
prs = Parser()
test_file = os.path.join(os.path.dirname(__file__), 'configs',
'delta.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file, Loader=yaml.FullLoader)
dev = jnpr.junos.device.Device(host="1.1.1.1",
user="******",
passwd="xyz")
dev.open()
m_op = mock_open()
with patch('jnpr.jsnapy.snap.open', m_op, create=True) as m_open:
prs.generate_reply(test_file, dev, "1.1.1.1_snap_mock", "1.1.1.1",
self.db)
self.assertEqual(prs.command_list, ['show chassis fpc'])
self.assertEqual(prs.rpc_list, [])
self.assertEqual(prs.test_included, ['check_chassis_fpc'])
dev.close()
def test_rpc_6(self, mock_etree, mock_parse):
prs = Parser()
test_file = os.path.join(os.path.dirname(__file__),
'configs', 'test_rpc_2_error.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file)
dev = jnpr.junos.device.Device(
host="1.1.1.1",
user="******",
passwd="abc")
with patch('logging.Logger.error') as mock_log:
prs.generate_reply(
test_file,
dev,
"1.1.1.1_snap_mock",
"1.1.1.1",
self.db)
c = mock_log.call_args_list[0]
self.assertNotEqual(c[0][0].find("ERROR occurred"), -1)
def generate_rpc_reply(self, dev, output_file, hostname, config_data, test_cases=None, **kwargs):
"""
Generates rpc-reply based on command/rpc given and stores them in snap_files
:param dev: device handler
:param output_file: filename to store snapshots
:param hostname: hostname of device
:param config_data : data of main config file
:param test_cases : list of test cases to be executed
"""
val = None
test_files = []
if test_cases is not None:
test_files = test_cases
else :
test_files = self.extract_test_cases(config_data)
g = Parser(**kwargs)
for tests in test_files:
val = g.generate_reply(tests, dev, output_file, hostname, self.db)
return val
def test_rpc_3(self, mock_etree, mock_parse, mock_dev):
prs = Parser()
test_file = os.path.join(os.path.dirname(__file__), 'configs',
'test_rpc_error.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file, Loader=yaml.FullLoader)
dev = jnpr.junos.device.Device(host="1.1.1.1",
user="******",
passwd="xyz")
dev.open()
with patch('logging.Logger.error') as mock_log:
prs.generate_reply(test_file, dev, "1.1.1.1_snap_mock", "1.1.1.1",
self.db)
c = mock_log.call_args_list[0]
self.assertNotEqual(
c[0][0].find(
"ERROR!!, filtering rpc works only for 'get-config' rpc"),
-1)
dev.close()
def test_rpc_3(self, mock_etree, mock_parse, mock_dev):
prs = Parser()
test_file = os.path.join(os.path.dirname(__file__),
'configs', 'test_rpc_error.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file)
dev = jnpr.junos.device.Device(
host="1.1.1.1",
user="******",
passwd="xyz")
dev.open()
with patch('logging.Logger.error') as mock_log:
prs.generate_reply(
test_file,
dev,
"1.1.1.1_snap_mock",
"1.1.1.1",
self.db)
c = mock_log.call_args_list[0]
self.assertNotEqual(
c[0][0].find("ERROR!!, filtering rpc works only for 'get-config' rpc"), -1)
dev.close()
def test_snap(self, mock_log, mock_etree, mock_dev):
prs = Parser()
test_file = os.path.join(os.path.dirname(__file__),
'configs', 'delta.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file)
dev = jnpr.junos.device.Device(
host="1.1.1.1",
user="******",
passwd="xyz")
dev.open()
m_op = mock_open()
with patch('jnpr.jsnapy.snap.open', m_op, create=True) as m_open:
prs.generate_reply(
test_file,
dev,
"1.1.1.1_snap_mock",
"1.1.1.1",
self.db)
self.assertEqual(prs.command_list, ['show chassis fpc'])
self.assertEqual(prs.rpc_list, [])
self.assertEqual(prs.test_included, ['check_chassis_fpc'])
dev.close()
def test_snap_sqlite_1(self, mock_sqlite, mock_etree, mock_dev):
prs = Parser()
test_file = os.path.join(os.path.dirname(__file__),
'configs', 'delta.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file)
dev = jnpr.junos.device.Device(
host="1.1.1.1",
user="******",
passwd="xyz")
dev.open()
m_op = mock_open()
self.db['store_in_sqlite'] = True
self.db['db_name'] = "abc.db"
with patch('jnpr.jsnapy.snap.open', m_op, create=True) as m_open:
prs.generate_reply(
test_file,
dev,
"1.1.1.1_snap_mock",
"1.1.1.1",
self.db)
mock_sqlite.assert_called_once_with('1.1.1.1', 'abc.db')
dev.close()
def test_snap_sqlite_2(self, mock_insert, mock_init, mock_etree, mock_dev):
mock_init.return_value = None
prs = Parser()
test_file = os.path.join(os.path.dirname(__file__),
'configs', 'delta.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file)
dev = jnpr.junos.device.Device(
host="1.1.1.1",
user="******",
passwd="xyz")
dev.open()
m_op = mock_open()
with patch('jnpr.jsnapy.snap.open', m_op, create=True) as m_open:
prs.generate_reply(
test_file,
dev,
"1.1.1.1_snap_mock",
"01.216.193.114",
self.db)
self.assertFalse(mock_insert.called)
self.assertFalse(mock_init.called)
dev.close()
def test_write_file_rpc_reply_true(self, mock_log):
par = Parser()
par._write_file(True, 'xml', 'mock.xml')
c = mock_log.call_args_list[0]
self.assertNotEqual(
c[0][0].find("Output of requested Command/RPC is empty"), -1)
def test_write_file(self, mock_info):
par = Parser()
res = par._check_reply(True, 'xml')
self.assertEqual(res, '')
mock_info.assert_called()
def test_snap_sqlite_4(
self, mock_insert, mock_reply, mock_init, mock_etree, mock_dev):
mock_init.return_value = None
prs = Parser()
calls = []
test_file = os.path.join(os.path.dirname(__file__),
'configs', 'test_rpc.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file)
dev = jnpr.junos.device.Device(
host="1.1.1.1",
user="******",
passwd="xyz")
dev.open()
m_op = mock_open()
self.db['store_in_sqlite'] = True
self.db['db_name'] = "abc.db"
with patch('jnpr.jsnapy.snap.open', m_op, create=True) as m_open:
prs.generate_reply(
test_file,
dev,
"snap_mock",
"1.1.1.1",
self.db)
db_dict = dict()
db_dict['cli_command'] = 'get-config'
db_dict['snap_name'] = "snap_mock_tcxj9UUkDu6z-Jv5vBTBtA=="
db_dict['filename'] = "1.1.1.1" + "_" + \
"snap_mock_tcxj9UUkDu6z-Jv5vBTBtA==" + "_" + "get-config" + "." + "xml"
db_dict['format'] = 'xml'
db_dict['data'] = mock_reply()
calls.append(call(db_dict))
db_dict2 = db_dict.copy()
db_dict2['snap_name'] = "snap_mock_8E_z-UpxqzKuorLyN66fYA=="
db_dict2['cli_command'] = 'get-interface-information'
db_dict2['filename'] = "1.1.1.1" + "_" + \
"snap_mock_8E_z-UpxqzKuorLyN66fYA==" + "_" + "get-interface-information" + "." + "xml"
calls.append(call(db_dict2))
mock_insert.assert_has_calls(calls)
dev.close()
@patch('logging.Logger.info')
def test_write_file(self, mock_info):
par = Parser()
res = par._check_reply(True, 'xml')
self.assertEqual(res, '')
mock_info.assert_called()
@patch('logging.Logger.error')
@patch('ncclient.manager.connect')
def test_generate_reply_error_1(self, mock_dev, mock_err):
par = Parser()
test_file = os.path.join(os.path.dirname(__file__),
'configs', 'bogus_testfile_1.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file)
dev = Device(user='******', host='abc', passwd='xyz')
dev.open()
par.generate_reply(test_file, dev, '1.1.1.1_snap_mock', self.hostname, self.db)
mock_err.assert_called()
@patch('logging.Logger.error')
@patch('ncclient.manager.connect')
def test_generate_reply_error_2(self, mock_dev, mock_err):
par = Parser()
test_file = os.path.join(os.path.dirname(__file__),
'configs', 'bogus_testfile_2.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file)
dev = Device(user='******', host='abc', passwd='xyz')
dev.open()
par.generate_reply(test_file, dev, '1.1.1.1_snap_mock', self.hostname, self.db)
mock_err.assert_called()
@patch('jnpr.jsnapy.snap.Parser._write_warning')
@patch('jnpr.junos.rpcmeta._RpcMetaExec.cli')
@patch('logging.Logger.error')
@patch('lxml.etree.tostring')
@patch('ncclient.manager.connect')
def test_generate_reply_error_3(self, mock_dev, mock_tostring, mock_err, mock_cli, mock_write_warn):
from jnpr.junos.exception import RpcError
mock_cli.side_effect = RpcError
par = Parser()
test_file = os.path.join(os.path.dirname(__file__),
'configs', 'bogus_testfile_3.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file)
dev = Device(host='10.221.136.250', user='******', passwd='xyz')
dev.open()
par.generate_reply(test_file, dev, 'mock.xml', self.hostname, self.db)
mock_err.assert_called()
mock_write_warn.assert_called()
@patch('jnpr.jsnapy.snap.Parser._write_warning')
@patch('jnpr.junos.rpcmeta._RpcMetaExec.__getattr__')
@patch('logging.Logger.error')
@patch('lxml.etree.tostring')
@patch('ncclient.manager.connect')
def test_generate_reply_rpc_error_4(self, mock_dev, mock_tostring, mock_err, mock_rpc, mock_write_warn):
from jnpr.junos.exception import RpcError
mock_rpc.side_effect = RpcError
par = Parser()
test_file = os.path.join(os.path.dirname(__file__),
'configs', 'bogus_testfile_4.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file)
dev = Device(host='10.221.136.250', user='******', passwd='xyz')
dev.open()
par.generate_reply(test_file, dev, 'mock.xml', self.hostname, self.db)
mock_err.assert_called()
mock_write_warn.assert_called()
@patch('jnpr.jsnapy.snap.Parser._write_warning')
@patch('jnpr.junos.rpcmeta._RpcMetaExec.__getattr__')
@patch('logging.Logger.error')
@patch('lxml.etree.tostring')
@patch('ncclient.manager.connect')
def test_generate_reply_rpc_error_5(self, mock_dev, mock_tostring, mock_err, mock_rpc, mock_write_warn):
from jnpr.junos.exception import RpcError
mock_rpc.side_effect = RpcError
par = Parser()
test_file = os.path.join(os.path.dirname(__file__),
'configs', 'bogus_testfile_5.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file)
dev = Device(host='10.221.136.250', user='******', passwd='xyz')
dev.open()
par.generate_reply(test_file, dev, 'mock.xml', self.hostname, self.db)
mock_err.assert_called()
mock_write_warn.assert_called()
@patch('logging.Logger.info')
def test_write_file_rpc_reply_true(self, mock_log):
par = Parser()
par._write_file(True, 'xml', 'mock.xml')
c = mock_log.call_args_list[0]
self.assertNotEqual(
c[0][0].find("Output of requested Command/RPC is empty"), -1)
@patch('jnpr.jsnapy.snap.Parser.store_in_sqlite')
def test_write_warning(self, mock_store_data):
self.db['store_in_sqlite'] = True
par = Parser()
par._write_warning("mock_reply", self.db, 'mock.xml', self.hostname
, 'mock_cmd', 'text', 'mock_output')
mock_store_data.assert_called()
def test_snap_sqlite_4(self, mock_insert, mock_reply, mock_init,
mock_etree, mock_dev):
mock_init.return_value = None
prs = Parser()
calls = []
test_file = os.path.join(os.path.dirname(__file__), 'configs',
'test_rpc.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file)
dev = jnpr.junos.device.Device(host="1.1.1.1",
user="******",
passwd="xyz")
dev.open()
m_op = mock_open()
self.db['store_in_sqlite'] = True
self.db['db_name'] = "abc.db"
with patch('jnpr.jsnapy.snap.open', m_op, create=True) as m_open:
prs.generate_reply(test_file, dev, "snap_mock", "1.1.1.1", self.db)
db_dict = dict()
db_dict['cli_command'] = 'get-config'
db_dict['snap_name'] = "snap_mock"
db_dict['filename'] = "1.1.1.1" + "_" + \
"snap_mock" + "_" + "get-config" + "." + "xml"
db_dict['format'] = 'xml'
db_dict['data'] = mock_reply()
calls.append(call(db_dict))
db_dict2 = db_dict.copy()
db_dict2['cli_command'] = 'get-interface-information'
db_dict2['filename'] = "1.1.1.1" + "_" + \
"snap_mock" + "_" + "get-interface-information" + "." + "xml"
calls.append(call(db_dict2))
mock_insert.assert_has_calls(calls)
dev.close()
@patch('logging.Logger.info')
def test_write_file(self, mock_info):
par = Parser()
res = par._check_reply(True, 'xml')
self.assertEqual(res, '')
mock_info.assert_called()
@patch('logging.Logger.error')
@patch('ncclient.manager.connect')
def test_generate_reply_error_1(self, mock_dev, mock_err):
par = Parser()
test_file = os.path.join(os.path.dirname(__file__), 'configs',
'bogus_testfile_1.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file)
dev = Device(user='******', host='abc', passwd='xyz')
dev.open()
par.generate_reply(test_file, dev, '1.1.1.1_snap_mock',
self.hostname, self.db)
mock_err.assert_called()
@patch('logging.Logger.error')
@patch('ncclient.manager.connect')
def test_generate_reply_error_2(self, mock_dev, mock_err):
par = Parser()
test_file = os.path.join(os.path.dirname(__file__), 'configs',
'bogus_testfile_2.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file)
dev = Device(user='******', host='abc', passwd='xyz')
dev.open()
par.generate_reply(test_file, dev, '1.1.1.1_snap_mock',
self.hostname, self.db)
mock_err.assert_called()
@patch('jnpr.jsnapy.snap.Parser._write_warning')
@patch('jnpr.junos.rpcmeta._RpcMetaExec.cli')
@patch('logging.Logger.error')
@patch('lxml.etree.tostring')
@patch('ncclient.manager.connect')
def test_generate_reply_error_3(self, mock_dev, mock_tostring,
mock_err, mock_cli, mock_write_warn):
from jnpr.junos.exception import RpcError
mock_cli.side_effect = RpcError
par = Parser()
test_file = os.path.join(os.path.dirname(__file__), 'configs',
'bogus_testfile_3.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file)
dev = Device(host='10.221.136.250', user='******', passwd='xyz')
dev.open()
par.generate_reply(test_file, dev, 'mock.xml', self.hostname,
self.db)
mock_err.assert_called()
mock_write_warn.assert_called()
@patch('jnpr.jsnapy.snap.Parser._write_warning')
@patch('jnpr.junos.rpcmeta._RpcMetaExec.__getattr__')
@patch('logging.Logger.error')
@patch('lxml.etree.tostring')
@patch('ncclient.manager.connect')
def test_generate_reply_rpc_error_4(self, mock_dev, mock_tostring,
mock_err, mock_rpc,
mock_write_warn):
from jnpr.junos.exception import RpcError
mock_rpc.side_effect = RpcError
par = Parser()
test_file = os.path.join(os.path.dirname(__file__), 'configs',
'bogus_testfile_4.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file)
dev = Device(host='10.221.136.250', user='******', passwd='xyz')
dev.open()
par.generate_reply(test_file, dev, 'mock.xml', self.hostname,
self.db)
mock_err.assert_called()
mock_write_warn.assert_called()
@patch('jnpr.jsnapy.snap.Parser._write_warning')
@patch('jnpr.junos.rpcmeta._RpcMetaExec.__getattr__')
@patch('logging.Logger.error')
@patch('lxml.etree.tostring')
@patch('ncclient.manager.connect')
def test_generate_reply_rpc_error_5(self, mock_dev, mock_tostring,
mock_err, mock_rpc,
mock_write_warn):
from jnpr.junos.exception import RpcError
mock_rpc.side_effect = RpcError
par = Parser()
test_file = os.path.join(os.path.dirname(__file__), 'configs',
'bogus_testfile_5.yml')
test_file = open(test_file, 'r')
test_file = yaml.load(test_file)
dev = Device(host='10.221.136.250', user='******', passwd='xyz')
dev.open()
par.generate_reply(test_file, dev, 'mock.xml', self.hostname,
self.db)
mock_err.assert_called()
mock_write_warn.assert_called()
@patch('logging.Logger.info')
def test_write_file_rpc_reply_true(self, mock_log):
par = Parser()
par._write_file(True, 'xml', 'mock.xml')
c = mock_log.call_args_list[0]
self.assertNotEqual(
c[0][0].find("Output of requested Command/RPC is empty"), -1)
@patch('jnpr.jsnapy.snap.Parser.store_in_sqlite')
def test_write_warning(self, mock_store_data):
self.db['store_in_sqlite'] = True
par = Parser()
par._write_warning("mock_reply", self.db, 'mock.xml',
self.hostname, 'mock_cmd', 'text',
'mock_output')
mock_store_data.assert_called()
def test_write_file(self, mock_info):
par = Parser()
res = par._check_reply(True, 'xml')
self.assertEqual(res,'')
mock_info.assert_called()
def test_write_file_rpc_reply_true(self, mock_log):
par = Parser()
par._write_file(True,'xml','mock.xml')
c = mock_log.call_args_list[0]
self.assertNotEqual(
c[0][0].find("Output of requested Command/RPC is empty"), -1)
def test_write_warning(self, mock_store_data):
self.db['store_in_sqlite'] = True
par = Parser()
par._write_warning("mock_reply", self.db, 'mock.xml',self.hostname
,'mock_cmd','text','mock_output')
mock_store_data.assert_called()
