Function SampleNetwork

state/utils_test.go:11–81

(t *testing.T, numClients, numRouters int, fullyConnected bool)

)

func SampleNetwork(t *testing.T, numClients, numRouters int, fullyConnected bool) (CentralCfg, map[string]NyPrivateKey) {
	t.Helper()
	keyStore := make(map[string]NyPrivateKey)
	keyStore["dist"] = GenerateKey()
	cfg := CentralCfg{
		Dist: &DistributionCfg{
			Key: keyStore["dist"].Pubkey(),
			Repos: []string{
				"https://example.com",
				"file:example.conf",
			},
		},
	}

	clients := make([]string, numClients)

	for idx := range numClients {
		client := fmt.Sprintf("client-%d", idx)
		clients[idx] = client
		keyStore[client] = GenerateKey()
		cfg.Clients = append(cfg.Clients, ClientCfg{NodeCfg{
			Id:     NodeId(client),
			PubKey: keyStore[client].Pubkey(),
			Prefixes: []PrefixHealthWrapper{
				{
					&StaticPrefixHealth{
						Prefix: netip.MustParsePrefix(fmt.Sprintf("10.1.0.%d/32", idx)),
						Metric: 0,
					},
				},
			},
		}})
	}

	routers := make([]string, numRouters)

	for idx := range numRouters {
		router := fmt.Sprintf("router-%d", idx)
		routers[idx] = router
		keyStore[router] = GenerateKey()
		cfg.Routers = append(cfg.Routers, RouterCfg{
			NodeCfg: NodeCfg{
				Id:     NodeId(router),
				PubKey: keyStore[router].Pubkey(),
				Prefixes: []PrefixHealthWrapper{
					{
						&StaticPrefixHealth{
							Prefix: netip.MustParsePrefix(fmt.Sprintf("10.1.0.%d/32", idx)),
							Metric: 0,
						},
					},
				},
			},
			Endpoints: []*DynamicEndpoint{
				NewDynamicEndpoint(fmt.Sprintf("192.168.0.%d:25565", idx)),
			},
		})
	}