annotate tests/test-annotate.py @ 188:5609a21fe39a

client: fail gracefully on unexpected prompts (issue5516) Right now, if hglib encounters an unexpected prompt, it fails with a rather opaque "unexpected data on required channel 'L'" error message. Furthermore, if subsequently another command is called on the same client instance, both the client and server processes lock up hard (at least on Windows), and the server process rapidly leaks ~2GB of memory. Fix this by responding with an empty string to any unexpected prompt. This will trigger an "abort: response expected" exception from the server, which is easily handled as a CommandError, and subsequent commands sent from the same client work as expected. This doesn't completely resolve bug 5516, as unexpected requests on another required channel (e.g. I) can still cause a lockup. However, it does fix the most common case of an unexpected password prompt.
author G?bor Stefanik <gabor.stefanik@nng.com>
date Tue, 12 Sep 2017 13:16:36 -0400
parents c1b966866ed7
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
148
c1b966866ed7 hglib: make all imports absolute (issue4520)
Brett Cannon <brett@python.org>
parents: 143
diff changeset
1 from tests import common
143
4359cabcb0cc hglib: move string literals in the test code to util.b() (issue4520)
Brett Cannon <brett@python.org>
parents: 134
diff changeset
2 from hglib.util import b
52
18f72b255553 client: add annotate command
Idan Kamara <idankk86@gmail.com>
parents:
diff changeset
3
18f72b255553 client: add annotate command
Idan Kamara <idankk86@gmail.com>
parents:
diff changeset
4 class test_annotate(common.basetest):
18f72b255553 client: add annotate command
Idan Kamara <idankk86@gmail.com>
parents:
diff changeset
5 def test_basic(self):
18f72b255553 client: add annotate command
Idan Kamara <idankk86@gmail.com>
parents:
diff changeset
6 self.append('a', 'a\n')
143
4359cabcb0cc hglib: move string literals in the test code to util.b() (issue4520)
Brett Cannon <brett@python.org>
parents: 134
diff changeset
7 rev, node0 = self.client.commit(b('first'), addremove=True)
52
18f72b255553 client: add annotate command
Idan Kamara <idankk86@gmail.com>
parents:
diff changeset
8 self.append('a', 'a\n')
143
4359cabcb0cc hglib: move string literals in the test code to util.b() (issue4520)
Brett Cannon <brett@python.org>
parents: 134
diff changeset
9 rev, node1 = self.client.commit(b('second'))
52
18f72b255553 client: add annotate command
Idan Kamara <idankk86@gmail.com>
parents:
diff changeset
10
143
4359cabcb0cc hglib: move string literals in the test code to util.b() (issue4520)
Brett Cannon <brett@python.org>
parents: 134
diff changeset
11 self.assertEquals(list(self.client.annotate(b('a'))),
4359cabcb0cc hglib: move string literals in the test code to util.b() (issue4520)
Brett Cannon <brett@python.org>
parents: 134
diff changeset
12 [(b('0'), b('a')), (b('1'), b('a'))])
134
1b47146a4a2c style: fix long lines
Matt Mackall <mpm@selenic.com>
parents: 52
diff changeset
13 self.assertEquals(list(
1b47146a4a2c style: fix long lines
Matt Mackall <mpm@selenic.com>
parents: 52
diff changeset
14 self.client.annotate(
143
4359cabcb0cc hglib: move string literals in the test code to util.b() (issue4520)
Brett Cannon <brett@python.org>
parents: 134
diff changeset
15 b('a'), user=True, file=True,
134
1b47146a4a2c style: fix long lines
Matt Mackall <mpm@selenic.com>
parents: 52
diff changeset
16 number=True, changeset=True, line=True, verbose=True)),
143
4359cabcb0cc hglib: move string literals in the test code to util.b() (issue4520)
Brett Cannon <brett@python.org>
parents: 134
diff changeset
17 [(b('test 0 ') + node0[:12] + b(' a:1'), b('a')),
4359cabcb0cc hglib: move string literals in the test code to util.b() (issue4520)
Brett Cannon <brett@python.org>
parents: 134
diff changeset
18 (b('test 1 ') + node1[:12] + b(' a:2'), b('a'))])
52
18f72b255553 client: add annotate command
Idan Kamara <idankk86@gmail.com>
parents:
diff changeset
19
18f72b255553 client: add annotate command
Idan Kamara <idankk86@gmail.com>
parents:
diff changeset
20 def test_files(self):
18f72b255553 client: add annotate command
Idan Kamara <idankk86@gmail.com>
parents:
diff changeset
21 self.append('a', 'a\n')
143
4359cabcb0cc hglib: move string literals in the test code to util.b() (issue4520)
Brett Cannon <brett@python.org>
parents: 134
diff changeset
22 rev, node0 = self.client.commit(b('first'), addremove=True)
52
18f72b255553 client: add annotate command
Idan Kamara <idankk86@gmail.com>
parents:
diff changeset
23 self.append('b', 'b\n')
143
4359cabcb0cc hglib: move string literals in the test code to util.b() (issue4520)
Brett Cannon <brett@python.org>
parents: 134
diff changeset
24 rev, node1 = self.client.commit(b('second'), addremove=True)
4359cabcb0cc hglib: move string literals in the test code to util.b() (issue4520)
Brett Cannon <brett@python.org>
parents: 134
diff changeset
25 self.assertEquals(list(self.client.annotate([b('a'), b('b')])),
4359cabcb0cc hglib: move string literals in the test code to util.b() (issue4520)
Brett Cannon <brett@python.org>
parents: 134
diff changeset
26 [(b('0'), b('a')), (b('1'), b('b'))])
52
18f72b255553 client: add annotate command
Idan Kamara <idankk86@gmail.com>
parents:
diff changeset
27
18f72b255553 client: add annotate command
Idan Kamara <idankk86@gmail.com>
parents:
diff changeset
28 def test_two_colons(self):
18f72b255553 client: add annotate command
Idan Kamara <idankk86@gmail.com>
parents:
diff changeset
29 self.append('a', 'a: b\n')
143
4359cabcb0cc hglib: move string literals in the test code to util.b() (issue4520)
Brett Cannon <brett@python.org>
parents: 134
diff changeset
30 self.client.commit(b('first'), addremove=True)
4359cabcb0cc hglib: move string literals in the test code to util.b() (issue4520)
Brett Cannon <brett@python.org>
parents: 134
diff changeset
31 self.assertEquals(list(self.client.annotate(b('a'))),
4359cabcb0cc hglib: move string literals in the test code to util.b() (issue4520)
Brett Cannon <brett@python.org>
parents: 134
diff changeset
32 [(b('0'), b('a: b'))])