percona/percona-toolkit
1
0
Fork 0
mirror of https://github.com/percona/percona-toolkit.git synced 2025-09-20 10:55:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

PMM-8575 migrate toolkit to modules (#512)

Browse Source 
* PMM-8575 Fixed test

Fixed sandbox used by go programs, updated all tests.

* PMM-8575 Fixed test

Fixed sandbox used by go programs, updated all tests.

* Removed Gopkg

* Switched to go 1.16

* Updated go.mod

* Upgraded deps

* Updated deps
This commit is contained in:
Carlos Salguero
2021-09-10 16:08:58 -03:00
committed by GitHub
parent ad593c6de9
commit dd2b144004
22 changed files with 491 additions and 1046 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
src/go/internal/testutils/env.go
View File

@@ -38,8 +38,8 @@ const (
envMongoDBStandalonePort = "TEST_MONGODB_STANDALONE_PORT"
//
envMongoDBUser = "TEST_MONGODB_ADMIN_USERNAME"
envMongoDBPassword = "TEST_MONGODB_ADMIN_PASSWORD"
envMongoDBUser = ""
envMongoDBPassword = ""
)
var (
@@ -86,9 +86,9 @@ var (
// MongoDBMongosPort mongos port
MongoDBMongosPort = getEnvDefault(envMongoDBMongosPort, "17000")
// MongoDBUser username for all instances
MongoDBUser = getEnvDefault(envMongoDBUser, "admin")
MongoDBUser = getEnvDefault(envMongoDBUser, "")
// MongoDBPassword password for all instances
MongoDBPassword = getEnvDefault(envMongoDBPassword, "admin123456")
MongoDBPassword = getEnvDefault(envMongoDBPassword, "")
// MongoDBTimeout global connection timeout
MongoDBTimeout = time.Duration(10) * time.Second
@@ -196,11 +196,13 @@ func TestClient(ctx context.Context, port string) (*mongo.Client, error) {
ConnectTimeout: &to,
Hosts: []string{net.JoinHostPort(hostname, port)},
Direct: &direct,
Auth: &options.Credential{
}
if MongoDBUser != "" {
co.Auth = &options.Credential{
Username: MongoDBUser,
Password: MongoDBPassword,
PasswordSet: true,
},
}
}
client, err := mongo.Connect(ctx, co)
@@ -215,3 +217,27 @@ func TestClient(ctx context.Context, port string) (*mongo.Client, error) {
return client, nil
}
func TestClientOptions(port string) *options.ClientOptions {
if port == "" {
port = MongoDBShard1PrimaryPort
}
hostname := "127.0.0.1"
direct := true
to := time.Second
co := &options.ClientOptions{
ConnectTimeout: &to,
Hosts: []string{net.JoinHostPort(hostname, port)},
Direct: &direct,
}
if MongoDBUser != "" {
co.Auth = &options.Credential{
Username: MongoDBUser,
Password: MongoDBPassword,
PasswordSet: true,
}
}
return co
}