server_test.go

package main

import (
	"context"
	"fmt"
	"goredis/client"
	"log"
	"sync"
	"testing"
	"time"
)

func TestFooBar(t *testing.T) {
	in := map[string]string{
		"first": "1",
		"second": "2",
	}
	out := respWriteMap(in)

	fmt.Println(out)

}

func TestServerWithClients(t *testing.T) {
	server := NewServer(Config{})
	go func() {
		log.Fatal(server.Start())
	}()

	time.Sleep(time.Second)

	nClinets := 10
	wg := sync.WaitGroup{}
	wg.Add(nClinets)
	for i := 0; i < nClinets; i++ {
		go func(i int) {
			c, err := client.New("localhost:5001")
			if err != nil {
				log.Fatal(err)
			}

			defer c.Close()
			key := fmt.Sprintf("client_foo_%d", i)
			value := fmt.Sprintf("client_bar_%d", i)
			if err := c.Set(context.TODO(), key, value); err != nil {
				log.Fatal(err)
			}

			val, err := c.Get(context.TODO(), key)
			if err != nil {
				log.Fatal(err)
			}

			fmt.Printf("client %d go this value back: %s\n", i, val)
			wg.Done()
		}(i)
	}

	wg.Wait()

	time.Sleep(time.Second)
	if len(server.peers) != 0 {
		t.Fatalf("expected 0 peers but got %d", len(server.peers))
	}

}