view tests/test-forget.py @ 197:6949fc164439 2.5

hgclient: look for an open server before closing it in ResponseError block

At least some of the codepaths that can throw a ResponseError close
the server first (the one in _readchannel does so in order to detect
server startup failures, for example), so we have to verify we have a
server to close before doing so, otherwise we can lose the
ResponseError and the user sees an AttributeError when we try to use
.close() on a NoneType.

from tests import common
from hglib.util import b

class test_forget(common.basetest):
    def test_basic(self):
        self.append('a', 'a')
        self.client.add([b('a')])
        self.assertTrue(self.client.forget(b('a')))

    def test_warnings(self):
        self.assertFalse(self.client.forget(b('a')))
        self.append('a', 'a')
        self.client.add([b('a')])
        self.assertFalse(self.client.forget([b('a'), b('b')]))