github.com/axw/juju@v0.0.0-20161005053422-4bd6544d08d4/worker/imagemetadataworker/package_test.go (about)

     1  // Copyright 2015 Canonical Ltd.
     2  // Licensed under the AGPLv3, see LICENCE file for details.
     3  
     4  package imagemetadataworker_test
     5  
     6  import (
     7  	"testing"
     8  
     9  	gc "gopkg.in/check.v1"
    10  
    11  	apitesting "github.com/juju/juju/api/base/testing"
    12  	"github.com/juju/juju/api/imagemetadata"
    13  	coretesting "github.com/juju/juju/testing"
    14  )
    15  
    16  func TestPackage(t *testing.T) {
    17  	gc.TestingT(t)
    18  }
    19  
    20  type baseMetadataSuite struct {
    21  	coretesting.BaseSuite
    22  	apiCalled bool
    23  }
    24  
    25  func (s *baseMetadataSuite) ImageClient(done chan struct{}) *imagemetadata.Client {
    26  	closer := apitesting.APICallerFunc(func(objType string, version int, id, request string, a, result interface{}) error {
    27  		s.apiCalled = false
    28  		if request == "UpdateFromPublishedImages" {
    29  			s.apiCalled = true
    30  			close(done)
    31  			return nil
    32  		}
    33  		return nil
    34  	})
    35  
    36  	return imagemetadata.NewClient(closer)
    37  }