Function testBlock

flate/huffman_bit_writer_test.go:191–308

testBlock tests a block against its references, or regenerate the references, if "-update" flag is set.

(t *testing.T, test huffTest, ttype string)

// testBlock tests a block against its references,
// or regenerate the references, if "-update" flag is set.
func testBlock(t *testing.T, test huffTest, ttype string) {
	if test.want != "" {
		test.want = fmt.Sprintf(test.want, ttype)
	}
	const gotSuffix = ".got"
	test.wantNoInput = fmt.Sprintf(test.wantNoInput, ttype)
	tokens := indexTokens(test.tokens)
	if *update {
		if test.input != "" {
			t.Logf("Updating %q", test.want)
			input, err := os.ReadFile(test.input)
			if err != nil {
				t.Error(err)
				return
			}

			f, err := os.Create(test.want)
			if err != nil {
				t.Error(err)
				return
			}
			defer f.Close()
			bw := newHuffmanBitWriter(f)
			writeToType(t, ttype, bw, tokens, input)
		}

		t.Logf("Updating %q", test.wantNoInput)
		f, err := os.Create(test.wantNoInput)
		if err != nil {
			t.Error(err)
			return
		}
		defer f.Close()
		bw := newHuffmanBitWriter(f)
		writeToType(t, ttype, bw, tokens, nil)
		return
	}

	if test.input != "" {
		t.Logf("Testing %q", test.want)
		input, err := os.ReadFile(test.input)
		if err != nil {
			t.Error(err)
			return
		}
		want, err := os.ReadFile(test.want)
		if err != nil {
			t.Error(err)
			return
		}
		var buf bytes.Buffer
		bw := newHuffmanBitWriter(&buf)
		writeToType(t, ttype, bw, tokens, input)

		got := buf.Bytes()
		if !bytes.Equal(got, want) {
			t.Errorf("writeBlock did not yield expected result for file %q with input. See %q", test.want, test.want+gotSuffix)
			if err := os.WriteFile(test.want+gotSuffix, got, 0666); err != nil {