package fetch import ( "context" "net/url" "os" "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() if testing.Verbose() { logger.SetLevel(slog.LevelDebug) } bus := memory.NewBus() server := app.NewServer( module.ContextModuleFactory(), module.ConsoleModuleFactory(), ModuleFactory(bus), ) path := "testdata/fetch.js" data, err := os.ReadFile(path) if err != nil { t.Fatalf("%+v", errors.WithStack(err)) } ctx := context.Background() if err := server.Start(ctx, path, string(data)); err != nil { t.Fatalf("%+v", errors.WithStack(err)) } defer server.Stop() // 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") reply, err := bus.Request(ctx, NewFetchRequestEnvelope(ctx, remoteAddr, url)) if err != nil { t.Fatalf("%+v", errors.WithStack(err)) } response, ok := reply.Message().(*FetchResponse) if !ok { t.Fatalf("unexpected reply message type '%T'", reply.Message()) } if e, g := true, response.Allow; e != g { t.Errorf("reply.Allow: expected '%v', got '%v'", e, g) } url, _ = url.Parse("https://google.com") reply, err = bus.Request(ctx, NewFetchRequestEnvelope(ctx, remoteAddr, url)) if err != nil { t.Fatalf("%+v", errors.WithStack(err)) } response, ok = reply.Message().(*FetchResponse) if !ok { t.Fatalf("unexpected reply message type '%T'", reply.Message()) } if e, g := false, response.Allow; e != g { t.Errorf("reply.Allow: expected '%v', got '%v'", e, g) } }