Function TestHTTPServiceStart

src/libraries/go/lib/pkg/core/httpservice_test.go:162–250

(t *testing.T)

}

func TestHTTPServiceStart(t *testing.T) {
	ctx := WithDefaultLogger(context.Background())

	get := func(c *http.Client, url string) (string, int) {
		resp, err := c.Get(url)
		require.NoError(t, err)
		body, err := io.ReadAll(resp.Body)
		require.NoError(t, err)
		return string(body), resp.StatusCode
	}

	{
		ctx, cancel := context.WithCancel(ctx)

		socketDir, err := os.MkdirTemp("/tmp", "nvcf")
		require.NoError(t, err)
		t.Cleanup(func() { os.RemoveAll(socketDir) })
		socketPath := filepath.Join(socketDir, "s.sock")

		s := NewHTTPService("unix://" + socketPath)
		s.AddHealthRoute(ctx)
		s.Use(NewHTTPMiddleware(ctx)...)
		s.Start(ctx)

		c := &http.Client{
			Timeout: 10 * time.Second,
			Transport: &http.Transport{
				DialContext: func(_ context.Context, _, _ string) (net.Conn, error) {
					return net.Dial("unix", socketPath)
				},
			},
		}
		body, code := get(c, "http://localhost/healthz")
		assert.Equal(t, "ok\n", body)
		assert.Equal(t, http.StatusOK, code)

		cancel()
	}

	{
		ctx, cancel := context.WithCancel(ctx)
		s := NewHTTPService("0.0.0.0:0")
		s.AddHealthRoute(ctx)
		s.Use(NewHTTPMiddleware(ctx)...)
		ln, err := s.Start(ctx)
		assert.NoError(t, err)

		t.Logf("addr: %s", ln.Addr().(*net.TCPAddr).String())
		addr := ln.Addr().(*net.TCPAddr).String()

		c := &http.Client{Timeout: 10 * time.Second}
		body, code := get(c, fmt.Sprintf("http://%s/healthz", addr))
		assert.Equal(t, "ok\n", body)
		assert.Equal(t, http.StatusOK, code)

		cancel()
	}