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() if err := server.Start(); 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) } }