Archive
/
DynamoDB
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add multiple keys to client test

master
Bel LaPointe 2019-03-15 09:33:00 -06:00
parent c5b6ad08e4
commit b7771dee17
2 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
client/client.go
View File

@ -4,6 +4,7 @@ import (
"bytes" "bytes"
"errors" "errors"
"io/ioutil" "io/ioutil"
"log"
"net" "net"
"net/http" "net/http"
"strings" "strings"
@ -57,6 +58,7 @@ func New(addrs ...string) (*Client, error) {
func (c *Client) Get(key string) ([]byte, error) { func (c *Client) Get(key string) ([]byte, error) {
addr := c.hash.LocateKey([]byte(key)).String() addr := c.hash.LocateKey([]byte(key)).String()
log.Printf("GET %s FROM %s", key, addr)
resp, err := http.Get(addr + "/" + key) resp, err := http.Get(addr + "/" + key)
if err != nil { if err != nil {
return nil, err return nil, err
@ -73,6 +75,7 @@ func (c *Client) Get(key string) ([]byte, error) {
func (c *Client) Set(key string, value []byte) error { func (c *Client) Set(key string, value []byte) error {
addr := c.hash.LocateKey([]byte(key)).String() addr := c.hash.LocateKey([]byte(key)).String()
log.Printf("SET %s FROM %s", key, addr)
r, err := http.NewRequest("PUT", addr+"/"+key, bytes.NewBuffer(value)) r, err := http.NewRequest("PUT", addr+"/"+key, bytes.NewBuffer(value))
if err != nil { if err != nil {
return err return err

8
client/client_test.go
View File

@ -1,6 +1,7 @@
package client package client
import ( import (
"fmt"
"local/dynamodb/server/config" "local/dynamodb/server/config"
"local/dynamodb/server/serve" "local/dynamodb/server/serve"
"net/http/httptest" "net/http/httptest"
@ -27,16 +28,19 @@ func TestAll(t *testing.T) {
t.Fatalf("cannot make client: %v", err) t.Fatalf("cannot make client: %v", err)
} }
if err := client.Set(validKey, []byte(validValue)); err != nil { for i := 0; i < 5; i++ {
key := fmt.Sprintf("%s-%d", validKey, i)
if err := client.Set(key, []byte(validValue)); err != nil {
t.Fatalf("cannot set with client: %v", err) t.Fatalf("cannot set with client: %v", err)
} }
if v, err := client.Get(validKey); err != nil { if v, err := client.Get(key); err != nil {
t.Fatalf("cannot get with client: %v", err) t.Fatalf("cannot get with client: %v", err)
} else if string(v) != validValue { } else if string(v) != validValue {
t.Fatalf("wrong get with client: got %q, want %q", v, validValue) t.Fatalf("wrong get with client: got %q, want %q", v, validValue)
} }
} }
}
func newServer(t *testing.T) *serve.Server { func newServer(t *testing.T) *serve.Server {
os.Setenv("DB", "MAP") os.Setenv("DB", "MAP")