FranklinWH-Toolbox/service/database/mongo_test.go

package database

import "testing"

func TestParseMongoDatabaseName(t *testing.T) {
	tests := []struct {
		uri, want string
		wantErr   bool
	}{
		{"mongodb://localhost:27017/toolbox?authSource=admin", "toolbox", false},
		{"mongodb://user:pass@mongo-a:27017,mongo-b:27017/toolbox?replicaSet=rs0", "toolbox", false},
		{"mongodb://localhost:27017/?authSource=admin", "", true},
		{"not-a-uri", "", true},
	}
	for _, test := range tests {
		got, err := parseMongoDatabaseName(test.uri)
		if (err != nil) != test.wantErr || got != test.want {
			t.Fatalf("parseMongoDatabaseName(%q) = %q, %v; want %q, err=%v", test.uri, got, err, test.want, test.wantErr)
		}
	}
}