Mirror/mochi
1
0
Fork 0
mirror of https://github.com/sot-tech/mochi.git synced 2026-04-26 07:30:00 -07:00
Code Issues Packages Projects Releases Wiki Activity

(done) replace redigo with go-redis

Browse Source 
* replace redis keys with RawString encoded values (delete SerializedPeer)
* merge peers got from pre-hools with store data
This commit is contained in:
Lawrence, Rendall
2022-04-14 00:55:58 +03:00
parent 1fcddf5102
commit 781fa9440f
20 changed files with 490 additions and 455 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
storage/test/storage_test_base.go
View File

@@ -171,64 +171,75 @@ func (th *testHolder) LeecherPutGraduateAnnounceDeleteAnnounce(t *testing.T) {
func (th *testHolder) CustomPutContainsLoadDelete(t *testing.T) {
for _, c := range testData {
th.st.Put("test", c.peer.String(), c.ih.RawString())
err := th.st.Put("test", storage.Entry{Key: c.peer.String(), Value: c.ih.RawString()})
require.Nil(t, err)
// check if exist in ctx we put
contains := th.st.Contains("test", c.peer.String())
contains, err := th.st.Contains("test", c.peer.String())
require.Nil(t, err)
require.True(t, contains)
// check if not exist in another ctx
contains = th.st.Contains("", c.peer.String())
contains, err = th.st.Contains("", c.peer.String())
require.Nil(t, err)
require.False(t, contains)
// check value and type in ctx we put
out := th.st.Load("test", c.peer.String())
out, err := th.st.Load("test", c.peer.String())
require.Nil(t, err)
ih, err := bittorrent.NewInfoHash(out)
require.Nil(t, err)
require.Equal(t, c.ih, ih)
// check value is nil in another ctx
dummy := th.st.Load("", c.peer.String())
dummy, err := th.st.Load("", c.peer.String())
require.Nil(t, err)
require.Nil(t, dummy)
th.st.Delete("test", c.peer.String())
err = th.st.Delete("test", c.peer.String())
require.Nil(t, err)
contains = th.st.Contains("peers", c.peer.String())
contains, err = th.st.Contains("peers", c.peer.String())
require.Nil(t, err)
require.False(t, contains)
}
}
func (th *testHolder) CustomBulkPutContainsLoadDelete(t *testing.T) {
pairs := make([]storage.Pair, 0, len(testData))
keys := make([]any, 0, len(testData))
pairs := make([]storage.Entry, 0, len(testData))
keys := make([]string, 0, len(testData))
for _, c := range testData {
key := c.peer.String()
keys = append(keys, key)
pairs = append(pairs, storage.Pair{
Left: key,
Right: c.ih.RawString(),
pairs = append(pairs, storage.Entry{
Key: key,
Value: c.ih.RawString(),
})
}
th.st.BulkPut("test", pairs...)
err := th.st.BulkPut("test", pairs...)
require.Nil(t, err)
// check if exist in ctx we put
for _, k := range keys {
contains := th.st.Contains("test", k)
contains, err := th.st.Contains("test", k)
require.Nil(t, err)
require.True(t, contains)
}
// check value and type in ctx we put
for _, p := range pairs {
out := th.st.Load("test", p.Left)
out, _ := th.st.Load("test", p.Key)
ih, err := bittorrent.NewInfoHash(out)
require.Nil(t, err)
require.Equal(t, p.Right, ih.RawString())
require.Equal(t, p.Value, ih.RawString())
}
th.st.Delete("test", keys...)
err = th.st.Delete("test", keys...)
require.Nil(t, err)
for _, k := range keys {
contains := th.st.Contains("test", k)
contains, err := th.st.Contains("test", k)
require.Nil(t, err)
require.False(t, contains)
}
}