105 lines
2.7 KiB
Go
105 lines
2.7 KiB
Go
|
package rabbitmq
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"reflect"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/hashicorp/vault/sdk/logical"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestBackend_ConfigConnection_DefaultUsernameTemplate(t *testing.T) {
|
||
|
|
var resp *logical.Response
|
||
|
|
var err error
|
||
|
|
config := logical.TestBackendConfig()
|
||
|
|
config.StorageView = &logical.InmemStorage{}
|
||
|
|
b := Backend()
|
||
|
|
if err = b.Setup(context.Background(), config); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
configData := map[string]interface{}{
|
||
|
|
"connection_uri": "uri",
|
||
|
|
"username": "username",
|
||
|
|
"password": "password",
|
||
|
|
"verify_connection": "false",
|
||
|
|
}
|
||
|
|
configReq := &logical.Request{
|
||
|
|
Operation: logical.UpdateOperation,
|
||
|
|
Path: "config/connection",
|
||
|
|
Storage: config.StorageView,
|
||
|
|
Data: configData,
|
||
|
|
}
|
||
|
|
resp, err = b.HandleRequest(context.Background(), configReq)
|
||
|
|
if err != nil || (resp != nil && resp.IsError()) {
|
||
|
|
t.Fatalf("bad: resp: %#v\nerr:%s", resp, err)
|
||
|
|
}
|
||
|
|
if resp != nil {
|
||
|
|
t.Fatal("expected a nil response")
|
||
|
|
}
|
||
|
|
|
||
|
|
actualConfig, err := readConfig(context.Background(), config.StorageView)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("unable to read configuration: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
expectedConfig := connectionConfig{
|
||
|
|
URI: "uri",
|
||
|
|
Username: "username",
|
||
|
|
Password: "password",
|
||
|
|
UsernameTemplate: "",
|
||
|
|
}
|
||
|
|
|
||
|
|
if !reflect.DeepEqual(actualConfig, expectedConfig) {
|
||
|
|
t.Fatalf("Expected: %#v\nActual: %#v", expectedConfig, actualConfig)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestBackend_ConfigConnection_CustomUsernameTemplate(t *testing.T) {
|
||
|
|
var resp *logical.Response
|
||
|
|
var err error
|
||
|
|
config := logical.TestBackendConfig()
|
||
|
|
config.StorageView = &logical.InmemStorage{}
|
||
|
|
b := Backend()
|
||
|
|
if err = b.Setup(context.Background(), config); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
configData := map[string]interface{}{
|
||
|
|
"connection_uri": "uri",
|
||
|
|
"username": "username",
|
||
|
|
"password": "password",
|
||
|
|
"verify_connection": "false",
|
||
|
|
"username_template": "{{ .DisplayName }}",
|
||
|
|
}
|
||
|
|
configReq := &logical.Request{
|
||
|
|
Operation: logical.UpdateOperation,
|
||
|
|
Path: "config/connection",
|
||
|
|
Storage: config.StorageView,
|
||
|
|
Data: configData,
|
||
|
|
}
|
||
|
|
resp, err = b.HandleRequest(context.Background(), configReq)
|
||
|
|
if err != nil || (resp != nil && resp.IsError()) {
|
||
|
|
t.Fatalf("bad: resp: %#v\nerr:%s", resp, err)
|
||
|
|
}
|
||
|
|
if resp != nil {
|
||
|
|
t.Fatal("expected a nil response")
|
||
|
|
}
|
||
|
|
|
||
|
|
actualConfig, err := readConfig(context.Background(), config.StorageView)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("unable to read configuration: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
expectedConfig := connectionConfig{
|
||
|
|
URI: "uri",
|
||
|
|
Username: "username",
|
||
|
|
Password: "password",
|
||
|
|
UsernameTemplate: "{{ .DisplayName }}",
|
||
|
|
}
|
||
|
|
|
||
|
|
if !reflect.DeepEqual(actualConfig, expectedConfig) {
|
||
|
|
t.Fatalf("Expected: %#v\nActual: %#v", expectedConfig, actualConfig)
|
||
|
|
}
|
||
|
|
}
|