aljaz/crawler_v2
1
0
Fork 0
mirror of https://github.com/aljazceru/crawler_v2.git synced 2025-12-17 23:44:27 +01:00
Code Issues Packages Projects Releases Wiki Activity

code refactor for rws

Browse Source
This commit is contained in:
pippellia-btc
2025-05-28 16:55:38 +02:00
parent 85a2eebc95
commit 364432a161
7 changed files with 780 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
pkg/redb/graph_test.go
View File

@@ -318,12 +318,12 @@ func TestPubkeys(t *testing.T) {
// ------------------------------------- HELPERS -------------------------------
func Empty() (RedisDB, error) {
return New(&redis.Options{Addr: testAddress}), nil
return RedisDB{client: redis.NewClient(&redis.Options{Addr: testAddress})}, nil
}
func OneNode() (RedisDB, error) {
db := New(&redis.Options{Addr: testAddress})
if _, err := db.AddNode(context.Background(), "0"); err != nil {
db := RedisDB{client: redis.NewClient(&redis.Options{Addr: testAddress})}
if _, err := db.AddNode(ctx, "0"); err != nil {
db.flushAll()
return RedisDB{}, err
}
@@ -332,9 +332,7 @@ func OneNode() (RedisDB, error) {
}
func Simple() (RedisDB, error) {
ctx := context.Background()
db := New(&redis.Options{Addr: testAddress})
db := RedisDB{client: redis.NewClient(&redis.Options{Addr: testAddress})}
for _, pk := range []string{"0", "1", "2"} {
if _, err := db.AddNode(ctx, pk); err != nil {
db.flushAll()