From b7771dee1715d98cbef4bc3649286ebd9f84d2e6 Mon Sep 17 00:00:00 2001
From: Bel LaPointe <breel@qualtrics.com>
Date: Fri, 15 Mar 2019 09:33:00 -0600
Subject: [PATCH] Add multiple keys to client test

---
 client/client.go      |  3 +++
 client/client_test.go | 18 +++++++++++-------
 2 files changed, 14 insertions(+), 7 deletions(-)

diff --git a/client/client.go b/client/client.go
index 442c256..c24420d 100644
--- a/client/client.go
+++ b/client/client.go
@@ -4,6 +4,7 @@ import (
 	"bytes"
 	"errors"
 	"io/ioutil"
+	"log"
 	"net"
 	"net/http"
 	"strings"
@@ -57,6 +58,7 @@ func New(addrs ...string) (*Client, error) {
 
 func (c *Client) Get(key string) ([]byte, error) {
 	addr := c.hash.LocateKey([]byte(key)).String()
+	log.Printf("GET %s FROM %s", key, addr)
 	resp, err := http.Get(addr + "/" + key)
 	if err != nil {
 		return nil, err
@@ -73,6 +75,7 @@ func (c *Client) Get(key string) ([]byte, error) {
 
 func (c *Client) Set(key string, value []byte) error {
 	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))
 	if err != nil {
 		return err
diff --git a/client/client_test.go b/client/client_test.go
index a1389f6..92c17a9 100644
--- a/client/client_test.go
+++ b/client/client_test.go
@@ -1,6 +1,7 @@
 package client
 
 import (
+	"fmt"
 	"local/dynamodb/server/config"
 	"local/dynamodb/server/serve"
 	"net/http/httptest"
@@ -27,14 +28,17 @@ func TestAll(t *testing.T) {
 		t.Fatalf("cannot make client: %v", err)
 	}
 
-	if err := client.Set(validKey, []byte(validValue)); err != nil {
-		t.Fatalf("cannot set with client: %v", err)
-	}
+	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)
+		}
 
-	if v, err := client.Get(validKey); err != nil {
-		t.Fatalf("cannot get with client: %v", err)
-	} else if string(v) != validValue {
-		t.Fatalf("wrong get with client: got %q, want %q", v, validValue)
+		if v, err := client.Get(key); err != nil {
+			t.Fatalf("cannot get with client: %v", err)
+		} else if string(v) != validValue {
+			t.Fatalf("wrong get with client: got %q, want %q", v, validValue)
+		}
 	}
 }