83 lines
1.7 KiB
Go
83 lines
1.7 KiB
Go
package cache
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/makeworld-the-better-one/amfora/structs"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
var p = structs.Page{Url: "example.com"}
|
|
var p2 = structs.Page{Url: "example.org"}
|
|
var queryPage = structs.Page{Url: "gemini://example.com/test?query"}
|
|
|
|
func reset() {
|
|
Clear()
|
|
SetMaxPages(0)
|
|
SetMaxSize(0)
|
|
}
|
|
|
|
func TestMaxPages(t *testing.T) {
|
|
reset()
|
|
SetMaxPages(1)
|
|
Add(&p)
|
|
Add(&p2)
|
|
assert.Equal(t, 1, NumPages(), "there should only be one page")
|
|
}
|
|
|
|
func TestMaxSize(t *testing.T) {
|
|
reset()
|
|
assert := assert.New(t)
|
|
SetMaxSize(p.Size())
|
|
Add(&p)
|
|
assert.Equal(1, NumPages(), "one page should be added")
|
|
Add(&p2)
|
|
assert.Equal(1, NumPages(), "there should still be just one page due to cache size limits")
|
|
assert.Equal(p2.Url, urls[0], "the only page url should be the second page one")
|
|
}
|
|
|
|
func TestRemove(t *testing.T) {
|
|
reset()
|
|
Add(&p)
|
|
Remove(p.Url)
|
|
assert.Equal(t, 0, NumPages(), "there shouldn't be any pages after the removal")
|
|
}
|
|
|
|
func TestClearAndNumPages(t *testing.T) {
|
|
reset()
|
|
Add(&p)
|
|
Clear()
|
|
assert.Equal(t, 0, len(pages), "map should be empty")
|
|
assert.Equal(t, 0, len(urls), "urls slice shoulde be empty")
|
|
assert.Equal(t, 0, NumPages(), "NumPages should report empty too")
|
|
}
|
|
|
|
func TestSize(t *testing.T) {
|
|
reset()
|
|
Add(&p)
|
|
assert.Equal(t, p.Size(), Size(), "sizes should match")
|
|
}
|
|
|
|
func TestGet(t *testing.T) {
|
|
reset()
|
|
Add(&p)
|
|
Add(&p2)
|
|
page, ok := Get(p.Url)
|
|
if !ok {
|
|
t.Fatal("Get should say that the page was found")
|
|
}
|
|
if page.Url != p.Url {
|
|
t.Error("page urls don't match")
|
|
}
|
|
}
|
|
|
|
func TestQueryString(t *testing.T) {
|
|
// Pages with URLs with query strings don't get added
|
|
reset()
|
|
Add(&queryPage)
|
|
_, ok := Get(queryPage.Url)
|
|
if ok {
|
|
t.Fatal("Get should not find the page, because it had query string")
|
|
}
|
|
}
|