mirror of
https://github.com/opentofu/opentofu.git
synced 2025-01-04 13:17:43 -06:00
31349a9c3a
This is part of a general effort to move all of Terraform's non-library package surface under internal in order to reinforce that these are for internal use within Terraform only. If you were previously importing packages under this prefix into an external codebase, you could pin to an earlier release tag as an interim solution until you've make a plan to achieve the same functionality some other way.
88 lines
1.9 KiB
Go
88 lines
1.9 KiB
Go
package terraform
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/hashicorp/terraform/internal/configs/configschema"
|
|
)
|
|
|
|
// MockResourceProvisioner implements ResourceProvisioner but mocks out all the
|
|
// calls for testing purposes.
|
|
type MockResourceProvisioner struct {
|
|
sync.Mutex
|
|
// Anything you want, in case you need to store extra data with the mock.
|
|
Meta interface{}
|
|
|
|
GetConfigSchemaCalled bool
|
|
GetConfigSchemaReturnSchema *configschema.Block
|
|
GetConfigSchemaReturnError error
|
|
|
|
ApplyCalled bool
|
|
ApplyOutput UIOutput
|
|
ApplyState *InstanceState
|
|
ApplyConfig *ResourceConfig
|
|
ApplyFn func(*InstanceState, *ResourceConfig) error
|
|
ApplyReturnError error
|
|
|
|
ValidateCalled bool
|
|
ValidateConfig *ResourceConfig
|
|
ValidateFn func(c *ResourceConfig) ([]string, []error)
|
|
ValidateReturnWarns []string
|
|
ValidateReturnErrors []error
|
|
|
|
StopCalled bool
|
|
StopFn func() error
|
|
StopReturnError error
|
|
}
|
|
|
|
var _ ResourceProvisioner = (*MockResourceProvisioner)(nil)
|
|
|
|
func (p *MockResourceProvisioner) GetConfigSchema() (*configschema.Block, error) {
|
|
p.GetConfigSchemaCalled = true
|
|
return p.GetConfigSchemaReturnSchema, p.GetConfigSchemaReturnError
|
|
}
|
|
|
|
func (p *MockResourceProvisioner) Validate(c *ResourceConfig) ([]string, []error) {
|
|
p.Lock()
|
|
defer p.Unlock()
|
|
|
|
p.ValidateCalled = true
|
|
p.ValidateConfig = c
|
|
if p.ValidateFn != nil {
|
|
return p.ValidateFn(c)
|
|
}
|
|
return p.ValidateReturnWarns, p.ValidateReturnErrors
|
|
}
|
|
|
|
func (p *MockResourceProvisioner) Apply(
|
|
output UIOutput,
|
|
state *InstanceState,
|
|
c *ResourceConfig) error {
|
|
p.Lock()
|
|
|
|
p.ApplyCalled = true
|
|
p.ApplyOutput = output
|
|
p.ApplyState = state
|
|
p.ApplyConfig = c
|
|
if p.ApplyFn != nil {
|
|
fn := p.ApplyFn
|
|
p.Unlock()
|
|
return fn(state, c)
|
|
}
|
|
|
|
defer p.Unlock()
|
|
return p.ApplyReturnError
|
|
}
|
|
|
|
func (p *MockResourceProvisioner) Stop() error {
|
|
p.Lock()
|
|
defer p.Unlock()
|
|
|
|
p.StopCalled = true
|
|
if p.StopFn != nil {
|
|
return p.StopFn()
|
|
}
|
|
|
|
return p.StopReturnError
|
|
}
|