fn-serverless/api/agent/static_pool_test.go

package agent

import (
	"context"
	"errors"
	"testing"

	pool "github.com/fnproject/fn/api/runnerpool"
)

func setupStaticPool(runners []string) pool.RunnerPool {
	return NewStaticRunnerPool(runners, nil, "", mockRunnerFactory)
}

var (
	ErrorGarbanzoBeans = errors.New("yes, that's right. Garbanzo beans...")
)

type mockStaticRunner struct {
	address string
}

func (r *mockStaticRunner) TryExec(ctx context.Context, call pool.RunnerCall) (bool, error) {
	return true, nil
}

func (r *mockStaticRunner) Close(context.Context) error {
	return ErrorGarbanzoBeans
}

func (r *mockStaticRunner) Address() string {
	return r.address
}

func mockRunnerFactory(addr, cn string, pki *pool.PKIData) (pool.Runner, error) {
	return &mockStaticRunner{address: addr}, nil
}

func TestNewStaticPool(t *testing.T) {
	addrs := []string{"127.0.0.1:8080", "127.0.0.1:8081"}
	np := setupStaticPool(addrs)

	runners, err := np.Runners(nil)
	if err != nil {
		t.Fatalf("Failed to list runners %v", err)
	}
	if len(runners) != len(addrs) {
		t.Fatalf("Invalid number of runners %v", len(runners))
	}
}

func TestEmptyPool(t *testing.T) {
	np := setupStaticPool(nil).(*staticRunnerPool)

	runners, err := np.Runners(nil)
	if err != nil {
		t.Fatalf("Failed to list runners %v", err)
	}
	if len(runners) != 0 {
		t.Fatalf("Invalid number of runners %v", len(runners))
	}

	err = np.AddRunner("127.0.0.1:8082")
	if err != nil {
		t.Fatalf("Failed to add runner %v", err)
	}

	runners, err = np.Runners(nil)
	if err != nil {
		t.Fatalf("Failed to list runners %v", err)
	}
	if len(runners) != 1 {
		t.Fatalf("Invalid number of runners %v", len(runners))
	}

	err = np.Shutdown(context.Background())
	if err != ErrorGarbanzoBeans {
		t.Fatalf("Expected garbanzo beans error from shutdown %v", err)
	}

	runners, err = np.Runners(nil)
	if err == nil {
		t.Fatalf("Should fail to list runners (shutdown)")
	}
	if len(runners) != 0 {
		t.Fatalf("Invalid number of runners %v", len(runners))
	}
}

func TestAddNodeToPool(t *testing.T) {
	addrs := []string{"127.0.0.1:8080", "127.0.0.1:8081"}
	np := setupStaticPool(addrs).(*staticRunnerPool)

	err := np.AddRunner("127.0.0.1:8082")
	if err != nil {
		t.Fatalf("Add Should not fail %v", err)
	}

	err = np.AddRunner("127.0.0.1:8082")
	if err != ErrorPoolRunnerExists {
		t.Fatalf("Add Should fail since duplicate %v", err)
	}

	runners, err := np.Runners(nil)
	if err != nil {
		t.Fatalf("Failed to list runners %v", err)
	}
	if len(runners) != 3 {
		t.Fatalf("Invalid number of runners %v", len(runners))
	}

	err = np.Shutdown(context.Background())
	if err != ErrorGarbanzoBeans {
		t.Fatalf("Expected garbanzo beans error from shutdown %v", err)
	}

	runners, err = np.Runners(nil)
	if err == nil {
		t.Fatalf("Should fail to list runners (shutdown)")
	}
	if len(runners) != 0 {
		t.Fatalf("Invalid number of runners %v", len(runners))
	}
}

func TestRemoveNodeFromPool(t *testing.T) {
	addrs := []string{"127.0.0.1:8080", "127.0.0.1:8081"}
	np := setupStaticPool(addrs).(*staticRunnerPool)

	np.RemoveRunner("127.0.0.1:8081")

	runners, err := np.Runners(nil)
	if err != nil {
		t.Fatalf("Failed to list runners %v", err)
	}

	if len(runners) != 1 {
		t.Fatalf("Invalid number of runners %v", len(runners))
	}

	np.RemoveRunner("127.0.0.1:8081")

	runners, err = np.Runners(nil)
	if err != nil {
		t.Fatalf("Failed to list runners %v", err)
	}
	if len(runners) != 1 {
		t.Fatalf("Invalid number of runners %v", len(runners))
	}

	np.RemoveRunner("127.0.0.1:8080")

	runners, err = np.Runners(nil)
	if err != nil {
		t.Fatalf("Failed to list runners %v", err)
	}
	if len(runners) != 0 {
		t.Fatalf("Invalid number of runners %v", len(runners))
	}

	np.RemoveRunner("127.0.0.1:8080")

	runners, err = np.Runners(nil)
	if err != nil {
		t.Fatalf("Failed to list runners %v", err)
	}
	if len(runners) != 0 {
		t.Fatalf("Invalid number of runners %v", len(runners))
	}

	// Let's try a double shutdown
	err = np.Shutdown(context.Background())
	if err != nil {
		t.Fatalf("Not expected error from shutdown I (empty pool) %v", err)
	}
	err = np.Shutdown(context.Background())
	if err != nil {
		t.Fatalf("Not expected error from shutdown II (empty pool) %v", err)
	}

	runners, err = np.Runners(nil)
	if err == nil {
		t.Fatalf("Should fail to list runners (shutdown)")
	}
	if len(runners) != 0 {
		t.Fatalf("Invalid number of runners %v", len(runners))
	}
}