47 lines
867 B
Go
47 lines
867 B
Go
|
package connector
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/bmizerany/assert"
|
||
|
|
"github.com/hoisie/redis"
|
||
|
|
)
|
||
|
|
|
||
|
|
func Test_key(t *testing.T) {
|
||
|
|
c := Checkpoint{
|
||
|
|
AppName: "app",
|
||
|
|
StreamName: "stream",
|
||
|
|
}
|
||
|
|
|
||
|
|
k := c.key("shard")
|
||
|
|
assert.Equal(t, k, "app:checkpoint:stream:shard")
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_CheckpointExists(t *testing.T) {
|
||
|
|
var rc redis.Client
|
||
|
|
rc.Set("app:checkpoint:stream:shard", []byte("testSeqNum"))
|
||
|
|
c := Checkpoint{
|
||
|
|
AppName: "app",
|
||
|
|
StreamName: "stream",
|
||
|
|
}
|
||
|
|
|
||
|
|
r := c.CheckpointExists("shard")
|
||
|
|
assert.Equal(t, r, true)
|
||
|
|
|
||
|
|
rc.Del("app:checkpoint:stream:shard")
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_SetCheckpoint(t *testing.T) {
|
||
|
|
var rc redis.Client
|
||
|
|
c := Checkpoint{
|
||
|
|
AppName: "app",
|
||
|
|
StreamName: "stream",
|
||
|
|
}
|
||
|
|
|
||
|
|
c.SetCheckpoint("shard", "testSeqNum")
|
||
|
|
r, _ := rc.Get("app:checkpoint:stream:shard")
|
||
|
|
assert.Equal(t, string(r), "testSeqNum")
|
||
|
|
|
||
|
|
rc.Del("app:checkpoint:stream:shard")
|
||
|
|
}
|