mirror of
https://github.com/therootcompany/golib.git
synced 2026-04-24 20:58:00 +00:00
test: add integration tests for httpcache and gitshallow
This commit is contained in:
parent
4e8321af97
commit
344246362f
76
net/gitshallow/gitshallow_integration_test.go
Normal file
76
net/gitshallow/gitshallow_integration_test.go
Normal file
@ -0,0 +1,76 @@
|
|||||||
|
//go:build integration
|
||||||
|
|
||||||
|
package gitshallow_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/therootcompany/golib/net/gitshallow"
|
||||||
|
)
|
||||||
|
|
||||||
|
const testRepoURL = "https://github.com/bitwire-it/ipblocklist"
|
||||||
|
|
||||||
|
func testdataDir(t *testing.T) string {
|
||||||
|
t.Helper()
|
||||||
|
dir, _ := filepath.Abs(".")
|
||||||
|
for {
|
||||||
|
if _, err := os.Stat(filepath.Join(dir, "go.mod")); err == nil {
|
||||||
|
return filepath.Join(dir, "testdata")
|
||||||
|
}
|
||||||
|
parent := filepath.Dir(dir)
|
||||||
|
if parent == dir {
|
||||||
|
t.Fatal("could not find module root (go.mod)")
|
||||||
|
}
|
||||||
|
dir = parent
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestRepo_Clone(t *testing.T) {
|
||||||
|
repoDir := filepath.Join(testdataDir(t), "gitshallow_ipblocklist")
|
||||||
|
os.RemoveAll(repoDir) // start fresh
|
||||||
|
|
||||||
|
repo := gitshallow.New(testRepoURL, repoDir, 1, "")
|
||||||
|
|
||||||
|
updated, err := repo.Fetch()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Fetch (clone): %v", err)
|
||||||
|
}
|
||||||
|
if !updated {
|
||||||
|
t.Error("first Fetch: expected updated=true after fresh clone")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify expected files are present.
|
||||||
|
for _, rel := range []string{
|
||||||
|
"tables/inbound/single_ips.txt",
|
||||||
|
"tables/inbound/networks.txt",
|
||||||
|
"tables/outbound/single_ips.txt",
|
||||||
|
"tables/outbound/networks.txt",
|
||||||
|
} {
|
||||||
|
p := filepath.Join(repoDir, rel)
|
||||||
|
if info, err := os.Stat(p); err != nil {
|
||||||
|
t.Errorf("expected file missing: %s", rel)
|
||||||
|
} else {
|
||||||
|
t.Logf("%s: %d bytes", rel, info.Size())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestRepo_Pull(t *testing.T) {
|
||||||
|
repoDir := filepath.Join(testdataDir(t), "gitshallow_ipblocklist")
|
||||||
|
|
||||||
|
repo := gitshallow.New(testRepoURL, repoDir, 1, "")
|
||||||
|
|
||||||
|
// Ensure cloned first.
|
||||||
|
if _, err := repo.Fetch(); err != nil {
|
||||||
|
t.Fatalf("initial Fetch: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Pull again — already up to date, updated may be true or false.
|
||||||
|
_, err := repo.Fetch()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("second Fetch (pull): %v", err)
|
||||||
|
}
|
||||||
|
t.Log("pull completed without error")
|
||||||
|
}
|
||||||
75
net/httpcache/httpcache_integration_test.go
Normal file
75
net/httpcache/httpcache_integration_test.go
Normal file
@ -0,0 +1,75 @@
|
|||||||
|
//go:build integration
|
||||||
|
|
||||||
|
package httpcache_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/therootcompany/golib/net/httpcache"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
testURL = "https://github.com/bitwire-it/ipblocklist/raw/refs/heads/main/tables/inbound/single_ips.txt"
|
||||||
|
testFile = "httpcache_inbound.txt"
|
||||||
|
)
|
||||||
|
|
||||||
|
func testdataDir(t *testing.T) string {
|
||||||
|
t.Helper()
|
||||||
|
dir, _ := filepath.Abs(".")
|
||||||
|
for {
|
||||||
|
if _, err := os.Stat(filepath.Join(dir, "go.mod")); err == nil {
|
||||||
|
return filepath.Join(dir, "testdata")
|
||||||
|
}
|
||||||
|
parent := filepath.Dir(dir)
|
||||||
|
if parent == dir {
|
||||||
|
t.Fatal("could not find module root (go.mod)")
|
||||||
|
}
|
||||||
|
dir = parent
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCacher_Download(t *testing.T) {
|
||||||
|
path := filepath.Join(testdataDir(t), testFile)
|
||||||
|
os.Remove(path) // start fresh
|
||||||
|
|
||||||
|
c := httpcache.New(testURL, path)
|
||||||
|
|
||||||
|
updated, err := c.Fetch()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("first Fetch: %v", err)
|
||||||
|
}
|
||||||
|
if !updated {
|
||||||
|
t.Error("first Fetch: expected updated=true")
|
||||||
|
}
|
||||||
|
|
||||||
|
info, err := os.Stat(path)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("file not created: %v", err)
|
||||||
|
}
|
||||||
|
if info.Size() == 0 {
|
||||||
|
t.Error("downloaded file is empty")
|
||||||
|
}
|
||||||
|
t.Logf("downloaded %d bytes to %s", info.Size(), path)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCacher_ConditionalGet(t *testing.T) {
|
||||||
|
path := filepath.Join(testdataDir(t), testFile)
|
||||||
|
|
||||||
|
// Ensure file exists from a prior download (or download it now).
|
||||||
|
c := httpcache.New(testURL, path)
|
||||||
|
if _, err := c.Fetch(); err != nil {
|
||||||
|
t.Fatalf("initial Fetch: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Second fetch on the same Cacher should use ETag/Last-Modified.
|
||||||
|
updated, err := c.Fetch()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("second Fetch: %v", err)
|
||||||
|
}
|
||||||
|
if updated {
|
||||||
|
t.Error("second Fetch: expected updated=false (content unchanged)")
|
||||||
|
}
|
||||||
|
t.Log("conditional GET correctly returned 304 / not-modified")
|
||||||
|
}
|
||||||
Loading…
x
Reference in New Issue
Block a user