Merge "vnf_generic: convert sshmanager to class"
authorJing Lu <lvjing5@huawei.com>
Tue, 14 Feb 2017 06:04:53 +0000 (06:04 +0000)
committerGerrit Code Review <gerrit@opnfv.org>
Tue, 14 Feb 2017 06:04:53 +0000 (06:04 +0000)
1  2 
tests/unit/benchmark/scenarios/networking/test_vnf_generic.py

  # Unittest for yardstick.benchmark.scenarios.networking.test_vnf_generic
  
  from __future__ import absolute_import
 +
 +import os
  import unittest
 +
  import mock
 -import os
  
  from yardstick.benchmark.scenarios.networking.vnf_generic import \
-     ssh_manager, NetworkServiceTestCase, IncorrectConfig, IncorrectSetup
+     SshManager, NetworkServiceTestCase, IncorrectConfig, IncorrectSetup
  from yardstick.network_services.collector.subscriber import Collector
  from yardstick.network_services.vnf_generic.vnf.base import \
      GenericTrafficGen, GenericVNF
@@@ -315,10 -301,10 +315,10 @@@ class TestNetworkServiceTestCase(unitte
          with mock.patch("yardstick.ssh.SSH") as ssh:
              ssh_mock = mock.Mock(autospec=ssh.SSH)
              ssh_mock.execute = \
 -                mock.Mock(return_value=(0, SYS_CLASS_NET+IP_ADDR_SHOW, ""))
 +                mock.Mock(return_value=(0, SYS_CLASS_NET + IP_ADDR_SHOW, ""))
              ssh.return_value = ssh_mock
              for node, node_dict in self.context_cfg["nodes"].items():
-                 with ssh_manager(node_dict) as conn:
+                 with SshManager(node_dict) as conn:
                      self.assertIsNotNone(conn)
  
      def test___init__(self):