server_test.go

// Copyright 2021 Peanutzhen. All rights reserved.
// Use of this source code is governed by a MIT style
// license that can be found in the LICENSE file.

package peanutcache

import (
	"fmt"
	"log"
	"math/rand"
	"reflect"
	"testing"
	"time"
)

func createTestSvr() (*Group, *server) {
	mysql := map[string]string{
		"Tom":  "630",
		"Jack": "589",
		"Sam":  "567",
	}

	g := NewGroup("scores", 2<<10, RetrieverFunc(
		func(key string) ([]byte, error) {
			log.Println("[Mysql] search key", key)
			if v, ok := mysql[key]; ok {
				return []byte(v), nil
			}
			return nil, fmt.Errorf("%s not exist", key)
		}))

	r := rand.New(rand.NewSource(time.Now().UnixNano()))
	// 随机一个端口 避免冲突
	port := 50000 + r.Intn(100)
	addr := fmt.Sprintf("localhost:%d", port)

	svr, err := NewServer(addr)
	if err != nil {
		log.Fatal(err)
	}
	svr.SetPeers(addr)
	g.RegisterSvr(svr)
	return g, svr
}

func TestServer_GetExistsKey(t *testing.T) {
	g, svr := createTestSvr()
	go func() {
		err := svr.Start()
		if err != nil {
			log.Fatal(err)
		}
	}()
	view, err := g.Get("Tom")
	if err != nil {
		t.Fatal(err)
	}
	if !reflect.DeepEqual(view.String(), "630") {
		t.Errorf("Tom %s(actual)/%s(ok)", view.String(), "630")
	}
	log.Printf("Tom -> %s", view.String())
	DestroyGroup(g.name)
}

func TestServer_GetUnknownKey(t *testing.T) {
	g, svr := createTestSvr()
	go func() {
		err := svr.Start()
		if err != nil {
			log.Fatal(err)
		}
	}()
	_, err := g.Get("Unknown")
	if err != nil {
		if err.Error() != "Unknown not exist" {
			t.Fatal(err)
		} else {
			t.Log(err.Error())
		}
	}
	DestroyGroup(g.name)
}