mirror of
https://github.com/gofiber/storage.git
synced 2025-10-04 00:06:35 +08:00
132 lines
2.5 KiB
Go
132 lines
2.5 KiB
Go
package clickhouse
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type TestOrBench interface {
|
|
Helper()
|
|
}
|
|
|
|
func getTestConnection(t TestOrBench) (*Storage, error) {
|
|
t.Helper()
|
|
|
|
client, err := New(Config{
|
|
Host: "127.0.0.1",
|
|
Port: 9000,
|
|
Engine: "Memory",
|
|
Table: "test_table",
|
|
Clean: true,
|
|
})
|
|
|
|
return client, err
|
|
}
|
|
|
|
func TestConnection(t *testing.T) {
|
|
_, err := getTestConnection(t)
|
|
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestSet(t *testing.T) {
|
|
client, err := getTestConnection(t)
|
|
require.NoError(t, err)
|
|
defer client.Close()
|
|
|
|
err = client.Set("somekey", []byte("somevalue"), 0)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestSetWithExp(t *testing.T) {
|
|
client, err := getTestConnection(t)
|
|
require.NoError(t, err)
|
|
defer client.Close()
|
|
|
|
err = client.Set("setsomekeywithexp", []byte("somevalue"), time.Second*1)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestGet(t *testing.T) {
|
|
client, err := getTestConnection(t)
|
|
require.NoError(t, err)
|
|
defer client.Close()
|
|
|
|
err = client.Set("somekey", []byte("somevalue"), 0)
|
|
require.NoError(t, err)
|
|
|
|
value, err := client.Get("somekey")
|
|
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, value)
|
|
assert.Equal(t, "somevalue", string(value))
|
|
}
|
|
|
|
func TestGetWithExp(t *testing.T) {
|
|
client, err := getTestConnection(t)
|
|
require.NoError(t, err)
|
|
defer client.Close()
|
|
|
|
err = client.Set("getsomekeywithexp", []byte("somevalue"), time.Second*5)
|
|
require.NoError(t, err)
|
|
|
|
value, err := client.Get("getsomekeywithexp")
|
|
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, value)
|
|
assert.Equal(t, "somevalue", string(value))
|
|
|
|
time.Sleep(time.Second * 10)
|
|
|
|
value, err = client.Get("getsomekeywithexp")
|
|
|
|
require.NoError(t, err)
|
|
assert.Nil(t, value)
|
|
}
|
|
|
|
func TestDelete(t *testing.T) {
|
|
client, err := getTestConnection(t)
|
|
require.NoError(t, err)
|
|
defer client.Close()
|
|
|
|
err = client.Set("somekeytodelete", []byte("somevalue"), time.Second*5)
|
|
require.NoError(t, err)
|
|
|
|
err = client.Delete("somekeytodelete")
|
|
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestReset(t *testing.T) {
|
|
client, err := getTestConnection(t)
|
|
require.NoError(t, err)
|
|
defer client.Close()
|
|
|
|
err = client.Set("testkey", []byte("somevalue"), 0)
|
|
require.NoError(t, err)
|
|
|
|
err = client.Reset()
|
|
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func BenchmarkClickhouse(b *testing.B) {
|
|
b.ReportAllocs()
|
|
b.ResetTimer()
|
|
|
|
client, err := getTestConnection(b)
|
|
|
|
require.NoError(b, err)
|
|
defer client.Close()
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
_ = client.Set("john", []byte("doe"), 0)
|
|
err = client.Delete("john")
|
|
}
|
|
|
|
require.NoError(b, err)
|
|
}
|