# -*- coding: utf-8 -*- ''' :codeauthor: :email:`Shane Lee ` ''' # Import Python libs from __future__ import absolute_import, print_function, unicode_literals # Import Salt Testing Libs from tests.support.unit import TestCase from tests.support.mock import ( patch, mock_open, MagicMock, ) # Import Salt Libs import salt.grains.fibre_channel as fibre_channel class FibreChannelGrainsTestCase(TestCase): ''' Test cases for iscsi grains ''' def test_windows_fibre_channel_wwns_grains(self): wwns = ['20:00:00:25:b5:11:11:4c', '20:00:00:25:b5:11:11:5c', '20:00:00:25:b5:44:44:4c', '20:00:00:25:b5:44:44:5c'] cmd_run_mock = MagicMock(return_value=wwns) with patch('salt.modules.cmdmod.powershell', cmd_run_mock): ret = fibre_channel._windows_wwns() assert ret == wwns, ret def test_linux_fibre_channel_wwns_grains(self): contents = ['0x500143802426baf4', '0x500143802426baf5'] files = ['file1', 'file2'] with patch('glob.glob', MagicMock(return_value=files)), \ patch('salt.utils.files.fopen', mock_open(read_data=contents)): ret = fibre_channel._linux_wwns() assert ret == ['500143802426baf4', '500143802426baf5'], ret