def test_read_invalid_json(self, unlink, _open): body = '__invalid__' _open.return_value.read.return_value = body path = '/tmp/123' document = Pending._read(path) _open.assert_called_once_with(path) _open.return_value.read.assert_called_once_with() _open.return_value.close.assert_called_once_with() unlink.assert_called_once_with(path) self.assertEqual(document, None)
def test_read(self, unlink, _open): body = '{"A": 1}' _open.return_value.read.return_value = body path = '/tmp/123' document = Pending._read(path) _open.assert_called_once_with(path) _open.return_value.read.assert_called_once_with() _open.return_value.close.assert_called_once_with() self.assertFalse(unlink.called) self.assertEqual(document.__dict__, {'A': 1})
def test_read_invalid_json(self, unlink, _open): def _enter(): return _open.return_value def _exit(*unused): _open.return_value.close() _open.return_value.__enter__ = Mock(side_effect=_enter) _open.return_value.__exit__ = Mock(side_effect=_exit) body = '__invalid__' _open.return_value.read.return_value = body path = '/tmp/123' document = Pending._read(path) _open.assert_called_once_with(path) _open.return_value.read.assert_called_once_with() _open.return_value.close.assert_called_once_with() unlink.assert_called_once_with(path) self.assertEqual(document, None)
def test_read(self, unlink, _open): def _enter(): return _open.return_value def _exit(*unused): _open.return_value.close() _open.return_value.__enter__ = Mock(side_effect=_enter) _open.return_value.__exit__ = Mock(side_effect=_exit) body = '{"A": 1}' _open.return_value.read.return_value = body path = '/tmp/123' document = Pending._read(path) _open.assert_called_once_with(path) _open.return_value.read.assert_called_once_with() _open.return_value.close.assert_called_once_with() self.assertFalse(unlink.called) self.assertEqual(document.__dict__, {'A': 1})