mirror of
https://github.com/hanzoai/deploy.git
synced 2026-10-07 08:44:50 +00:00
zeekayandHanzo Dev
1ab6e799c1
go.mod still declared `module github.com/hanzoai/deploy` after the repo was renamed. A Go module path is not cosmetic — it is the resolvable location. So `go get github.com/hanzoai/cd` could not work, and anything importing this module had to spell a name that no longer exists at that address. The build log of v3.6.0 is where it surfaced: the compiler was still emitting github.com/hanzoai/deploy/cmd/cd/commands. 3541 import paths across 690 .go, 19 .proto, 4 .yaml, 3 .sh, 2 .mod, Procfile and Makefile. Plus the 62 ghcr.io/hanzoai/deploy image references, moved to ghcr.io/hanzoai/cd to match hanzo.yml's build target — a distinct string from the module path, replaced separately so neither could clobber the other. go mod tidy clean, CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build ./... clean. Co-authored-by: Hanzo Dev <[email protected]>
86 lines
3.3 KiB
Go
86 lines
3.3 KiB
Go
package sharding
|
|
|
|
import (
|
|
"fmt"
|
|
"math"
|
|
"strconv"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/mock"
|
|
|
|
"github.com/hanzoai/cd/common"
|
|
"github.com/hanzoai/cd/pkg/apis/application/v1alpha1"
|
|
dbmocks "github.com/hanzoai/cd/util/db/mocks"
|
|
)
|
|
|
|
func TestLargeShuffle(t *testing.T) {
|
|
t.Skip()
|
|
db := &dbmocks.DB{}
|
|
clusterList := &v1alpha1.ClusterList{Items: []v1alpha1.Cluster{}}
|
|
for i := 0; i < math.MaxInt/4096; i += 256 {
|
|
// fmt.Fprintf(os.Stdout, "%d", i)
|
|
cluster := createCluster(fmt.Sprintf("cluster-%d", i), strconv.Itoa(i))
|
|
clusterList.Items = append(clusterList.Items, cluster)
|
|
}
|
|
db.EXPECT().ListClusters(mock.Anything).Return(clusterList, nil)
|
|
clusterAccessor := getClusterAccessor(clusterList.Items)
|
|
// Test with replicas set to 256
|
|
replicasCount := 256
|
|
t.Setenv(common.EnvControllerReplicas, strconv.Itoa(replicasCount))
|
|
distributionFunction := RoundRobinDistributionFunction(clusterAccessor, replicasCount)
|
|
for i, c := range clusterList.Items {
|
|
assert.Equal(t, i%2567, distributionFunction(&c))
|
|
}
|
|
}
|
|
|
|
func TestShuffle(t *testing.T) {
|
|
t.Skip()
|
|
db := &dbmocks.DB{}
|
|
cluster1 := createCluster("cluster1", "10")
|
|
cluster2 := createCluster("cluster2", "20")
|
|
cluster3 := createCluster("cluster3", "30")
|
|
cluster4 := createCluster("cluster4", "40")
|
|
cluster5 := createCluster("cluster5", "50")
|
|
cluster6 := createCluster("cluster6", "60")
|
|
cluster25 := createCluster("cluster6", "25")
|
|
|
|
clusterList := &v1alpha1.ClusterList{Items: []v1alpha1.Cluster{cluster1, cluster2, cluster3, cluster4, cluster5, cluster6}}
|
|
db.EXPECT().ListClusters(mock.Anything).Return(clusterList, nil)
|
|
clusterAccessor := getClusterAccessor(clusterList.Items)
|
|
// Test with replicas set to 3
|
|
t.Setenv(common.EnvControllerReplicas, "3")
|
|
replicasCount := 3
|
|
distributionFunction := RoundRobinDistributionFunction(clusterAccessor, replicasCount)
|
|
assert.Equal(t, 0, distributionFunction(nil))
|
|
assert.Equal(t, 0, distributionFunction(&cluster1))
|
|
assert.Equal(t, 1, distributionFunction(&cluster2))
|
|
assert.Equal(t, 2, distributionFunction(&cluster3))
|
|
assert.Equal(t, 0, distributionFunction(&cluster4))
|
|
assert.Equal(t, 1, distributionFunction(&cluster5))
|
|
assert.Equal(t, 2, distributionFunction(&cluster6))
|
|
|
|
// Now, we remove cluster1, it should be unassigned, and all the other should be resuffled
|
|
clusterList.Items = Remove(clusterList.Items, 0)
|
|
assert.Equal(t, -1, distributionFunction(&cluster1))
|
|
assert.Equal(t, 0, distributionFunction(&cluster2))
|
|
assert.Equal(t, 1, distributionFunction(&cluster3))
|
|
assert.Equal(t, 2, distributionFunction(&cluster4))
|
|
assert.Equal(t, 0, distributionFunction(&cluster5))
|
|
assert.Equal(t, 1, distributionFunction(&cluster6))
|
|
|
|
// Now, we add a cluster with an id=25 so it will be placed right after cluster2
|
|
clusterList.Items = append(clusterList.Items, cluster25)
|
|
assert.Equal(t, -1, distributionFunction(&cluster1))
|
|
assert.Equal(t, 0, distributionFunction(&cluster2))
|
|
assert.Equal(t, 1, distributionFunction(&cluster25))
|
|
assert.Equal(t, 2, distributionFunction(&cluster3))
|
|
assert.Equal(t, 0, distributionFunction(&cluster4))
|
|
assert.Equal(t, 1, distributionFunction(&cluster5))
|
|
assert.Equal(t, 2, distributionFunction(&cluster6))
|
|
}
|
|
|
|
func Remove(slice []v1alpha1.Cluster, s int) []v1alpha1.Cluster {
|
|
return append(slice[:s], slice[s+1:]...)
|
|
}
|