func TestResourceProvider_datasources(t *testing.T) { p := new(terraform.MockResourceProvider) // Create a mock provider client, _ := plugin.TestPluginRPCConn(t, pluginMap(&ServeOpts{ ProviderFunc: testProviderFixed(p), })) defer client.Close() // Request the provider raw, err := client.Dispense(ProviderPluginName) if err != nil { t.Fatalf("err: %s", err) } provider := raw.(terraform.ResourceProvider) expected := []terraform.DataSource{ {Name: "foo"}, {Name: "bar"}, } p.DataSourcesReturn = expected // DataSources result := provider.DataSources() if !p.DataSourcesCalled { t.Fatal("DataSources should be called") } if !reflect.DeepEqual(result, expected) { t.Fatalf("bad: %#v", result) } }