package server

import (
	"testing"

	"github.com/stretchr/testify/assert"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"

	cloudv1 "github.com/tidbcloud/infra-api/api/v1"
)

func Test_generateDBAuditLogBucketURLInSharedBucket(t *testing.T) {
	cluster := &cloudv1.Cluster{
		ObjectMeta: metav1.ObjectMeta{
			Name: "cluster1",
		},
		Spec: cloudv1.ClusterSpec{
			SharedBucket: &cloudv1.SharedBucket{
				BucketURL: "s3://bucket1/tenant1/project1/cluster1/",
			},
		},
	}
	assert.Equal(t, allocateDBAuditLogBucketURLInSharedBucket(cluster), "s3://bucket1/tenant1/project1/cluster1/audit_log/")
}