103 lines
3.6 KiB
Go
103 lines
3.6 KiB
Go
package terraform
|
|
|
|
import (
|
|
"github.com/hashicorp/terraform/internal/configs/configschema"
|
|
"github.com/hashicorp/terraform/internal/providers"
|
|
"github.com/zclconf/go-cty/cty"
|
|
)
|
|
|
|
// mockProviderWithConfigSchema is a test helper to concisely create a mock
|
|
// provider with the given schema for its own configuration.
|
|
func mockProviderWithConfigSchema(schema *configschema.Block) *MockProvider {
|
|
return &MockProvider{
|
|
GetProviderSchemaResponse: &providers.GetProviderSchemaResponse{
|
|
Provider: providers.Schema{Block: schema},
|
|
},
|
|
}
|
|
}
|
|
|
|
// mockProviderWithResourceTypeSchema is a test helper to concisely create a mock
|
|
// provider with a schema containing a single resource type.
|
|
func mockProviderWithResourceTypeSchema(name string, schema *configschema.Block) *MockProvider {
|
|
return &MockProvider{
|
|
GetProviderSchemaResponse: &providers.GetProviderSchemaResponse{
|
|
Provider: providers.Schema{
|
|
Block: &configschema.Block{
|
|
Attributes: map[string]*configschema.Attribute{
|
|
"string": {
|
|
Type: cty.String,
|
|
Optional: true,
|
|
},
|
|
"list": {
|
|
Type: cty.List(cty.String),
|
|
Optional: true,
|
|
},
|
|
"root": {
|
|
Type: cty.Map(cty.String),
|
|
Optional: true,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
ResourceTypes: map[string]providers.Schema{
|
|
name: providers.Schema{Block: schema},
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
// getProviderSchemaResponseFromProviderSchema is a test helper to convert a
|
|
// ProviderSchema to a GetProviderSchemaResponse for use when building a mock provider.
|
|
func getProviderSchemaResponseFromProviderSchema(providerSchema *ProviderSchema) *providers.GetProviderSchemaResponse {
|
|
resp := &providers.GetProviderSchemaResponse{
|
|
Provider: providers.Schema{Block: providerSchema.Provider},
|
|
ProviderMeta: providers.Schema{Block: providerSchema.ProviderMeta},
|
|
ResourceTypes: map[string]providers.Schema{},
|
|
DataSources: map[string]providers.Schema{},
|
|
}
|
|
|
|
for name, schema := range providerSchema.ResourceTypes {
|
|
resp.ResourceTypes[name] = providers.Schema{
|
|
Block: schema,
|
|
Version: int64(providerSchema.ResourceTypeSchemaVersions[name]),
|
|
}
|
|
}
|
|
|
|
for name, schema := range providerSchema.DataSources {
|
|
resp.DataSources[name] = providers.Schema{Block: schema}
|
|
}
|
|
|
|
return resp
|
|
}
|
|
|
|
// simpleMockProvider returns a MockProvider that is pre-configured
|
|
// with schema for its own config, for a resource type called "test_object" and
|
|
// for a data source also called "test_object".
|
|
//
|
|
// All three schemas have the same content as returned by function
|
|
// simpleTestSchema.
|
|
//
|
|
// For most reasonable uses the returned provider must be registered in a
|
|
// componentFactory under the name "test". Use simpleMockComponentFactory
|
|
// to obtain a pre-configured componentFactory containing the result of
|
|
// this function along with simpleMockProvisioner, both registered as "test".
|
|
//
|
|
// The returned provider has no other behaviors by default, but the caller may
|
|
// modify it in order to stub any other required functionality, or modify
|
|
// the default schema stored in the field GetSchemaReturn. Each new call to
|
|
// simpleTestProvider produces entirely new instances of all of the nested
|
|
// objects so that callers can mutate without affecting mock objects.
|
|
func simpleMockProvider() *MockProvider {
|
|
return &MockProvider{
|
|
GetProviderSchemaResponse: &providers.GetProviderSchemaResponse{
|
|
Provider: providers.Schema{Block: simpleTestSchema()},
|
|
ResourceTypes: map[string]providers.Schema{
|
|
"test_object": providers.Schema{Block: simpleTestSchema()},
|
|
},
|
|
DataSources: map[string]providers.Schema{
|
|
"test_object": providers.Schema{Block: simpleTestSchema()},
|
|
},
|
|
},
|
|
}
|
|
}
|