helper/schema: Provider.Resources
This commit is contained in:
parent
eaac13dd9b
commit
afd3a7b811
|
@ -13,7 +13,7 @@ import (
|
|||
// and is recommended to be used over that.
|
||||
type Provider struct {
|
||||
Schema map[string]*Schema
|
||||
Resources map[string]*Resource
|
||||
ResourcesMap map[string]*Resource
|
||||
|
||||
ConfigureFunc ConfigureFunc
|
||||
}
|
||||
|
@ -30,7 +30,7 @@ func (p *Provider) Validate(c *terraform.ResourceConfig) ([]string, []error) {
|
|||
// proper schema.
|
||||
func (p *Provider) ValidateResource(
|
||||
t string, c *terraform.ResourceConfig) ([]string, []error) {
|
||||
r, ok := p.Resources[t]
|
||||
r, ok := p.ResourcesMap[t]
|
||||
if !ok {
|
||||
return nil, []error{fmt.Errorf(
|
||||
"Provider doesn't support resource: %s", t)}
|
||||
|
@ -62,3 +62,15 @@ func (p *Provider) Configure(c *terraform.ResourceConfig) error {
|
|||
|
||||
return p.ConfigureFunc(data)
|
||||
}
|
||||
|
||||
// Resources implementation of terraform.ResourceProvider interface.
|
||||
func (p *Provider) Resources() []terraform.ResourceType {
|
||||
result := make([]terraform.ResourceType, 0, len(p.ResourcesMap))
|
||||
for k, _ := range p.ResourcesMap {
|
||||
result = append(result, terraform.ResourceType{
|
||||
Name: k,
|
||||
})
|
||||
}
|
||||
|
||||
return result
|
||||
}
|
||||
|
|
|
@ -2,6 +2,7 @@ package schema
|
|||
|
||||
import (
|
||||
"fmt"
|
||||
"reflect"
|
||||
"testing"
|
||||
|
||||
"github.com/hashicorp/terraform/config"
|
||||
|
@ -80,6 +81,38 @@ func TestProviderConfigure(t *testing.T) {
|
|||
}
|
||||
}
|
||||
|
||||
func TestProviderResources(t *testing.T) {
|
||||
cases := []struct {
|
||||
P *Provider
|
||||
Result []terraform.ResourceType
|
||||
}{
|
||||
{
|
||||
P: &Provider{},
|
||||
Result: []terraform.ResourceType{},
|
||||
},
|
||||
|
||||
{
|
||||
P: &Provider{
|
||||
ResourcesMap: map[string]*Resource{
|
||||
"foo": nil,
|
||||
"bar": nil,
|
||||
},
|
||||
},
|
||||
Result: []terraform.ResourceType{
|
||||
terraform.ResourceType{Name: "foo"},
|
||||
terraform.ResourceType{Name: "bar"},
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
for i, tc := range cases {
|
||||
actual := tc.P.Resources()
|
||||
if !reflect.DeepEqual(actual, tc.Result) {
|
||||
t.Fatalf("%d: %#v", i, actual)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestProviderValidateResource(t *testing.T) {
|
||||
cases := []struct {
|
||||
P *Provider
|
||||
|
@ -96,7 +129,7 @@ func TestProviderValidateResource(t *testing.T) {
|
|||
|
||||
{
|
||||
P: &Provider{
|
||||
Resources: map[string]*Resource{
|
||||
ResourcesMap: map[string]*Resource{
|
||||
"foo": &Resource{},
|
||||
},
|
||||
},
|
||||
|
|
Loading…
Reference in New Issue