def test_save(self): ''' Test if it save the current in-memory rules to disk ''' with patch.dict(nftables.__grains__, {'os_family': 'Debian'}): mock = MagicMock(return_value=False) with patch.dict(nftables.__salt__, {'cmd.run': mock}): with patch.object(salt.utils, 'fopen', MagicMock(mock_open())): self.assertEqual(nftables.save(), '#! nft -f\n\n') with patch.object(salt.utils, 'fopen', MagicMock(side_effect=IOError)): self.assertRaises(CommandExecutionError, nftables.save)
def test_save(self): ''' Test if it save the current in-memory rules to disk ''' with patch.dict(nftables.__grains__, {'os_family': 'Debian'}): mock = MagicMock(return_value=False) with patch.dict(nftables.__salt__, {'cmd.run': mock}): with patch.object(salt.utils, 'fopen', MagicMock(mock_open())): self.assertEqual(nftables.save(), '#! nft -f\n\n') with patch.object(salt.utils, 'fopen', MagicMock(side_effect=IOError)): self.assertRaises(CommandExecutionError, nftables.save)
def test_save(self): """ Test if it save the current in-memory rules to disk """ with patch.dict(nftables.__grains__, {"os_family": "Debian"}): mock = MagicMock(return_value=False) with patch.dict(nftables.__salt__, {"cmd.run": mock}): with patch.object(salt.utils.files, "fopen", MagicMock(mock_open())): self.assertEqual(nftables.save(), "#! nft -f\n\n") with patch.object(salt.utils.files, "fopen", MagicMock(side_effect=IOError)): self.assertRaises(CommandExecutionError, nftables.save)