mirror of https://github.com/poanetwork/gecko.git
78 lines
1.3 KiB
Go
78 lines
1.3 KiB
Go
|
package nat
|
||
|
|
||
|
// go test -run 'HTTP'
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"fmt"
|
||
|
"net/http"
|
||
|
"os"
|
||
|
"os/signal"
|
||
|
"strconv"
|
||
|
"testing"
|
||
|
"time"
|
||
|
|
||
|
"github.com/ava-labs/gecko/utils/logging"
|
||
|
)
|
||
|
|
||
|
const (
|
||
|
externalPort = 9876
|
||
|
localPort = 8080
|
||
|
)
|
||
|
|
||
|
func TestHTTP(t *testing.T) {
|
||
|
config, err := logging.DefaultConfig()
|
||
|
if err != nil {
|
||
|
return
|
||
|
}
|
||
|
factory := logging.NewFactory(config)
|
||
|
defer factory.Close()
|
||
|
|
||
|
log, err := factory.Make()
|
||
|
if err != nil {
|
||
|
return
|
||
|
}
|
||
|
defer log.Stop()
|
||
|
defer log.StopOnPanic()
|
||
|
|
||
|
log.Info("Logger Initialized")
|
||
|
|
||
|
n := GetNATRouter()
|
||
|
if n == nil {
|
||
|
log.Error("Unable to get UPnP Device")
|
||
|
return
|
||
|
}
|
||
|
|
||
|
ip, err := n.ExternalIP()
|
||
|
if err != nil {
|
||
|
log.Error("Unable to get external IP: %v", err)
|
||
|
return
|
||
|
}
|
||
|
log.Info("External Address %s:%d", ip.String(), externalPort)
|
||
|
|
||
|
r := NewRouter(log, n)
|
||
|
defer r.UnmapAllPorts()
|
||
|
|
||
|
r.Map("TCP", localPort, externalPort, "AVA UPnP Test")
|
||
|
|
||
|
log.Info("Starting HTTP Service")
|
||
|
server := &http.Server{Addr: ":" + strconv.Itoa(localPort)}
|
||
|
http.HandleFunc("/", hello)
|
||
|
go func() {
|
||
|
server.ListenAndServe()
|
||
|
}()
|
||
|
|
||
|
stop := make(chan os.Signal, 1)
|
||
|
signal.Notify(stop, os.Interrupt)
|
||
|
|
||
|
<-stop
|
||
|
|
||
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||
|
defer cancel()
|
||
|
server.Shutdown(ctx)
|
||
|
}
|
||
|
|
||
|
func hello(w http.ResponseWriter, req *http.Request) {
|
||
|
fmt.Fprintf(w, "AVA UPnP Test\n")
|
||
|
}
|