1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73
|
// -*- Mode: Go; indent-tabs-mode: t -*-
/*
* Copyright (C) 2016 Canonical Ltd
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 3 as
* published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
*/
package client_test
import (
"errors"
"fmt"
"net/http"
"golang.org/x/xerrors"
. "gopkg.in/check.v1"
)
const (
pkgID = "chatroom.ogra"
)
func (cs *clientSuite) TestClientIconCallsEndpoint(c *C) {
_, _ = cs.cli.Icon(pkgID)
c.Assert(cs.req.Method, Equals, "GET")
c.Assert(cs.req.URL.Path, Equals, fmt.Sprintf("/v2/icons/%s/icon", pkgID))
}
func (cs *clientSuite) TestClientIconHttpError(c *C) {
cs.err = errors.New("fail")
_, err := cs.cli.Icon(pkgID)
c.Assert(err, ErrorMatches, ".*server: fail")
}
func (cs *clientSuite) TestClientIconResponseNotFound(c *C) {
cs.status = 404
_, err := cs.cli.Icon(pkgID)
c.Assert(err, ErrorMatches, `.*Not Found`)
}
func (cs *clientSuite) TestClientIconInvalidContentDisposition(c *C) {
cs.header = http.Header{"Content-Disposition": {"invalid"}}
_, err := cs.cli.Icon(pkgID)
c.Assert(err, ErrorMatches, `.*cannot determine filename`)
}
func (cs *clientSuite) TestClientIcon(c *C) {
cs.rsp = "pixels"
cs.header = http.Header{"Content-Disposition": {"attachment; filename=myicon.png"}}
icon, err := cs.cli.Icon(pkgID)
c.Assert(err, IsNil)
c.Assert(icon.Filename, Equals, "myicon.png")
c.Assert(icon.Content, DeepEquals, []byte("pixels"))
}
func (cs *clientSuite) TestClientIconErrIsWrapped(c *C) {
cs.err = errors.New("boom")
_, err := cs.cli.Icon("something")
var e xerrors.Wrapper
c.Assert(err, Implements, &e)
}
|