cipd: Add -tag option to 'create' and 'pkg-register' subcomands.

go/src/infra/tools/cipd/uploader_test.go

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 package cipd
 
 import (
         "bytes"
         "fmt"
         "io/ioutil"
@@ skipping 196 matching lines @@
                         UploadOptions: UploadOptions{
                                 Client: client,
                                 Log:    logging.DefaultLogger,
                         },
                 })
                 So(err, ShouldBeNil)
                 So(uploaded.Bytes(), ShouldResemble, []byte(dataToUpload))
         })
 }
 
+func TestAttachTagsWhenReady(t *testing.T) {
+        Convey("Mocking clock", t, func() {
+                mockClock(time.Now())
+
+                Convey("attachTagsWhenReady works", func() {
+                        remote := mockRemoteServiceWithExpectations([]expectedHTTPCall{
+                                {
+                                        Method: "POST",
+                                        Path:   "/_ah/api/repo/v1/tags",
+                                        Query: url.Values{
+                                                "instance_id":  []string{"aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"},
+                                                "package_name": []string{"pkgname"},
+                                        },
+                                        Reply: `{"status": "PROCESSING_NOT_FINISHED_YET"}`,
+                                },
+                                {
+                                        Method: "POST",
+                                        Path:   "/_ah/api/repo/v1/tags",
+                                        Query: url.Values{
+                                                "instance_id":  []string{"aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"},
+                                                "package_name": []string{"pkgname"},
+                                        },
+                                        Reply: `{"status": "SUCCESS"}`,
+                                },
+                        })
+                        err := attachTagsWhenReady(
+                                remote, "pkgname", "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa",
+                                []string{"tag1:value1"}, logging.DefaultLogger)
+                        So(err, ShouldBeNil)
+                })
+
+                Convey("attachTagsWhenReady timeout", func() {
+                        calls := []expectedHTTPCall{}
+                        for i := 0; i < 20; i++ {
+                                calls = append(calls, expectedHTTPCall{
+                                        Method: "POST",
+                                        Path:   "/_ah/api/repo/v1/tags",
+                                        Query: url.Values{
+                                                "instance_id":  []string{"aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"},
+                                                "package_name": []string{"pkgname"},
+                                        },
+                                        Reply: `{"status": "PROCESSING_NOT_FINISHED_YET"}`,
+                                })
+                        }
+                        remote := mockRemoteServiceWithExpectations(calls)
+                        err := attachTagsWhenReady(
+                                remote, "pkgname", "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa",
+                                []string{"tag1:value1"}, logging.DefaultLogger)
+                        So(err, ShouldEqual, ErrAttachTagsTimeout)
+                })
+        })
+}
+
 func mockResumableUpload() {
         prev := resumableUpload
         resumableUpload = func(string, int64, UploadToCASOptions) error {
                 return nil
         }
         Reset(func() { resumableUpload = prev })
 }