/
cdc_test.go
67 lines (57 loc) · 1.19 KB
/
cdc_test.go
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
package cdc_test
import (
"io"
"io/ioutil"
"strconv"
"testing"
"github.com/schorlet/cdc"
)
func TestCrawl(t *testing.T) {
cache, err := cdc.OpenCache("testdata")
if err != nil {
t.Fatal(err)
}
for _, url := range cache.URLs() {
entry, err := cache.OpenURL(url)
if err != nil {
t.Fatal(err)
}
if entry.URL() != url {
t.Fatalf("url: %s, want: %s", entry.URL(), url)
}
header, err := entry.Header()
if err != nil {
t.Fatalf("header: %v", err)
}
if len(header) == 0 {
t.Fatal("header is empty")
}
clength := header.Get("Content-Length")
nlength, err := strconv.ParseInt(clength, 10, 64)
if err != nil {
t.Fatal(err)
}
body, err := entry.Body()
if err != nil {
t.Fatalf("body: %v", err)
}
n, err := io.Copy(ioutil.Discard, body)
if err != nil {
t.Fatalf("discard body: %v", err)
}
_ = body.Close()
if n != nlength {
t.Fatalf("body stream-length: %d, want: %d", n, nlength)
}
}
}
func TestBadURL(t *testing.T) {
cache, err := cdc.OpenCache("testdata")
if err != nil {
t.Fatal(err)
}
_, err = cache.OpenURL("http://foo.com")
if err == nil || err != cdc.ErrNotFound {
t.Fatalf("err: %v, want: %v", err, cdc.ErrNotFound)
}
}