summaryrefslogtreecommitdiff
path: root/internal/cache/status_test.go
diff options
context:
space:
mode:
authorLibravatar kim <89579420+NyaaaWhatsUpDoc@users.noreply.github.com>2021-09-01 10:08:21 +0100
committerLibravatar GitHub <noreply@github.com>2021-09-01 11:08:21 +0200
commit7d193de25fbccc00923d6d791d6d4e0d2d5d498e (patch)
treeda993d91aeaccc07c2fff461067325fbe89bba73 /internal/cache/status_test.go
parentAdd SQLite support, fix un-thread-safe DB caches, small performance f… (#172) (diff)
downloadgotosocial-7d193de25fbccc00923d6d791d6d4e0d2d5d498e.tar.xz
Improve GetRemoteStatus and db.GetStatus() logic (#174)
* only fetch status parents / children if explicity requested when dereferencing Signed-off-by: kim (grufwub) <grufwub@gmail.com> * Remove recursive DB GetStatus logic, don't fetch parent unless requested Signed-off-by: kim (grufwub) <grufwub@gmail.com> * StatusCache copies status so there are no thread-safety issues with modified status objects Signed-off-by: kim (grufwub) <grufwub@gmail.com> * remove sqlite test files Signed-off-by: kim (grufwub) <grufwub@gmail.com> * fix bugs introduced by previous commit Signed-off-by: kim (grufwub) <grufwub@gmail.com> * fix not continue on error in loop Signed-off-by: kim (grufwub) <grufwub@gmail.com> * use our own RunInTx implementation (possible fix for nested tx error) Signed-off-by: kim (grufwub) <grufwub@gmail.com> * fix cast statement to work with SQLite Signed-off-by: kim (grufwub) <grufwub@gmail.com> * be less strict about valid status in cache Signed-off-by: kim (grufwub) <grufwub@gmail.com> * add cache=shared ALWAYS for SQLite db instances Signed-off-by: kim (grufwub) <grufwub@gmail.com> * Fix EnrichRemoteAccount when updating account fails Signed-off-by: kim (grufwub) <grufwub@gmail.com> * add nolint tag Signed-off-by: kim (grufwub) <grufwub@gmail.com> * ensure file: prefixes the filename in sqlite addr Signed-off-by: kim (grufwub) <grufwub@gmail.com> * add an account cache, add status author account from db Signed-off-by: kim (grufwub) <grufwub@gmail.com> * Fix incompatible SQLite query Signed-off-by: kim (grufwub) <grufwub@gmail.com> * *actually* use the new getAccount() function in accountsDB Signed-off-by: kim (grufwub) <grufwub@gmail.com> * update cache tests to use test suite Signed-off-by: kim (grufwub) <grufwub@gmail.com> * add RelationshipTestSuite, add tests for methods with changed SQL Signed-off-by: kim (grufwub) <grufwub@gmail.com>
Diffstat (limited to 'internal/cache/status_test.go')
-rw-r--r--internal/cache/status_test.go66
1 files changed, 44 insertions, 22 deletions
diff --git a/internal/cache/status_test.go b/internal/cache/status_test.go
index 10dee5bca..222961025 100644
--- a/internal/cache/status_test.go
+++ b/internal/cache/status_test.go
@@ -3,39 +3,61 @@ package cache_test
import (
"testing"
+ "github.com/stretchr/testify/suite"
"github.com/superseriousbusiness/gotosocial/internal/cache"
"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
+ "github.com/superseriousbusiness/gotosocial/testrig"
)
-func TestStatusCache(t *testing.T) {
- cache := cache.NewStatusCache()
+type StatusCacheTestSuite struct {
+ suite.Suite
+ data map[string]*gtsmodel.Status
+ cache *cache.StatusCache
+}
- // Attempt to place a status
- status := gtsmodel.Status{
- ID: "id",
- URI: "uri",
- URL: "url",
- }
- cache.Put(&status)
+func (suite *StatusCacheTestSuite) SetupSuite() {
+ suite.data = testrig.NewTestStatuses()
+}
- var ok bool
- var check *gtsmodel.Status
+func (suite *StatusCacheTestSuite) SetupTest() {
+ suite.cache = cache.NewStatusCache()
+}
- // Check we can retrieve
- check, ok = cache.GetByID(status.ID)
- if !ok || !statusIs(&status, check) {
- t.Fatal("Could not find expected status")
- }
- check, ok = cache.GetByURI(status.URI)
- if !ok || !statusIs(&status, check) {
- t.Fatal("Could not find expected status")
+func (suite *StatusCacheTestSuite) TearDownTest() {
+ suite.data = nil
+ suite.cache = nil
+}
+
+func (suite *StatusCacheTestSuite) TestStatusCache() {
+ for _, status := range suite.data {
+ // Place in the cache
+ suite.cache.Put(status)
}
- check, ok = cache.GetByURL(status.URL)
- if !ok || !statusIs(&status, check) {
- t.Fatal("Could not find expected status")
+
+ for _, status := range suite.data {
+ var ok bool
+ var check *gtsmodel.Status
+
+ // Check we can retrieve
+ check, ok = suite.cache.GetByID(status.ID)
+ if !ok && !statusIs(status, check) {
+ suite.Fail("Failed to fetch expected account with ID: %s", status.ID)
+ }
+ check, ok = suite.cache.GetByURI(status.URI)
+ if status.URI != "" && !ok && !statusIs(status, check) {
+ suite.Fail("Failed to fetch expected account with URI: %s", status.URI)
+ }
+ check, ok = suite.cache.GetByURL(status.URL)
+ if status.URL != "" && !ok && !statusIs(status, check) {
+ suite.Fail("Failed to fetch expected account with URL: %s", status.URL)
+ }
}
}
+func TestStatusCache(t *testing.T) {
+ suite.Run(t, &StatusCacheTestSuite{})
+}
+
func statusIs(status1, status2 *gtsmodel.Status) bool {
return status1.ID == status2.ID && status1.URI == status2.URI && status1.URL == status2.URL
}