fix unittests after rebase

canonical · Sep 27, 2024 · a85147a · a85147a
1 parent 9e0a43c
commit a85147a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/k8s/pkg/k8sd/setup/k8s_apiserver_proxy_test.go b/src/k8s/pkg/k8sd/setup/k8s_apiserver_proxy_test.go
@@ -138,13 +138,13 @@ func TestK8sApiServerProxy(t *testing.T) {
 		s := mustSetupSnapAndDirectories(t, setKubeletMock)
 		s.Mock.Hostname = "dev"
 
-		g.Expect(setup.K8sAPIServerProxy(s, nil, "[2001:db8::]", nil)).To(Succeed())
+		g.Expect(setup.K8sAPIServerProxy(s, nil, 1234, nil)).To(Succeed())
 
 		tests := []struct {
 			key         string
 			expectedVal string
 		}{
-			{key: "--listen", expectedVal: "[2001:db8::]:6443"},
+			{key: "--listen", expectedVal: ":1234"},
 		}
 		for _, tc := range tests {
 			t.Run(tc.key, func(t *testing.T) {

diff --git a/src/k8s/pkg/k8sd/setup/kube_apiserver_test.go b/src/k8s/pkg/k8sd/setup/kube_apiserver_test.go
@@ -253,7 +253,7 @@ func TestKubeAPIServer(t *testing.T) {
 		s := mustSetupSnapAndDirectories(t, setKubeletMock)
 		s.Mock.Hostname = "dev"
 
-		g.Expect(setup.KubeAPIServer(s, net.ParseIP("2001:db8::"), "fd98::/108", "https://auth-webhook.url", false, types.Datastore{Type: utils.Pointer("k8s-dqlite")}, "Node,RBAC", nil)).To(Succeed())
+		g.Expect(setup.KubeAPIServer(s, 6443, net.ParseIP("2001:db8::"), "fd98::/108", "https://auth-webhook.url", false, types.Datastore{Type: utils.Pointer("k8s-dqlite")}, "Node,RBAC", nil)).To(Succeed())
 
 		tests := []struct {
 			key         string