diff options
author | 2021-09-01 10:08:21 +0100 | |
---|---|---|
committer | 2021-09-01 11:08:21 +0200 | |
commit | 7d193de25fbccc00923d6d791d6d4e0d2d5d498e (patch) | |
tree | da993d91aeaccc07c2fff461067325fbe89bba73 /internal/cache/status_test.go | |
parent | Add SQLite support, fix un-thread-safe DB caches, small performance f… (#172) (diff) | |
download | gotosocial-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.go | 66 |
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 } |