ipinfo-csv-lite/release_test.go at main · Alice39s/ipinfo-csv-lite · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
package main

import (
	"bytes"
	"io"
	"os"
	"path/filepath"
	"testing"

	"github.com/ulikunitz/xz"
)

func TestCompressXZFileRoundTrip(t *testing.T) {
	dir := t.TempDir()
	src := filepath.Join(dir, "input.csv")
	dst := filepath.Join(dir, "output.csv.xz")
	want := bytes.Repeat([]byte("1.0.0.0/24,AU,OC,13335,Cloudflare\n"), 1000)
	if err := os.WriteFile(src, want, 0o644); err != nil {
		t.Fatal(err)
	}
	if err := compressXZFile(src, dst, 4); err != nil {
		t.Fatal(err)
	}

	f, err := os.Open(dst)
	if err != nil {
		t.Fatal(err)
	}
	defer f.Close()
	r, err := xz.NewReader(f)
	if err != nil {
		t.Fatal(err)
	}
	got, err := io.ReadAll(r)
	if err != nil {
		t.Fatal(err)
	}
	if !bytes.Equal(got, want) {
		t.Fatalf("decompressed content differs: got %d bytes, want %d", len(got), len(want))
	}
}

func TestCompressXZFilePureGoRoundTrip(t *testing.T) {
	dir := t.TempDir()
	src := filepath.Join(dir, "input.csv")
	dst := filepath.Join(dir, "output.csv.xz")
	want := bytes.Repeat([]byte("2001:db8::/32,ZZ,XX,0,Example\n"), 1000)
	if err := os.WriteFile(src, want, 0o644); err != nil {
		t.Fatal(err)
	}
	config := xz.WriterConfig{DictCap: xzDictionarySize}
	if err := compressXZFileWithConfig(src, dst, 4, config); err != nil {
		t.Fatal(err)
	}

	f, err := os.Open(dst)
	if err != nil {
		t.Fatal(err)
	}
	defer f.Close()
	r, err := xz.NewReader(f)
	if err != nil {
		t.Fatal(err)
	}
	got, err := io.ReadAll(r)
	if err != nil {
		t.Fatal(err)
	}
	if !bytes.Equal(got, want) {
		t.Fatalf("decompressed content differs: got %d bytes, want %d", len(got), len(want))
	}
}

func TestCompressXZFileRejectsInvalidStreamCount(t *testing.T) {
	if err := compressXZFile("unused", "unused", 0); err == nil {
		t.Fatal("expected invalid stream count error")
	}
}