github.com/wallyworld/juju@v0.0.0-20161013125918-6cf1bc9d917a/resource/api/private/server/handler_test.go

// Copyright 2016 Canonical Ltd.
// Licensed under the AGPLv3, see LICENCE file for details.

package server_test

import (
	"bytes"
	"fmt"
	"io"
	"net/http"
	"net/url"

	"github.com/juju/errors"
	"github.com/juju/testing"
	jc "github.com/juju/testing/checkers"
	"github.com/juju/testing/filetesting"
	gc "gopkg.in/check.v1"

	"github.com/juju/juju/resource"
	"github.com/juju/juju/resource/api"
	"github.com/juju/juju/resource/api/private/server"
	"github.com/juju/juju/resource/resourcetesting"
)

var _ = gc.Suite(&LegacyHTTPHandlerSuite{})

type LegacyHTTPHandlerSuite struct {
	testing.IsolationSuite

	stub   *testing.Stub
	opener *stubResourceOpener
	deps   *stubLegacyHTTPHandlerDeps
	resp   *stubResponseWriter
}

func (s *LegacyHTTPHandlerSuite) SetUpTest(c *gc.C) {
	s.IsolationSuite.SetUpTest(c)

	s.stub = &testing.Stub{}
	s.opener = &stubResourceOpener{Stub: s.stub}
	s.deps = &stubLegacyHTTPHandlerDeps{Stub: s.stub}
	s.resp = newStubResponseWriter(s.stub)
}

func (s *LegacyHTTPHandlerSuite) TestIntegration(c *gc.C) {
	opened := resourcetesting.NewResource(c, s.stub, "spam", "a-application", "some data")
	s.opener.ReturnOpenResource = opened
	s.deps.ReturnNewResourceOpener = s.opener
	deps := server.NewLegacyHTTPHandlerDeps(s.deps)
	h := server.NewLegacyHTTPHandler(deps)
	req, err := api.NewHTTPDownloadRequest("spam")
	c.Assert(err, jc.ErrorIsNil)
	req.URL, err = url.ParseRequestURI("https://api:17018/units/eggs/1/resources/spam?:resource=spam")
	c.Assert(err, jc.ErrorIsNil)
	resp := &fakeResponseWriter{
		stubResponseWriter: s.resp,
	}

	c.Logf("%#v", opened.ReadCloser)
	h.ServeHTTP(resp, req)

	resp.checkWritten(c, "some data", http.Header{
		"Content-Type":   []string{api.ContentTypeRaw},
		"Content-Length": []string{"9"}, // len("some data")
		"Content-Sha384": []string{opened.Fingerprint.String()},
	})
}

func (s *LegacyHTTPHandlerSuite) TestNewLegacyHTTPHandler(c *gc.C) {
	h := server.NewLegacyHTTPHandler(s.deps)

	s.stub.CheckNoCalls(c)
	c.Check(h, gc.NotNil)
}

func (s *LegacyHTTPHandlerSuite) TestServeHTTPDownloadOkay(c *gc.C) {
	s.deps.ReturnNewResourceOpener = s.opener
	opened := resourcetesting.NewResource(c, s.stub, "spam", "a-application", "some data")
	s.deps.ReturnHandleDownload = opened
	h := &server.LegacyHTTPHandler{
		LegacyHTTPHandlerDeps: s.deps,
	}
	req, err := http.NewRequest("GET", "...", nil)
	c.Assert(err, jc.ErrorIsNil)

	h.ServeHTTP(s.resp, req)

	s.stub.CheckCallNames(c,
		"NewResourceOpener",
		"HandleDownload",
		"UpdateDownloadResponse",
		"WriteHeader",
		"Copy",
		"Close",
	)
	s.stub.CheckCall(c, 0, "NewResourceOpener", req)
	s.stub.CheckCall(c, 1, "HandleDownload", s.opener, req)
	s.stub.CheckCall(c, 2, "UpdateDownloadResponse", s.resp, opened.Resource)
	s.stub.CheckCall(c, 3, "WriteHeader", http.StatusOK)
	s.stub.CheckCall(c, 4, "Copy", s.resp, opened)
}

func (s *LegacyHTTPHandlerSuite) TestServeHTTPDownloadHandlerFailed(c *gc.C) {
	h := &server.LegacyHTTPHandler{
		LegacyHTTPHandlerDeps: s.deps,
	}
	failure := errors.New("<failure>")
	s.stub.SetErrors(nil, failure)
	req, err := http.NewRequest("GET", "...", nil)
	c.Assert(err, jc.ErrorIsNil)

	h.ServeHTTP(s.resp, req)

	s.stub.CheckCallNames(c,
		"NewResourceOpener",
		"HandleDownload",
		"SendHTTPError",
	)
	s.stub.CheckCall(c, 2, "SendHTTPError", s.resp, failure)
}

func (s *LegacyHTTPHandlerSuite) TestServeHTTPDownloadCopyFailed(c *gc.C) {
	s.deps.ReturnHandleDownload = resourcetesting.NewResource(c, s.stub, "spam", "a-application", "some data")
	h := &server.LegacyHTTPHandler{
		LegacyHTTPHandlerDeps: s.deps,
	}
	failure := errors.New("<failure>")
	s.stub.SetErrors(nil, nil, failure)
	req, err := http.NewRequest("GET", "...", nil)
	c.Assert(err, jc.ErrorIsNil)

	h.ServeHTTP(s.resp, req)

	s.stub.CheckCallNames(c,
		"NewResourceOpener",
		"HandleDownload",
		"UpdateDownloadResponse",
		"WriteHeader",
		"Copy",
		"Close",
	)
}

func (s *LegacyHTTPHandlerSuite) TestServeHTTPConnectFailed(c *gc.C) {
	h := &server.LegacyHTTPHandler{
		LegacyHTTPHandlerDeps: s.deps,
	}
	failure := errors.New("<failure>")
	s.stub.SetErrors(failure)
	req, err := http.NewRequest("GET", "...", nil)
	c.Assert(err, jc.ErrorIsNil)

	h.ServeHTTP(s.resp, req)

	s.stub.CheckCallNames(c,
		"NewResourceOpener",
		"SendHTTPError",
	)
	s.stub.CheckCall(c, 1, "SendHTTPError", s.resp, failure)
}

func (s *LegacyHTTPHandlerSuite) TestServeHTTPUnsupportedMethod(c *gc.C) {
	h := &server.LegacyHTTPHandler{
		LegacyHTTPHandlerDeps: s.deps,
	}
	req, err := http.NewRequest("HEAD", "...", nil)
	c.Assert(err, jc.ErrorIsNil)

	h.ServeHTTP(s.resp, req)

	s.stub.CheckCallNames(c,
		"NewResourceOpener",
		"SendHTTPError",
	)
}

type stubLegacyHTTPHandlerDeps struct {
	*testing.Stub

	ReturnNewResourceOpener resource.Opener
	ReturnHandleDownload    resource.Opened
}

func (s *stubLegacyHTTPHandlerDeps) NewResourceOpener(req *http.Request) (resource.Opener, error) {
	s.AddCall("NewResourceOpener", req)
	if err := s.NextErr(); err != nil {
		return nil, err
	}

	return s.ReturnNewResourceOpener, nil
}

func (s *stubLegacyHTTPHandlerDeps) SendHTTPError(resp http.ResponseWriter, err error) {
	s.AddCall("SendHTTPError", resp, err)
	s.NextErr() // Pop one off.
}

func (s *stubLegacyHTTPHandlerDeps) UpdateDownloadResponse(resp http.ResponseWriter, info resource.Resource) {
	s.AddCall("UpdateDownloadResponse", resp, info)
	s.NextErr() // Pop one off.
}

func (s *stubLegacyHTTPHandlerDeps) HandleDownload(opener resource.Opener, req *http.Request) (resource.Opened, error) {
	s.AddCall("HandleDownload", opener, req)
	if err := s.NextErr(); err != nil {
		return resource.Opened{}, err
	}

	return s.ReturnHandleDownload, nil
}

type stubResourceOpener struct {
	*testing.Stub

	ReturnOpenResource resource.Opened
}

func (s *stubResourceOpener) OpenResource(name string) (resource.Opened, error) {
	s.AddCall("OpenResource", name)
	if err := s.NextErr(); err != nil {
		return resource.Opened{}, err
	}

	return s.ReturnOpenResource, nil
}

func (s *stubLegacyHTTPHandlerDeps) Copy(w io.Writer, r io.Reader) error {
	s.AddCall("Copy", w, r)
	if err := s.NextErr(); err != nil {
		return err
	}

	return nil
}

type stubResponseWriter struct {
	*testing.Stub
	io.Writer
	buf *bytes.Buffer

	ReturnHeader http.Header
}

func newStubResponseWriter(stub *testing.Stub) *stubResponseWriter {
	writer, buf := filetesting.NewStubWriter(stub)
	return &stubResponseWriter{
		Stub:   stub,
		Writer: writer,
		buf:    buf,

		ReturnHeader: make(http.Header),
	}
}

func (s *stubResponseWriter) Header() http.Header {
	s.AddCall("Header")
	s.NextErr() // Pop one off.

	return s.ReturnHeader
}

func (s *stubResponseWriter) WriteHeader(code int) {
	s.AddCall("WriteHeader", code)
	s.NextErr() // Pop one off.
}

type fakeResponseWriter struct {
	*stubResponseWriter

	writeCalled   bool
	writtenHeader http.Header
}

func (f *fakeResponseWriter) checkWritten(c *gc.C, body string, header http.Header) {
	if !c.Check(f.writeCalled, jc.IsTrue) {
		return
	}
	c.Check(f.buf.String(), gc.Equals, body)
	c.Check(f.writtenHeader, jc.DeepEquals, header)
	c.Check(f.writtenHeader.Get("Content-Length"), gc.Equals, fmt.Sprint(len(body)))
}

func (f *fakeResponseWriter) WriteHeader(code int) {
	f.stubResponseWriter.WriteHeader(code)

	// See http.Header.clone() in the stdlib (net/http/header.go).
	header := make(http.Header)
	for k, vv := range f.ReturnHeader {
		vv2 := make([]string, len(vv))
		copy(vv2, vv)
		header[k] = vv2
	}
	f.writtenHeader = header
}

func (f *fakeResponseWriter) Write(data []byte) (int, error) {
	f.writeCalled = true
	if f.writtenHeader == nil {
		f.WriteHeader(http.StatusOK)
	}
	return f.stubResponseWriter.Write(data)
}