package cockroach import ( "database/sql" "fmt" "testing" "time" "github.com/kr/pretty" "github.com/micro/go-micro/store" ) func TestSQL(t *testing.T) { connection := fmt.Sprintf( "host=%s port=%d user=%s sslmode=disable dbname=%s", "localhost", 5432, "jake", "test", ) db, err := sql.Open("postgres", connection) if err != nil { t.Fatal(err) } if err := db.Ping(); err != nil { t.Skip(err) } db.Close() sqlStore := NewStore( store.Namespace("testsql"), store.Nodes(connection), ) records, err := sqlStore.List() if err != nil { t.Error(err) } else { t.Logf("%# v\n", pretty.Formatter(records)) } err = sqlStore.Write( &store.Record{ Key: "test", Value: []byte("foo"), }, ) if err != nil { t.Error(err) } err = sqlStore.Write( &store.Record{ Key: "bar", Value: []byte("baz"), }, ) if err != nil { t.Error(err) } err = sqlStore.Write( &store.Record{ Key: "qux", Value: []byte("aasad"), }, ) if err != nil { t.Error(err) } err = sqlStore.Delete("qux") if err != nil { t.Error(err) } err = sqlStore.Write(&store.Record{ Key: "test", Value: []byte("bar"), Expiry: time.Minute, }) if err != nil { t.Error(err) } records, err = sqlStore.Read("test") if err != nil { t.Error(err) } t.Logf("%# v\n", pretty.Formatter(records)) if string(records[0].Value) != "bar" { t.Error("Expected bar, got ", string(records[0].Value)) } time.Sleep(61 * time.Second) _, err = sqlStore.Read("test") switch err { case nil: t.Error("Key test should have expired") default: t.Error(err) case store.ErrNotFound: break } }