github.com/wallyworld/juju@v0.0.0-20161013125918-6cf1bc9d917a/resource/context/cmd/stub_test.go (about)

     1  // Copyright 2016 Canonical Ltd.
     2  // Licensed under the AGPLv3, see LICENCE file for details.
     3  
     4  package cmd
     5  
     6  import (
     7  	"github.com/juju/errors"
     8  	"github.com/juju/testing"
     9  )
    10  
    11  type stubHookContext struct {
    12  	stub *testing.Stub
    13  
    14  	ReturnDownload string
    15  }
    16  
    17  func (s *stubHookContext) Download(name string) (string, error) {
    18  	s.stub.AddCall("Download", name)
    19  	if err := s.stub.NextErr(); err != nil {
    20  		return "", errors.Trace(err)
    21  	}
    22  
    23  	return s.ReturnDownload, nil
    24  }
    25  
    26  func (s *stubHookContext) Flush() error { return nil }