Update databse backend tests to use the APIClientMeta for the plugin conns

This commit is contained in:
Brian Kassouf 2017-05-03 16:34:09 -07:00
parent 9faf234869
commit cbcb8635a4
2 changed files with 27 additions and 5 deletions

View file

@ -3,6 +3,7 @@ package database
import (
"database/sql"
"fmt"
"io/ioutil"
"log"
stdhttp "net/http"
"os"
@ -10,7 +11,6 @@ import (
"sync"
"testing"
"github.com/hashicorp/vault/api"
"github.com/hashicorp/vault/builtin/logical/database/dbplugin"
"github.com/hashicorp/vault/helper/pluginutil"
"github.com/hashicorp/vault/http"
@ -109,11 +109,28 @@ func TestBackend_PluginMain(t *testing.T) {
return
}
err := postgresql.Run(&api.TLSConfig{Insecure: true})
content := []byte(vault.TestClusterCACert)
tmpfile, err := ioutil.TempFile("", "example")
if err != nil {
t.Fatal(err)
}
t.Fatal("We shouldn't get here")
defer os.Remove(tmpfile.Name()) // clean up
if _, err := tmpfile.Write(content); err != nil {
t.Fatal(err)
}
if err := tmpfile.Close(); err != nil {
t.Fatal(err)
}
args := []string{"--ca-cert=" + tmpfile.Name()}
apiClientMeta := &pluginutil.APIClientMeta{}
flags := apiClientMeta.FlagSet()
flags.Parse(args)
postgresql.Run(apiClientMeta.GetTLSConfig())
}
func TestBackend_config_connection(t *testing.T) {

View file

@ -7,7 +7,6 @@ import (
"testing"
"time"
"github.com/hashicorp/vault/api"
"github.com/hashicorp/vault/builtin/logical/database/dbplugin"
"github.com/hashicorp/vault/helper/pluginutil"
"github.com/hashicorp/vault/http"
@ -107,7 +106,13 @@ func TestPlugin_Main(t *testing.T) {
users: make(map[string][]string),
}
plugins.Serve(plugin, &api.TLSConfig{Insecure: true})
args := []string{"--tls-skip-verify=true"}
apiClientMeta := &pluginutil.APIClientMeta{}
flags := apiClientMeta.FlagSet()
flags.Parse(args)
plugins.Serve(plugin, apiClientMeta.GetTLSConfig())
}
func TestPlugin_Initialize(t *testing.T) {