mirror of
https://github.com/replicatedhq/troubleshoot.git
synced 2026-02-14 18:29:53 +00:00
a523551da9
* feat(redactors): Run redactors on an existing support bundle Add redact subcommand to support-bundle to allow running redactors on an existing bundle to creating a new redacted bundle. The command will be launched like so support-bundle redact <redactor urls> --bundle support-bundle.tar.gz Fixes: #705
110 lines
2.5 KiB
Go
110 lines
2.5 KiB
Go
package collect
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/replicatedhq/troubleshoot/internal/testutils"
|
|
v1beta2 "github.com/replicatedhq/troubleshoot/pkg/apis/troubleshoot/v1beta2"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
testclient "k8s.io/client-go/kubernetes/fake"
|
|
)
|
|
|
|
func Test_extractServerVersion(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
info string
|
|
want string
|
|
}{
|
|
{
|
|
name: "extracts version successfully",
|
|
info: `
|
|
# Server
|
|
redis_version:7.0.5
|
|
redis_git_sha1:00000000
|
|
redis_git_dirty:0
|
|
redis_build_id:eb3578384289228a
|
|
`,
|
|
want: "7.0.5",
|
|
},
|
|
{
|
|
name: "extracts version but fails",
|
|
info: "",
|
|
want: "",
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got := extractServerVersion(tt.info)
|
|
assert.Equalf(t, tt.want, got, "extractServerVersion() = %v, want %v", got, tt.want)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestCollectRedis_createPlainTextClient(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
uri string
|
|
hasError bool
|
|
}{
|
|
{
|
|
name: "valid uri creates redis client successfully",
|
|
uri: "redis://localhost:6379",
|
|
},
|
|
{
|
|
name: "empty uri fails to create client with error",
|
|
uri: "",
|
|
hasError: true,
|
|
},
|
|
{
|
|
name: "invalid redis protocol fails to create client with error",
|
|
uri: "http://localhost:6379",
|
|
hasError: true,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
c := &CollectRedis{
|
|
Collector: &v1beta2.Database{
|
|
URI: tt.uri,
|
|
},
|
|
}
|
|
|
|
client, err := c.createClient()
|
|
assert.Equal(t, tt.hasError, err != nil)
|
|
if err == nil {
|
|
require.NotNil(t, client)
|
|
assert.Equal(t, client.Options().Addr, "localhost:6379")
|
|
} else {
|
|
t.Log(err)
|
|
assert.Nil(t, client)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestCollectRedis_createTLSClient(t *testing.T) {
|
|
k8sClient := testclient.NewSimpleClientset()
|
|
|
|
c := &CollectRedis{
|
|
Client: k8sClient,
|
|
Collector: &v1beta2.Database{
|
|
URI: "redis://localhost:6379",
|
|
TLS: &v1beta2.TLSParams{
|
|
CACert: testutils.GetTestFixture(t, "db/ca.pem"),
|
|
ClientCert: testutils.GetTestFixture(t, "db/client.pem"),
|
|
ClientKey: testutils.GetTestFixture(t, "db/client-key.pem"),
|
|
},
|
|
},
|
|
}
|
|
|
|
client, err := c.createClient()
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, client)
|
|
opt := client.Options()
|
|
assert.Equal(t, opt.Addr, "localhost:6379")
|
|
assert.NotNil(t, opt.TLSConfig.Certificates)
|
|
assert.NotNil(t, opt.TLSConfig.RootCAs)
|
|
assert.False(t, opt.TLSConfig.InsecureSkipVerify)
|
|
}
|