def test_execute_command_failed(self, mock_ssh):
options = {"interval": 1, "count": 1}
args = {'options': options}
n = netutilization.NetUtilization(args, self.ctx)
mock_ssh.SSH.from_node().execute.return_value = (0, '', '')
n.setup()
mock_ssh.SSH.from_node().execute.return_value = (127, '', 'abcdefg')
self.assertRaises(RuntimeError, n._execute_command, "failed")
def test_setup_success(self, mock_ssh):
options = {"interval": 1, "count": 1}
args = {'options': options}
n = netutilization.NetUtilization(args, self.ctx)
mock_ssh.SSH.from_node().execute.return_value = (0, '', '')
n.setup()
self.assertIsNotNone(n.client)
self.assertTrue(n.setup_done)
def test_execute_command_success(self, mock_ssh):
options = {"interval": 1, "count": 1}
args = {'options': options}
n = netutilization.NetUtilization(args, self.ctx)
mock_ssh.SSH.from_node().execute.return_value = (0, '', '')
n.setup()
expected_result = 'abcdefg'
mock_ssh.SSH.from_node().execute.return_value = (0, expected_result,
'')
result = n._execute_command("foo")
self.assertEqual(result, expected_result)
def test_get_network_utilization_success(self, mock_ssh):
options = {
"interval": 1,
"count": 1
}
args = {'options': options}
n = netutilization.NetUtilization(args, self.ctx)
mock_ssh.SSH().execute.return_value = (0, '', '')
n.setup()
mpstat_output = self._read_file("netutilization_sample_output1.txt")
mock_ssh.SSH().execute.return_value = (0, mpstat_output, '')
result = n._get_network_utilization()
expected_result = \
{"network_utilization_maximun": {
"lo": {"rxcmp/s": "0.00",
"%ifutil": "0.00",
"txcmp/s": "0.00",
"txkB/s": "0.00",
"rxkB/s": "0.00",
"rxpck/s": "0.00",
"txpck/s": "0.00",
"rxmcst/s": "0.00"},
"eth0": {"rxcmp/s": "0.00",
"%ifutil": "0.00",
"txcmp/s": "0.00",
"txkB/s": "0.00",
"rxkB/s": "0.00",
"rxpck/s": "0.00",
"txpck/s": "0.00",
"rxmcst/s": "0.00"}},
"network_utilization_average": {
"lo": {"rxcmp/s": "0.00",
"%ifutil": "0.00",
"txcmp/s": "0.00",
"txkB/s": "0.00",
"rxkB/s": "0.00",
"rxpck/s": "0.00",
"txpck/s": "0.00",
"rxmcst/s": "0.00"},
"eth0": {"rxcmp/s": "0.00",
"%ifutil": "0.00",
"txcmp/s": "0.00",
"txkB/s": "0.00",
"rxkB/s": "0.00",
"rxpck/s": "0.00",
"txpck/s": "0.00",
"rxmcst/s": "0.00"}},
"network_utilization_minimum": {
"lo": {"rxcmp/s": "0.00",
"%ifutil": "0.00",
"txcmp/s": "0.00",
"txkB/s": "0.00",
"rxkB/s": "0.00",
"rxpck/s": "0.00",
"txpck/s": "0.00",
"rxmcst/s": "0.00"},
"eth0": {"rxcmp/s": "0.00",
"%ifutil": "0.00",
"txcmp/s": "0.00",
"txkB/s": "0.00",
"rxkB/s": "0.00",
"rxpck/s": "0.00",
"txpck/s": "0.00",
"rxmcst/s": "0.00"}}}
self.assertDictEqual(result, expected_result)