def test_is_swap(self, mock_open): magic = mock.MagicMock(spec=io.TextIOBase) mock_open.return_value = magic magic.__enter__.return_value = magic magic.__iter__.return_value = iter(_proc_swaps) self.assertTrue(misc.is_swap('/dev/sda5')) self.assertFalse(misc.is_swap('/dev/sda'))
def test_is_swap_fail(self, mock_open): mock_open.side_effect = Exception('Ka-blamo!') self.assertFalse(misc.is_swap('/dev/sda5'))