gatekeeper_test.go

package cuckooc

import (
	"context"
	"sync"
	"testing"
)

func TestGatekeeper_IntegrationTests(t *testing.T) {
	tests := []struct {
		cmd    string
		result string
	}{
		{
			cmd:    "test new 30",
			result: "true",
		},

		{
			cmd:    "test set a b c d e",
			result: "true true true true true",
		},

		{
			cmd:    "test setu a b f g",
			result: "true true true true",
		},

		{
			cmd:    "test check a b 1 2 c d e f g",
			result: "true true false false true true true true true",
		},

		{
			cmd:    "test count",
			result: "7",
		},

		{
			cmd:    "test loadfactor",
			result: "0.2188",
		},

		{
			cmd:    "test delete a c d f z",
			result: "true true true true false",
		},

		{
			cmd:    "test count",
			result: "3",
		},

		{
			cmd:    "test loadfactor",
			result: "0.0938",
		},

		{
			cmd:    "test stop",
			result: "true",
		},
	}

	respCh := make(chan string)
	gk := NewGatekeeper(make(chan Executor))
	wg := new(sync.WaitGroup)
	ctx, cancel := context.WithCancel(context.Background())
	wg.Add(1)
	go gk.Start(ctx, Config{}, wg)
	for _, c := range tests {
		exe, err := parseCommand(c.cmd, respCh)
		if err != nil {
			t.Fatalf("unexpected error: %v", err)
		}

		gk.CMDCh <- exe
		result := <-respCh

		if result != c.result {
			t.Fatalf("expected result %s but got %s", c.result, result)
		}
	}

	cancel()
	wg.Wait()

	_, ok := gk.filters["test"]
	if ok {
		t.Fatalf("expected filter to be missing but got one")
	}
}