(t *testing.T)
| 79 | } |
| 80 | |
| 81 | func TestClient_IsCoderConnectRunning(t *testing.T) { |
| 82 | t.Parallel() |
| 83 | ctx := testutil.Context(t, testutil.WaitShort) |
| 84 | |
| 85 | srv := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) { |
| 86 | assert.Equal(t, "/api/v2/workspaceagents/connection", r.URL.Path) |
| 87 | httpapi.Write(ctx, rw, http.StatusOK, workspacesdk.AgentConnectionInfo{ |
| 88 | HostnameSuffix: "test", |
| 89 | }) |
| 90 | })) |
| 91 | defer srv.Close() |
| 92 | |
| 93 | apiURL, err := url.Parse(srv.URL) |
| 94 | require.NoError(t, err) |
| 95 | sdkClient := codersdk.New(apiURL) |
| 96 | client := workspacesdk.New(sdkClient) |
| 97 | |
| 98 | // Right name, right IP |
| 99 | expectedName := fmt.Sprintf(tailnet.IsCoderConnectEnabledFmtString, "test") |
| 100 | ctxResolveExpected := workspacesdk.WithTestOnlyCoderContextResolver(ctx, |
| 101 | &fakeResolver{t: t, hostMap: map[string][]net.IP{ |
| 102 | expectedName: {net.ParseIP(tsaddr.CoderServiceIPv6().String())}, |
| 103 | }}) |
| 104 | |
| 105 | result, err := client.IsCoderConnectRunning(ctxResolveExpected, workspacesdk.CoderConnectQueryOptions{}) |
| 106 | require.NoError(t, err) |
| 107 | require.True(t, result) |
| 108 | |
| 109 | // Wrong name |
| 110 | result, err = client.IsCoderConnectRunning(ctxResolveExpected, workspacesdk.CoderConnectQueryOptions{HostnameSuffix: "coder"}) |
| 111 | require.NoError(t, err) |
| 112 | require.False(t, result) |
| 113 | |
| 114 | // Not found |
| 115 | ctxResolveNotFound := workspacesdk.WithTestOnlyCoderContextResolver(ctx, |
| 116 | &fakeResolver{t: t, err: &net.DNSError{IsNotFound: true}}) |
| 117 | result, err = client.IsCoderConnectRunning(ctxResolveNotFound, workspacesdk.CoderConnectQueryOptions{}) |
| 118 | require.NoError(t, err) |
| 119 | require.False(t, result) |
| 120 | |
| 121 | // Some other error |
| 122 | ctxResolverErr := workspacesdk.WithTestOnlyCoderContextResolver(ctx, |
| 123 | &fakeResolver{t: t, err: xerrors.New("a bad thing happened")}) |
| 124 | _, err = client.IsCoderConnectRunning(ctxResolverErr, workspacesdk.CoderConnectQueryOptions{}) |
| 125 | require.Error(t, err) |
| 126 | |
| 127 | // Right name, wrong IP |
| 128 | ctxResolverWrongIP := workspacesdk.WithTestOnlyCoderContextResolver(ctx, |
| 129 | &fakeResolver{t: t, hostMap: map[string][]net.IP{ |
| 130 | expectedName: {net.ParseIP("2001::34")}, |
| 131 | }}) |
| 132 | result, err = client.IsCoderConnectRunning(ctxResolverWrongIP, workspacesdk.CoderConnectQueryOptions{}) |
| 133 | require.NoError(t, err) |
| 134 | require.False(t, result) |
| 135 | } |
| 136 | |
| 137 | type fakeResolver struct { |
| 138 | t testing.TB |
nothing calls this directly
no test coverage detected