93 lines
2.1 KiB
Go
93 lines
2.1 KiB
Go
package cloudflare
|
|
|
|
import (
|
|
"math/rand"
|
|
"os"
|
|
"strconv"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/micro/go-micro/config/options"
|
|
"github.com/micro/go-micro/store"
|
|
)
|
|
|
|
func TestCloudflare(t *testing.T) {
|
|
apiToken, accountID := os.Getenv("CF_API_TOKEN"), os.Getenv("CF_ACCOUNT_ID")
|
|
kvID := os.Getenv("KV_NAMESPACE_ID")
|
|
if len(apiToken) == 0 || len(accountID) == 0 || len(kvID) == 0 {
|
|
t.Skip("No Cloudflare API keys available, skipping test")
|
|
}
|
|
rand.Seed(time.Now().UnixNano())
|
|
randomK := strconv.Itoa(rand.Int())
|
|
randomV := strconv.Itoa(rand.Int())
|
|
|
|
wkv, err := New(
|
|
options.WithValue("CF_API_TOKEN", apiToken),
|
|
options.WithValue("CF_ACCOUNT_ID", accountID),
|
|
options.WithValue("KV_NAMESPACE_ID", kvID),
|
|
)
|
|
|
|
if err != nil {
|
|
t.Fatal(err.Error())
|
|
}
|
|
|
|
records, err := wkv.Sync()
|
|
if err != nil {
|
|
t.Fatalf("Sync: %s\n", err.Error())
|
|
} else {
|
|
t.Log("Synced " + strconv.Itoa(len(records)) + " records")
|
|
}
|
|
|
|
err = wkv.Write(
|
|
&store.Record{
|
|
Key: randomK,
|
|
Value: []byte(randomV),
|
|
},
|
|
&store.Record{
|
|
Key: "expirationtest",
|
|
Value: []byte("This message will self destruct"),
|
|
Expiry: 75 * time.Second,
|
|
},
|
|
)
|
|
if err != nil {
|
|
t.Errorf("Write: %s", err.Error())
|
|
}
|
|
|
|
// This might be needed for cloudflare eventual consistency
|
|
time.Sleep(1 * time.Minute)
|
|
|
|
r, err := wkv.Read(randomK)
|
|
if err != nil {
|
|
t.Errorf("Read: %s\n", err.Error())
|
|
}
|
|
if len(r) != 1 {
|
|
t.Errorf("Expected to read 1 key, got %d keys\n", len(r))
|
|
}
|
|
if string(r[0].Value) != randomV {
|
|
t.Errorf("Read: expected %s, got %s\n", randomK, string(r[0].Value))
|
|
}
|
|
|
|
r, err = wkv.Read("expirationtest")
|
|
if err != nil {
|
|
t.Errorf("Read: expirationtest should still exist")
|
|
}
|
|
if r[0].Expiry == 0 {
|
|
t.Error("Expected r to have an expiry")
|
|
} else {
|
|
t.Log(r[0].Expiry)
|
|
}
|
|
|
|
time.Sleep(20 * time.Second)
|
|
r, err = wkv.Read("expirationtest")
|
|
if err == nil && len(r) != 0 {
|
|
t.Error("Read: Managed to read expirationtest, but it should have expired")
|
|
t.Log(err, r[0].Key, string(r[0].Value), r[0].Expiry, len(r))
|
|
}
|
|
|
|
err = wkv.Delete(randomK)
|
|
if err != nil {
|
|
t.Errorf("Delete: %s\n", err.Error())
|
|
}
|
|
|
|
}
|