Function testUpdateModelVersion

scheduler/pkg/envoy/processor/server_test.go:117–146

(g *WithT, inc *IncrementalProcessor, c client.ADSClient)

}

func testUpdateModelVersion(g *WithT, inc *IncrementalProcessor, c client.ADSClient) func(t *testing.T) {
	secondFetch := []string{"model_2_grpc", "model_2_http"}

	return func(t *testing.T) {
		ops := []func(inc *IncrementalProcessor, g *WithT){
			createTestModel("model", "server", 1, []int{0}, 2, []store.ModelReplicaState{store.Available}),
		}
		go func() {
			for _, op := range ops {
				op(inc, g)
			}
		}()

		g.Eventually(func() bool {
			for _, key := range []string{"model_2_grpc", "model_2_http"} {
				_, ok := inc.xdsCache.Clusters.Load(key)
				if ok {
					return true
				}
			}
			return false
		}).WithPolling(100 * time.Millisecond).WithTimeout(5 * time.Second).Should(BeTrue())

		result := fetch(c, g, secondFetch)

		// version 2 exists
		g.Expect(result).Should(ContainElements(permanentClusterNames))
		g.Expect(result).Should(ContainElements(secondFetch))
	}
}

func fetch(c client.ADSClient, g *WithT, expectedClusterNames []string) []string {
	actualClusterNames := make([]string, 0)