edge/pkg/module/fetch/module_test.go

package fetch

import (
	"context"
	"io/ioutil"
	"net/url"
	"testing"
	"time"

	"cdr.dev/slog"
	"forge.cadoles.com/arcad/edge/pkg/app"
	"forge.cadoles.com/arcad/edge/pkg/bus/memory"
	"forge.cadoles.com/arcad/edge/pkg/module"
	"github.com/pkg/errors"
	"gitlab.com/wpetit/goweb/logger"
)

func TestFetchModule(t *testing.T) {
	t.Parallel()

	logger.SetLevel(slog.LevelDebug)

	bus := memory.NewBus()

	server := app.NewServer(
		module.ContextModuleFactory(),
		module.ConsoleModuleFactory(),
		ModuleFactory(bus),
	)

	data, err := ioutil.ReadFile("testdata/fetch.js")
	if err != nil {
		t.Fatalf("%+v", errors.WithStack(err))
	}

	if err := server.Load("testdata/fetch.js", string(data)); err != nil {
		t.Fatalf("%+v", errors.WithStack(err))
	}

	defer server.Stop()

	ctx := context.Background()
	if err := server.Start(ctx); err != nil {
		t.Fatalf("%+v", errors.WithStack(err))
	}

	// Wait for module to startup
	time.Sleep(1 * time.Second)

	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	defer cancel()

	remoteAddr := "127.0.0.1"
	url, _ := url.Parse("http://example.com")

	rawReply, err := bus.Request(ctx, NewMessageFetchRequest(ctx, remoteAddr, url))
	if err != nil {
		t.Fatalf("%+v", errors.WithStack(err))
	}

	reply, ok := rawReply.(*MessageFetchResponse)
	if !ok {
		t.Fatalf("unexpected reply type '%T'", rawReply)
	}

	if e, g := true, reply.Allow; e != g {
		t.Errorf("reply.Allow: expected '%v', got '%v'", e, g)
	}

	url, _ = url.Parse("https://google.com")

	rawReply, err = bus.Request(ctx, NewMessageFetchRequest(ctx, remoteAddr, url))
	if err != nil {
		t.Fatalf("%+v", errors.WithStack(err))
	}

	reply, ok = rawReply.(*MessageFetchResponse)
	if !ok {
		t.Fatalf("unexpected reply type '%T'", rawReply)
	}

	if e, g := false, reply.Allow; e != g {
		t.Errorf("reply.Allow: expected '%v', got '%v'", e, g)
	}
}