helper/schema: ability to force set Meta

Author: mitchellh

helper/schema/provider.go

@@ -58,6 +58,13 @@ func (p *Provider) Meta() interface{} {
 	return p.meta
 }
 
+// SetMeta can be used to forcefully set the Meta object of the provider.
+// Note that if Configure is called the return value will override anything
+// set here.
+func (p *Provider) SetMeta(v interface{}) {
+	p.meta = v
+}
+
 // Validate validates the provider configuration against the schema.
 func (p *Provider) Validate(c *terraform.ResourceConfig) ([]string, []error) {
 	return schemaMap(p.Schema).Validate(c)

helper/schema/provider_test.go

@@ -155,3 +155,16 @@ func TestProviderValidateResource(t *testing.T) {
 		}
 	}
 }
+
+func TestProviderMeta(t *testing.T) {
+	p := new(Provider)
+	if v := p.Meta(); v != nil {
+		t.Fatalf("bad: %#v", v)
+	}
+
+	expected := 42
+	p.SetMeta(42)
+	if v := p.Meta(); !reflect.DeepEqual(v, expected) {
+		t.Fatalf("bad: %#v")
+	}
+}