76 lines
1.9 KiB
Go
76 lines
1.9 KiB
Go
|
package storage
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"github.com/spf13/viper"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestEtcdGroupCacheHandler(t *testing.T) {
|
||
|
|
cfg := createConfig()
|
||
|
|
handler, err := NewEtcdGroupCacheHandler(cfg)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("failed to create EtcdGroupCacheHandler: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
key := "test-key"
|
||
|
|
value := map[string]string{"key1": "value1", "key2": "value2"}
|
||
|
|
|
||
|
|
// test Put and Get
|
||
|
|
err = handler.Put(key, value)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("failed to put key: %v", err)
|
||
|
|
}
|
||
|
|
resp, err := handler.Get(key)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("failed to get key: %v", err)
|
||
|
|
}
|
||
|
|
var result map[string]string
|
||
|
|
err = json.Unmarshal(resp.([]byte), &result)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("failed to unmarshal response: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(result) != len(value) || result["key1"] != value["key1"] || result["key2"] != value["key2"] {
|
||
|
|
t.Fatalf("unexpected response: %v", result)
|
||
|
|
}
|
||
|
|
|
||
|
|
// test PutWithTTL and Get
|
||
|
|
err = handler.PutWithTTL(key, value, time.Second*2)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("failed to put key with TTL: %v", err)
|
||
|
|
}
|
||
|
|
resp, err = handler.Get(key)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("failed to get key: %v", err)
|
||
|
|
}
|
||
|
|
err = json.Unmarshal(resp.([]byte), &result)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("failed to unmarshal response: %v", err)
|
||
|
|
}
|
||
|
|
if len(result) != len(value) || result["key1"] != value["key1"] || result["key2"] != value["key2"] {
|
||
|
|
t.Fatalf("unexpected response: %v", result)
|
||
|
|
}
|
||
|
|
|
||
|
|
// test Delete
|
||
|
|
err = handler.Delete(key)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("failed to delete key: %v", err)
|
||
|
|
}
|
||
|
|
resp, err = handler.Get(key)
|
||
|
|
if err == nil {
|
||
|
|
t.Fatalf("expected key to be deleted")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func createConfig() *viper.Viper {
|
||
|
|
cfg := viper.New()
|
||
|
|
cfg.Set("storage.kv.etcd.endpoints", []string{"localhost:2379"})
|
||
|
|
cfg.Set("storage.kv.etcd.username", "")
|
||
|
|
cfg.Set("storage.kv.etcd.prefix", "parcoursmob/cache/")
|
||
|
|
cfg.Set("storage.kv.etcd.password", "")
|
||
|
|
cfg.Set("storage.kv.groupcache.endpoint", "localhost:8080")
|
||
|
|
return cfg
|
||
|
|
}
|