package cache import ( "strings" "testing" "time" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" "github.com/MichaelMure/git-bug/bug" "github.com/MichaelMure/git-bug/query" "github.com/MichaelMure/git-bug/repository" ) func TestCache(t *testing.T) { repo := repository.CreateGoGitTestRepo(t, false) cache, err := NewRepoCache(repo) require.NoError(t, err) // Create, set and get user identity iden1, err := cache.NewIdentity("René Descartes", "rene@descartes.fr") require.NoError(t, err) err = cache.SetUserIdentity(iden1) require.NoError(t, err) userIden, err := cache.GetUserIdentity() require.NoError(t, err) require.Equal(t, iden1.Id(), userIden.Id()) // it's possible to create two identical identities iden2, err := cache.NewIdentity("René Descartes", "rene@descartes.fr") require.NoError(t, err) // Two identical identities yield a different id require.NotEqual(t, iden1.Id(), iden2.Id()) // There is now two identities in the cache require.Len(t, cache.AllIdentityIds(), 2) require.Len(t, cache.identitiesExcerpts, 2) require.Len(t, cache.identities, 2) // Create a bug bug1, _, err := cache.NewBug("title", "message") require.NoError(t, err) // It's possible to create two identical bugs bug2, _, err := cache.NewBug("title", "message") require.NoError(t, err) // two identical bugs yield a different id require.NotEqual(t, bug1.Id(), bug2.Id()) // There is now two bugs in the cache require.Len(t, cache.AllBugsIds(), 2) require.Len(t, cache.bugExcerpts, 2) require.Len(t, cache.bugs, 2) // Resolving _, err = cache.ResolveIdentity(iden1.Id()) require.NoError(t, err) _, err = cache.ResolveIdentityExcerpt(iden1.Id()) require.NoError(t, err) _, err = cache.ResolveIdentityPrefix(iden1.Id().String()[:10]) require.NoError(t, err) _, err = cache.ResolveBug(bug1.Id()) require.NoError(t, err) _, err = cache.ResolveBugExcerpt(bug1.Id()) require.NoError(t, err) _, err = cache.ResolveBugPrefix(bug1.Id().String()[:10]) require.NoError(t, err) // Querying q, err := query.Parse("status:open author:descartes sort:edit-asc") require.NoError(t, err) res, err := cache.QueryBugs(q) require.NoError(t, err) require.Len(t, res, 2) // Close require.NoError(t, cache.Close()) require.Empty(t, cache.bugs) require.Empty(t, cache.bugExcerpts) require.Empty(t, cache.identities) require.Empty(t, cache.identitiesExcerpts) // Reload, only excerpt are loaded, but as we need to load the identities used in the bugs // to check the signatures, we also load the identity used above cache, err = NewRepoCache(repo) require.NoError(t, err) require.Empty(t, cache.bugs) require.Len(t, cache.identities, 1) require.Len(t, cache.bugExcerpts, 2) require.Len(t, cache.identitiesExcerpts, 2) // Resolving load from the disk _, err = cache.ResolveIdentity(iden1.Id()) require.NoError(t, err) _, err = cache.ResolveIdentityExcerpt(iden1.Id()) require.NoError(t, err) _, err = cache.ResolveIdentityPrefix(iden1.Id().String()[:10]) require.NoError(t, err) _, err = cache.ResolveBug(bug1.Id()) require.NoError(t, err) _, err = cache.ResolveBugExcerpt(bug1.Id()) require.NoError(t, err) _, err = cache.ResolveBugPrefix(bug1.Id().String()[:10]) require.NoError(t, err) } func TestCachePushPull(t *testing.T) { repoA, repoB, _ := repository.SetupGoGitReposAndRemote(t) cacheA, err := NewRepoCache(repoA) require.NoError(t, err) cacheB, err := NewRepoCache(repoB) require.NoError(t, err) // Create, set and get user identity reneA, err := cacheA.NewIdentity("René Descartes", "rene@descartes.fr") require.NoError(t, err) err = cacheA.SetUserIdentity(reneA) require.NoError(t, err) isaacB, err := cacheB.NewIdentity("Isaac Newton", "isaac@newton.uk") require.NoError(t, err) err = cacheB.SetUserIdentity(isaacB) require.NoError(t, err) // distribute the identity _, err = cacheA.Push("origin") require.NoError(t, err) err = cacheB.Pull("origin") require.NoError(t, err) // Create a bug in A _, _, err = cacheA.NewBug("bug1", "message") require.NoError(t, err) // A --> remote --> B _, err = cacheA.Push("origin") require.NoError(t, err) err = cacheB.Pull("origin") require.NoError(t, err) require.Len(t, cacheB.AllBugsIds(), 1) // retrieve and set identity reneB, err := cacheB.ResolveIdentity(reneA.Id()) require.NoError(t, err) err = cacheB.SetUserIdentity(reneB) require.NoError(t, err) // B --> remote --> A _, _, err = cacheB.NewBug("bug2", "message") require.NoError(t, err) _, err = cacheB.Push("origin") require.NoError(t, err) err = cacheA.Pull("origin") require.NoError(t, err) require.Len(t, cacheA.AllBugsIds(), 2) } func TestRemove(t *testing.T) { repo := repository.CreateGoGitTestRepo(t, false) remoteA := repository.CreateGoGitTestRepo(t, true) remoteB := repository.CreateGoGitTestRepo(t, true) err := repo.AddRemote("remoteA", remoteA.GetLocalRemote()) require.NoError(t, err) err = repo.AddRemote("remoteB", remoteB.GetLocalRemote()) require.NoError(t, err) repoCache, err := NewRepoCache(repo) require.NoError(t, err) rene, err := repoCache.NewIdentity("René Descartes", "rene@descartes.fr") require.NoError(t, err) err = repoCache.SetUserIdentity(rene) require.NoError(t, err) _, _, err = repoCache.NewBug("title", "message") require.NoError(t, err) // and one more for testing b1, _, err := repoCache.NewBug("title", "message") require.NoError(t, err) _, err = repoCache.Push("remoteA") require.NoError(t, err) _, err = repoCache.Push("remoteB") require.NoError(t, err) _, err = repoCache.Fetch("remoteA") require.NoError(t, err) _, err = repoCache.Fetch("remoteB") require.NoError(t, err) err = repoCache.RemoveBug(b1.Id().String()) require.NoError(t, err) assert.Equal(t, 1, len(repoCache.bugs)) assert.Equal(t, 1, len(repoCache.bugExcerpts)) _, err = repoCache.ResolveBug(b1.Id()) assert.Error(t, bug.ErrBugNotExist, err) } func TestCacheEviction(t *testing.T) { repo := repository.CreateGoGitTestRepo(t, false) repoCache, err := NewRepoCache(repo) require.NoError(t, err) repoCache.setCacheSize(2) require.Equal(t, 2, repoCache.maxLoadedBugs) require.Equal(t, 0, repoCache.loadedBugs.Len()) require.Equal(t, 0, len(repoCache.bugs)) // Generating some bugs rene, err := repoCache.NewIdentity("René Descartes", "rene@descartes.fr") require.NoError(t, err) err = repoCache.SetUserIdentity(rene) require.NoError(t, err) bug1, _, err := repoCache.NewBug("title", "message") require.NoError(t, err) checkBugPresence(t, repoCache, bug1, true) require.Equal(t, 1, repoCache.loadedBugs.Len()) require.Equal(t, 1, len(repoCache.bugs)) bug2, _, err := repoCache.NewBug("title", "message") require.NoError(t, err) checkBugPresence(t, repoCache, bug1, true) checkBugPresence(t, repoCache, bug2, true) require.Equal(t, 2, repoCache.loadedBugs.Len()) require.Equal(t, 2, len(repoCache.bugs)) // Number of bugs should not exceed max size of lruCache, oldest one should be evicted bug3, _, err := repoCache.NewBug("title", "message") require.NoError(t, err) require.Equal(t, 2, repoCache.loadedBugs.Len()) require.Equal(t, 2, len(repoCache.bugs)) checkBugPresence(t, repoCache, bug1, false) checkBugPresence(t, repoCache, bug2, true) checkBugPresence(t, repoCache, bug3, true) // Accessing bug should update position in lruCache and therefore it should not be evicted repoCache.loadedBugs.Get(bug2.Id()) oldestId, _ := repoCache.loadedBugs.GetOldest() require.Equal(t, bug3.Id(), oldestId) checkBugPresence(t, repoCache, bug1, false) checkBugPresence(t, repoCache, bug2, true) checkBugPresence(t, repoCache, bug3, true) require.Equal(t, 2, repoCache.loadedBugs.Len()) require.Equal(t, 2, len(repoCache.bugs)) } func checkBugPresence(t *testing.T, cache *RepoCache, bug *BugCache, presence bool) { id := bug.Id() require.Equal(t, presence, cache.loadedBugs.Contains(id)) b, ok := cache.bugs[id] require.Equal(t, presence, ok) if ok { require.Equal(t, bug, b) } } func TestLongDescription(t *testing.T) { // See https://github.com/MichaelMure/git-bug/issues/606 text := strings.Repeat("x", 65536) repo := repository.CreateGoGitTestRepo(t, false) backend, err := NewRepoCache(repo) require.NoError(t, err) i, err := backend.NewIdentity("René Descartes", "rene@descartes.fr") require.NoError(t, err) _, _, err = backend.NewBugRaw(i, time.Now().Unix(), text, text, nil, nil) require.NoError(t, err) }