[PATCH 1 of 2 v2] bundle2.bundlepart: make mandatory part flag explicit in API
Eric Sumner
ericsumner at fb.com
Wed Dec 17 23:42:11 UTC 2014
# HG changeset patch
# User Eric Sumner <ericsumner at fb.com>
# Date 1418856629 28800
# Wed Dec 17 14:50:29 2014 -0800
# Node ID 72d491a98c7a589f905b76328f584728c12f1d79
# Parent a4e1268f4ce180063d0ead294041d07dc816cb32
bundle2.bundlepart: make mandatory part flag explicit in API
This makes all bundle2 parts mandatory unless they are expressly made advisory
via the keyword parameter or the bundlepart.mandatory property.
diff --git a/mercurial/bundle2.py b/mercurial/bundle2.py
--- a/mercurial/bundle2.py
+++ b/mercurial/bundle2.py
@@ -588,7 +588,7 @@
"""
def __init__(self, parttype, mandatoryparams=(), advisoryparams=(),
- data=''):
+ data='', mandatory = True):
self.id = None
self.type = parttype
self._data = data
@@ -605,6 +605,7 @@
# - False: currently generated,
# - True: generation done.
self._generated = None
+ self.mandatory = mandatory
# methods used to defines the part content
def __setdata(self, data):
@@ -642,9 +643,13 @@
raise RuntimeError('part can only be consumed once')
self._generated = False
#### header
+ if self.mandatory:
+ parttype = self.type.upper()
+ else:
+ parttype = self.type.lower()
## parttype
- header = [_pack(_fparttypesize, len(self.type)),
- self.type, _pack(_fpartid, self.id),
+ header = [_pack(_fparttypesize, len(parttype)),
+ parttype, _pack(_fpartid, self.id),
]
## parameters
# count
diff --git a/tests/test-bundle2-format.t b/tests/test-bundle2-format.t
--- a/tests/test-bundle2-format.t
+++ b/tests/test-bundle2-format.t
@@ -55,7 +55,8 @@
> op.ui.write('received ping request (id %i)\n' % part.id)
> if op.reply is not None and 'ping-pong' in op.reply.capabilities:
> op.ui.write_err('replying to ping request (id %i)\n' % part.id)
- > op.reply.newpart('test:pong', [('in-reply-to', str(part.id))])
+ > op.reply.newpart('test:pong', [('in-reply-to', str(part.id))],
+ > mandatory = False)
>
> @bundle2.parthandler('test:debugreply')
> def debugreply(op, part):
@@ -111,29 +112,30 @@
> bundler.newpart('b2x:changegroup', data=cg.getchunks())
>
> if opts['parts']:
- > bundler.newpart('test:empty')
+ > bundler.newpart('test:empty', mandatory = False)
> # add a second one to make sure we handle multiple parts
- > bundler.newpart('test:empty')
- > bundler.newpart('test:song', data=ELEPHANTSSONG)
- > bundler.newpart('test:debugreply')
+ > bundler.newpart('test:empty', mandatory = False)
+ > bundler.newpart('test:song', data=ELEPHANTSSONG, mandatory = False)
+ > bundler.newpart('test:debugreply', mandatory = False)
> mathpart = bundler.newpart('test:math')
> mathpart.addparam('pi', '3.14')
> mathpart.addparam('e', '2.72')
> mathpart.addparam('cooking', 'raw', mandatory=False)
> mathpart.data = '42'
+ > mathpart.mandatory = False
> # advisory known part with unknown mandatory param
- > bundler.newpart('test:song', [('randomparam','')])
+ > bundler.newpart('test:song', [('randomparam','')], mandatory = False)
> if opts['unknown']:
> bundler.newpart('test:UNKNOWN', data='some random content')
> if opts['unknownparams']:
> bundler.newpart('test:SONG', [('randomparams', '')])
> if opts['parts']:
- > bundler.newpart('test:ping')
+ > bundler.newpart('test:ping', mandatory = False)
> if opts['genraise']:
> def genraise():
> yield 'first line\n'
> raise RuntimeError('Someone set up us the bomb!')
- > bundler.newpart('b2x:output', data=genraise())
+ > bundler.newpart('b2x:output', data=genraise(), mandatory = False)
>
> if path is None:
> file = sys.stdout
@@ -608,12 +610,12 @@
$ cat ../reply.hg2
HG2Y\x00\x00\x00\x00\x00\x00\x00\x1f (esc)
- b2x:output\x00\x00\x00\x00\x00\x01\x0b\x01in-reply-to3\x00\x00\x00\xd9The choir starts singing: (esc)
+ B2X:OUTPUT\x00\x00\x00\x00\x00\x01\x0b\x01in-reply-to3\x00\x00\x00\xd9The choir starts singing: (esc)
Patali Dirapata, Cromda Cromda Ripalo, Pata Pata, Ko Ko Ko
Bokoro Dipoulito, Rondi Rondi Pepino, Pata Pata, Ko Ko Ko
Emana Karassoli, Loucra Loucra Ponponto, Pata Pata, Ko Ko Ko.
\x00\x00\x00\x00\x00\x00\x00\x1f (esc)
- b2x:output\x00\x00\x00\x01\x00\x01\x0b\x01in-reply-to4\x00\x00\x00\xc9debugreply: capabilities: (esc)
+ B2X:OUTPUT\x00\x00\x00\x01\x00\x01\x0b\x01in-reply-to4\x00\x00\x00\xc9debugreply: capabilities: (esc)
debugreply: 'city=!'
debugreply: 'celeste,ville'
debugreply: 'elephants'
@@ -621,7 +623,7 @@
debugreply: 'celeste'
debugreply: 'ping-pong'
\x00\x00\x00\x00\x00\x00\x00\x1e test:pong\x00\x00\x00\x02\x01\x00\x0b\x01in-reply-to7\x00\x00\x00\x00\x00\x00\x00\x1f (esc)
- b2x:output\x00\x00\x00\x03\x00\x01\x0b\x01in-reply-to7\x00\x00\x00=received ping request (id 7) (esc)
+ B2X:OUTPUT\x00\x00\x00\x03\x00\x01\x0b\x01in-reply-to7\x00\x00\x00=received ping request (id 7) (esc)
replying to ping request (id 7)
\x00\x00\x00\x00\x00\x00\x00\x00 (no-eol) (esc)
@@ -729,7 +731,7 @@
end of bundle
$ cat ../rev.hg2
- HG2Y\x00\x00\x00\x00\x00\x00\x00\x16\x0fb2x:changegroup\x00\x00\x00\x00\x00\x00\x00\x00\x06\x13\x00\x00\x00\xa42\xafv\x86\xd4\x03\xcfE\xb5\xd9_-p\xce\xbe\xa5\x87\xac\x80j_\xdd\xd9\x89W\xc8\xa5JMCm\xfe\x1d\xa9\xd8\x7f!\xa1\xb9{\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x002\xafv\x86\xd4\x03\xcfE\xb5\xd9_-p\xce\xbe\xa5\x87\xac\x80j\x00\x00\x00\x00\x00\x00\x00)\x00\x00\x00)6e1f4c47ecb533ffd0c8e52cdc88afb6cd39e20c (esc)
+ HG2Y\x00\x00\x00\x00\x00\x00\x00\x16\x0fB2X:CHANGEGROUP\x00\x00\x00\x00\x00\x00\x00\x00\x06\x13\x00\x00\x00\xa42\xafv\x86\xd4\x03\xcfE\xb5\xd9_-p\xce\xbe\xa5\x87\xac\x80j_\xdd\xd9\x89W\xc8\xa5JMCm\xfe\x1d\xa9\xd8\x7f!\xa1\xb9{\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x002\xafv\x86\xd4\x03\xcfE\xb5\xd9_-p\xce\xbe\xa5\x87\xac\x80j\x00\x00\x00\x00\x00\x00\x00)\x00\x00\x00)6e1f4c47ecb533ffd0c8e52cdc88afb6cd39e20c (esc)
\x00\x00\x00f\x00\x00\x00h\x00\x00\x00\x02D (esc)
\x00\x00\x00i\x00\x00\x00j\x00\x00\x00\x01D\x00\x00\x00\xa4\x95 \xee\xa7\x81\xbc\xca\x16\xc1\xe1Z\xcc\x0b\xa1C5\xa0\xe8\xe5\xba\xcd\x01\x0b\x8c\xd9\x98\xf3\x98\x1aZ\x81\x15\xf9O\x8d\xa4\xabP`\x89\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x95 \xee\xa7\x81\xbc\xca\x16\xc1\xe1Z\xcc\x0b\xa1C5\xa0\xe8\xe5\xba\x00\x00\x00\x00\x00\x00\x00)\x00\x00\x00)4dece9c826f69490507b98c6383a3009b295837d (esc)
\x00\x00\x00f\x00\x00\x00h\x00\x00\x00\x02E (esc)
@@ -768,8 +770,8 @@
addchangegroup return: 1
$ cat ../rev-reply.hg2
- HG2Y\x00\x00\x00\x00\x00\x00\x003\x15b2x:reply:changegroup\x00\x00\x00\x00\x00\x02\x0b\x01\x06\x01in-reply-to1return1\x00\x00\x00\x00\x00\x00\x00\x1f (esc)
- b2x:output\x00\x00\x00\x01\x00\x01\x0b\x01in-reply-to1\x00\x00\x00dadding changesets (esc)
+ HG2Y\x00\x00\x00\x00\x00\x00\x003\x15B2X:REPLY:CHANGEGROUP\x00\x00\x00\x00\x00\x02\x0b\x01\x06\x01in-reply-to1return1\x00\x00\x00\x00\x00\x00\x00\x1f (esc)
+ B2X:OUTPUT\x00\x00\x00\x01\x00\x01\x0b\x01in-reply-to1\x00\x00\x00dadding changesets (esc)
adding manifests
adding file changes
added 0 changesets with 0 changes to 3 files
@@ -786,7 +788,7 @@
$ cat ../genfailed.hg2
HG2Y\x00\x00\x00\x00\x00\x00\x00\x11 (esc)
- b2x:output\x00\x00\x00\x00\x00\x00\xff\xff\xff\xff\x00\x00\x00L\x0fb2x:error:abort\x00\x00\x00\x00\x01\x00\x07-messageunexpected error: Someone set up us the bomb!\x00\x00\x00\x00\x00\x00\x00\x00 (no-eol) (esc)
+ b2x:output\x00\x00\x00\x00\x00\x00\xff\xff\xff\xff\x00\x00\x00L\x0fB2X:ERROR:ABORT\x00\x00\x00\x00\x01\x00\x07-messageunexpected error: Someone set up us the bomb!\x00\x00\x00\x00\x00\x00\x00\x00 (no-eol) (esc)
And its handling on the other size raise a clean exception
More information about the Mercurial-devel
mailing list