Files
edgevpn/pkg/services/services_test.go
Ettore Di Giacinto f5ea51d82e 📝 Switch to Apachev2
As EdgeVPN can be used as library and it turns out to be quite useful, change license for follow up versions
to Apache so can be used as a library easilier.
2022-08-27 21:55:21 +00:00

90 lines
2.4 KiB
Go

/*
Copyright © 2021-2022 Ettore Di Giacinto <mudler@mocaccino.org>
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package services_test
import (
"context"
"io/ioutil"
"net/http"
"time"
"github.com/ipfs/go-log"
. "github.com/onsi/ginkgo/v2"
. "github.com/onsi/gomega"
"github.com/mudler/edgevpn/pkg/blockchain"
"github.com/mudler/edgevpn/pkg/logger"
node "github.com/mudler/edgevpn/pkg/node"
. "github.com/mudler/edgevpn/pkg/services"
)
func get(url string) string {
client := &http.Client{
CheckRedirect: func(req *http.Request, via []*http.Request) error {
return http.ErrUseLastResponse
},
Timeout: 1 * time.Second,
}
resp, err := client.Get(url)
if err != nil {
return ""
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
return ""
}
return string(body)
}
var _ = Describe("Expose services", func() {
token := node.GenerateNewConnectionData().Base64()
logg := logger.New(log.LevelFatal)
l := node.Logger(logg)
serviceUUID := "test"
Context("Service sharing", func() {
PIt("expose services and can connect to them", func() {
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
opts := RegisterService(logg, 5*time.Second, serviceUUID, "142.250.184.35:80")
opts = append(opts, node.FromBase64(true, true, token, nil, nil), node.WithDiscoveryInterval(10*time.Second), node.WithStore(&blockchain.MemoryStore{}), l)
e, _ := node.New(opts...)
// First node expose a service
// redirects to google:80
e.Start(ctx)
go func() {
e2, _ := node.New(
node.WithNetworkService(ConnectNetworkService(5*time.Second, serviceUUID, "127.0.0.1:9999")),
node.WithDiscoveryInterval(10*time.Second),
node.FromBase64(true, true, token, nil, nil), node.WithStore(&blockchain.MemoryStore{}), l)
e2.Start(ctx)
}()
Eventually(func() string {
return get("http://127.0.0.1:9999")
}, 360*time.Second, 1*time.Second).Should(ContainSubstring("The document has moved"))
})
})
})