github.com/niedbalski/juju@v0.0.0-20190215020005-8ff100488e47/resource/context/internal/base_test.go (about)

     1  // Copyright 2016 Canonical Ltd.
     2  // Licensed under the AGPLv3, see LICENCE file for details.
     3  
     4  package internal_test
     5  
     6  import (
     7  	"io"
     8  	"time"
     9  
    10  	"github.com/juju/testing"
    11  	gc "gopkg.in/check.v1"
    12  	charmresource "gopkg.in/juju/charm.v6/resource"
    13  
    14  	"github.com/juju/juju/resource"
    15  	"github.com/juju/juju/resource/resourcetesting"
    16  )
    17  
    18  func newCharmResource(c *gc.C, stub *testing.Stub, name, content string, resType charmresource.Type) (resource.Resource, io.ReadCloser) {
    19  	opened := resourcetesting.NewResource(c, stub, name, "a-application", content)
    20  	res := opened.Resource
    21  	res.Type = resType
    22  	if content != "" {
    23  		return res, opened.ReadCloser
    24  	}
    25  	res.Username = ""
    26  	res.Timestamp = time.Time{}
    27  	return res, nil
    28  }
    29  
    30  func newResource(c *gc.C, stub *testing.Stub, name, content string) (resource.Resource, io.ReadCloser) {
    31  	return newCharmResource(c, stub, name, content, charmresource.TypeFile)
    32  }
    33  
    34  func newDockerResource(c *gc.C, stub *testing.Stub, name, content string) (resource.Resource, io.ReadCloser) {
    35  	return newCharmResource(c, stub, name, content, charmresource.TypeContainerImage)
    36  }