diff options
Diffstat (limited to 'clients/common')
| -rw-r--r-- | clients/common/common.go | 16 | ||||
| -rw-r--r-- | clients/common/common_test.go | 13 |
2 files changed, 16 insertions, 13 deletions
diff --git a/clients/common/common.go b/clients/common/common.go index 460eb10..ef35440 100644 --- a/clients/common/common.go +++ b/clients/common/common.go @@ -268,8 +268,16 @@ func (r *GitUploadPackInfo) Bytes() []byte { } type GitUploadPackRequest struct { - Want []core.Hash - Have []core.Hash + Wants []core.Hash + Haves []core.Hash +} + +func (r *GitUploadPackRequest) Want(h ...core.Hash) { + r.Wants = append(r.Wants, h...) +} + +func (r *GitUploadPackRequest) Have(h ...core.Hash) { + r.Haves = append(r.Haves, h...) } func (r *GitUploadPackRequest) String() string { @@ -279,11 +287,11 @@ func (r *GitUploadPackRequest) String() string { func (r *GitUploadPackRequest) Reader() *strings.Reader { e := pktline.NewEncoder() - for _, want := range r.Want { + for _, want := range r.Wants { e.AddLine(fmt.Sprintf("want %s", want)) } - for _, have := range r.Have { + for _, have := range r.Haves { e.AddLine(fmt.Sprintf("have %s", have)) } diff --git a/clients/common/common_test.go b/clients/common/common_test.go index 5ffb402..dac15b6 100644 --- a/clients/common/common_test.go +++ b/clients/common/common_test.go @@ -108,15 +108,10 @@ func (s *SuiteCommon) TestGitUploadPackEncode(c *C) { } func (s *SuiteCommon) TestGitUploadPackRequest(c *C) { - r := &GitUploadPackRequest{ - Want: []core.Hash{ - core.NewHash("d82f291cde9987322c8a0c81a325e1ba6159684c"), - core.NewHash("2b41ef280fdb67a9b250678686a0c3e03b0a9989"), - }, - Have: []core.Hash{ - core.NewHash("6ecf0ef2c2dffb796033e5a02219af86ec6584e5"), - }, - } + r := &GitUploadPackRequest{} + r.Want(core.NewHash("d82f291cde9987322c8a0c81a325e1ba6159684c")) + r.Want(core.NewHash("2b41ef280fdb67a9b250678686a0c3e03b0a9989")) + r.Have(core.NewHash("6ecf0ef2c2dffb796033e5a02219af86ec6584e5")) c.Assert(r.String(), Equals, "0032want d82f291cde9987322c8a0c81a325e1ba6159684c\n"+ |