github.com/mhilton/juju-juju@v0.0.0-20150901100907-a94dd2c73455/environs/jujutest/metadata_test.go (about)

     1  // Copyright 2013 Canonical Ltd.
     2  // Licensed under the AGPLv3, see LICENCE file for details.
     3  
     4  package jujutest
     5  
     6  import (
     7  	"io/ioutil"
     8  	"net/http"
     9  	"net/url"
    10  
    11  	jc "github.com/juju/testing/checkers"
    12  	gc "gopkg.in/check.v1"
    13  )
    14  
    15  type metadataSuite struct{}
    16  
    17  var _ = gc.Suite(&metadataSuite{})
    18  
    19  func (s *metadataSuite) TestCannedRoundTripper(c *gc.C) {
    20  	aContent := "a-content"
    21  	vrt := NewCannedRoundTripper(map[string]string{
    22  		"a": aContent,
    23  		"b": "b-content",
    24  	}, nil)
    25  	c.Assert(vrt, gc.NotNil)
    26  	req := &http.Request{URL: &url.URL{Path: "a"}}
    27  	resp, err := vrt.RoundTrip(req)
    28  	c.Assert(err, jc.ErrorIsNil)
    29  	c.Assert(resp, gc.NotNil)
    30  	content, err := ioutil.ReadAll(resp.Body)
    31  	c.Assert(string(content), gc.Equals, aContent)
    32  	c.Assert(resp.ContentLength, gc.Equals, int64(len(aContent)))
    33  	c.Assert(resp.StatusCode, gc.Equals, http.StatusOK)
    34  	c.Assert(resp.Status, gc.Equals, "200 OK")
    35  }
    36  
    37  func (s *metadataSuite) TestCannedRoundTripperMissing(c *gc.C) {
    38  	vrt := NewCannedRoundTripper(map[string]string{"a": "a-content"}, nil)
    39  	c.Assert(vrt, gc.NotNil)
    40  	req := &http.Request{URL: &url.URL{Path: "no-such-file"}}
    41  	resp, err := vrt.RoundTrip(req)
    42  	c.Assert(err, jc.ErrorIsNil)
    43  	c.Assert(resp, gc.NotNil)
    44  	content, err := ioutil.ReadAll(resp.Body)
    45  	c.Assert(string(content), gc.Equals, "")
    46  	c.Assert(resp.ContentLength, gc.Equals, int64(0))
    47  	c.Assert(resp.StatusCode, gc.Equals, http.StatusNotFound)
    48  	c.Assert(resp.Status, gc.Equals, "404 Not Found")
    49  }