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

go/src/infra/tools/cipd/remote_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 (
         "fmt"
         "io/ioutil"
         "net/http"
@@ skipping 66 matching lines @@
                         c.So(r.URL.Query().Get("package_path"), ShouldEqual, "pkgname")
                         c.So(r.Method, ShouldEqual, "POST")
                         body, err := ioutil.ReadAll(r.Body)
                         c.So(err, ShouldBeNil)
                         c.So(string(body), ShouldEqual, request)
                         w.Write([]byte(response))
                 })
                 return remote.modifyACL("pkgname", changes)
         }
 
+        mockAttachTags := func(c C, tags []string, request, response string) error {
+                remote := mockRemoteService(func(w http.ResponseWriter, r *http.Request) {
+                        c.So(r.URL.Path, ShouldEqual, "/_ah/api/repo/v1/tags")
+                        c.So(r.URL.Query().Get("package_name"), ShouldEqual, "pkgname")
+                        c.So(r.URL.Query().Get("instance_id"), ShouldEqual, "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa")
+                        c.So(r.Method, ShouldEqual, "POST")
+                        body, err := ioutil.ReadAll(r.Body)
+                        c.So(err, ShouldBeNil)
+                        c.So(string(body), ShouldEqual, request)
+                        w.Write([]byte(response))
+                })
+                return remote.attachTags("pkgname", "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa", tags)
+        }
+
         Convey("makeRequest POST works", t, func(c C) {
                 remote := mockRemoteService(func(w http.ResponseWriter, r *http.Request) {
                         c.So(r.Method, ShouldEqual, "POST")
                         c.So(r.URL.Path, ShouldEqual, "/_ah/api/cas/v1/method")
                         w.Write([]byte(`{"value":"123"}`))
                 })
                 var reply struct {
                         Value string `json:"value"`
                 }
                 err := remote.makeRequest("cas/v1/method", "POST", nil, &reply)
@@ skipping 330 matching lines @@
                 So(err, ShouldBeNil)
         })
 
         Convey("modifyACL ERROR", t, func(c C) {
                 err := mockModifyACL(c, []PackageACLChange{}, `{"changes":null}`, `{
                         "status": "ERROR",
                         "error_message": "Error message"
                 }`)
                 So(err, ShouldNotBeNil)
         })
+
+        Convey("attachTags SUCCESS", t, func(c C) {
+                err := mockAttachTags(
+                        c, []string{"tag1:value1", "tag2:value2"},
+                        `{"tags":["tag1:value1","tag2:value2"]}`,
+                        `{"status":"SUCCESS"}`)
+                So(err, ShouldBeNil)
+        })
+
+        Convey("attachTags no tags", t, func(c C) {
+                err := mockAttachTags(c, nil, "", "")
+                So(err, ShouldNotBeNil)
+        })
+
+        Convey("attachTags bad tag", t, func(c C) {
+                err := mockAttachTags(c, []string{"BADTAG"}, "", "")
+                So(err, ShouldNotBeNil)
+        })
+
+        Convey("attachTags PROCESSING_NOT_FINISHED_YET", t, func(c C) {
+                err := mockAttachTags(
+                        c, []string{"tag1:value1", "tag2:value2"},
+                        `{"tags":["tag1:value1","tag2:value2"]}`,
+                        `{"status":"PROCESSING_NOT_FINISHED_YET", "error_message":"Blah"}`)
+                So(err, ShouldResemble, &pendingProcessingError{message: "Blah"})
+        })
+
+        Convey("attachTags ERROR", t, func(c C) {
+                err := mockAttachTags(
+                        c, []string{"tag1:value1", "tag2:value2"},
+                        `{"tags":["tag1:value1","tag2:value2"]}`,
+                        `{"status":"ERROR", "error_message":"Blah"}`)
+                So(err, ShouldNotBeNil)
+        })
 }
 
 ////////////////////////////////////////////////////////////////////////////////
 
 type expectedHTTPCall struct {
         Method string
         Path   string
         Reply  string
         Query  url.Values
         Status int
@@ skipping 69 matching lines @@
                 // Mocked reply.
                 if exp.Status != 0 {
                         w.WriteHeader(exp.Status)
                 }
                 if exp.Reply != "" {
                         w.Write([]byte(exp.Reply))
                 }
                 index++
         })
 }