Function TestCompressorHeadings

coderd/cachecompress/compress_internal_test.go:194–262

nolint: tparallel // we want to assert the state of the cache, so run synchronously

(t *testing.T)

// nolint: tparallel // we want to assert the state of the cache, so run synchronously
func TestCompressorHeadings(t *testing.T) {
	t.Parallel()
	logger := testutil.Logger(t)
	tempDir := t.TempDir()
	cacheDir := filepath.Join(tempDir, "cache")
	err := os.MkdirAll(cacheDir, 0o700)
	require.NoError(t, err)
	srcDir := filepath.Join(tempDir, "src")
	err = os.MkdirAll(srcDir, 0o700)
	require.NoError(t, err)
	err = os.WriteFile(filepath.Join(srcDir, "file.html"), []byte("textstring"), 0o600)
	require.NoError(t, err)

	compressor := NewCompressor(logger, 5, cacheDir, http.FS(os.DirFS(srcDir)))

	ts := httptest.NewServer(compressor)
	defer ts.Close()

	tests := []struct {
		name string
		path string
	}{
		{
			name: "exists",
			path: "/file.html",
		},
		{
			name: "not found",
			path: "/missing.html",
		},
		{
			name: "not found directory",
			path: "/a_directory/",
		},
	}

	// nolint: paralleltest // we want to assert the state of the cache, so run synchronously
	for _, tc := range tests {
		t.Run(tc.name, func(t *testing.T) {
			ctx := testutil.Context(t, testutil.WaitShort)
			req := httptest.NewRequestWithContext(ctx, "GET", tc.path, nil)

			// request directly from http.FileServer as our baseline response
			respROrig := httptest.NewRecorder()
			http.FileServer(http.Dir(srcDir)).ServeHTTP(respROrig, req)
			respOrig := respROrig.Result()

			req.Header.Add("Accept-Encoding", "gzip")
			// serve twice so that we go thru cache hit and cache miss code
			for range 2 {
				respRec := httptest.NewRecorder()
				compressor.ServeHTTP(respRec, req)
				respComp := respRec.Result()

				require.Equal(t, respOrig.StatusCode, respComp.StatusCode)
				for key, values := range respOrig.Header {
					if key == "Content-Length" {
						continue // we don't get length on compressed responses