class HuaweiMacTest(unittest2.TestCase):
    """test get mac from huawei device."""
    def setUp(self):
        super(HuaweiMacTest, self).setUp()
        logsetting.init()
        host = '192.168.1.1'
        credential = {'version': '2c', 'community': 'public'}
        self.mac_plugin = Mac(host, credential)

    def tearDown(self):
        del self.mac_plugin
        super(HuaweiMacTest, self).tearDown()

    @patch("compass.hdsdiscovery.utils.snmpwalk_by_cl")
    def test_process_data(self, mock_snmpwalk):
        """get progress data function."""
        # GET operation haven't been implemeneted.
        self.assertIsNone(self.mac_plugin.process_data('GET'))

        # SNMP Walk Timeout
        # utils.snmpwalk_by_cl = Mock(return_value=None)
        mock_snmpwalk.return_value = None
        self.assertIsNone(self.mac_plugin.process_data())

        # Successfully get MAC addresses from the switch
        mock_snmp_walk_result = [{
            "iid": "40.110.212.77.198.190.88.1.48",
            "value": "10"
        }, {
            "iid": "40.110.212.100.199.74.88.1.48",
            "value": "11"
        }, {
            "iid": "0.12.41.53.220.2.88.1.48",
            "value": "12"
        }]
        expected_mac_info = [{
            "mac": "28:6e:d4:4d:c6:be",
            "port": "1",
            "vlan": "88"
        }, {
            "mac": "28:6e:d4:64:c7:4a",
            "port": "2",
            "vlan": "88"
        }, {
            "mac": "00:0c:29:35:dc:02",
            "port": "3",
            "vlan": "88"
        }]
        # utils.snmpwalk_by_cl = Mock(return_value=mock_snmp_walk_result)
        mock_snmpwalk.return_value = mock_snmp_walk_result
        self.mac_plugin.get_port = Mock()
        self.mac_plugin.get_port.side_effect = ["1", "2", "3"]
        result = self.mac_plugin.process_data()
        self.assertEqual(expected_mac_info, result)
class HuaweiMacTest(unittest2.TestCase):
    """test get mac from huawei device."""

    def setUp(self):
        super(HuaweiMacTest, self).setUp()
        logsetting.init()
        host = '192.168.1.1'
        credential = {'version': '2c', 'community': 'public'}
        self.mac_plugin = Mac(host, credential)

    def tearDown(self):
        del self.mac_plugin
        super(HuaweiMacTest, self).tearDown()

    @patch("compass.hdsdiscovery.utils.snmpwalk_by_cl")
    def test_process_data(self, mock_snmpwalk):
        """get progress data function."""
        # GET operation haven't been implemeneted.
        self.assertIsNone(self.mac_plugin.process_data('GET'))

        # SNMP Walk Timeout
        # utils.snmpwalk_by_cl = Mock(return_value=None)
        mock_snmpwalk.return_value = None
        self.assertIsNone(self.mac_plugin.process_data())

        # Successfully get MAC addresses from the switch
        mock_snmp_walk_result = [
            {"iid": "40.110.212.77.198.190.88.1.48", "value": "10"},
            {"iid": "40.110.212.100.199.74.88.1.48", "value": "11"},
            {"iid": "0.12.41.53.220.2.88.1.48", "value": "12"}
        ]
        expected_mac_info = [
            {"mac": "28:6e:d4:4d:c6:be", "port": "1", "vlan": "88"},
            {"mac": "28:6e:d4:64:c7:4a", "port": "2", "vlan": "88"},
            {"mac": "00:0c:29:35:dc:02", "port": "3", "vlan": "88"}
        ]
        # utils.snmpwalk_by_cl = Mock(return_value=mock_snmp_walk_result)
        mock_snmpwalk.return_value = mock_snmp_walk_result
        self.mac_plugin.get_port = Mock()
        self.mac_plugin.get_port.side_effect = ["1", "2", "3"]
        result = self.mac_plugin.process_data()
        self.assertEqual(expected_mac_info, result)
class HuaweiMacTest(unittest2.TestCase):
    def setUp(self):
        host = '172.29.8.40'
        credential = {'Version': 'v2c', 'Community': 'public'}
        self.mac = Mac(host, credential)

    def tearDown(self):
        del self.mac

    def test_ProcessData_Operation(self):
        # GET operation haven't been implemeneted.
        self.assertIsNone(self.mac.process_data('GET'))
 def setUp(self):
     host = '172.29.8.40'
     credential = {'Version': 'v2c', 'Community': 'public'}
     self.mac = Mac(host, credential)
 def setUp(self):
     super(HuaweiMacTest, self).setUp()
     logsetting.init()
     host = '192.168.1.1'
     credential = {'version': '2c', 'community': 'public'}
     self.mac_plugin = Mac(host, credential)
 def setUp(self):
     super(HuaweiMacTest, self).setUp()
     logsetting.init()
     host = '192.168.1.1'
     credential = {'version': '2c', 'community': 'public'}
     self.mac_plugin = Mac(host, credential)