Fix tests for get_hostname

This commit is contained in:
twangboy 2017-01-18 11:27:50 -07:00
parent 7ca3fd7484
commit d57a51f9f9

View file

@ -291,11 +291,11 @@ class WinSystemTestCase(TestCase):
'''
Test setting a new hostname
'''
cmd_run_mock = MagicMock(return_value="Name\nMINION")
cmd_run_mock = MagicMock(return_value="MINION")
with patch.dict(win_system.__salt__, {'cmd.run': cmd_run_mock}):
ret = win_system.get_hostname()
self.assertEqual(ret, "MINION")
cmd_run_mock.assert_called_once_with(cmd="wmic computersystem get name")
cmd_run_mock.assert_called_once_with(cmd="hostname")
if __name__ == '__main__':