mirror of
https://github.com/mudler/edgevpn.git
synced 2025-09-26 21:01:44 +08:00
f5ea51d82e
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.
87 lines
2.5 KiB
Go
87 lines
2.5 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"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"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"
|
|
)
|
|
|
|
var _ = Describe("File services", func() {
|
|
token := node.GenerateNewConnectionData(25).Base64()
|
|
|
|
logg := logger.New(log.LevelError)
|
|
l := node.Logger(logg)
|
|
|
|
e2, _ := node.New(
|
|
node.WithDiscoveryInterval(10*time.Second),
|
|
node.WithNetworkService(AliveNetworkService(2*time.Second, 4*time.Second, 15*time.Minute)),
|
|
node.FromBase64(true, true, token, nil, nil), node.WithStore(&blockchain.MemoryStore{}), l)
|
|
|
|
Context("File sharing", func() {
|
|
It("sends and receive files between two nodes", func() {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
fileUUID := "test"
|
|
|
|
f, err := ioutil.TempFile("", "test")
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
defer os.RemoveAll(f.Name())
|
|
|
|
ioutil.WriteFile(f.Name(), []byte("testfile"), os.ModePerm)
|
|
|
|
// First node expose a file
|
|
opts, err := ShareFile(logg, 10*time.Second, fileUUID, f.Name())
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
opts = append(opts, node.FromBase64(true, true, token, nil, nil), node.WithStore(&blockchain.MemoryStore{}), l)
|
|
e, _ := node.New(opts...)
|
|
|
|
e.Start(ctx)
|
|
e2.Start(ctx)
|
|
|
|
Eventually(func() string {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
|
|
defer cancel()
|
|
|
|
f, err := ioutil.TempFile("", "test")
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
defer os.RemoveAll(f.Name())
|
|
|
|
ll, _ := e2.Ledger()
|
|
ll1, _ := e.Ledger()
|
|
By(fmt.Sprint(ll.CurrentData(), ll.LastBlock().Index, ll1.CurrentData()))
|
|
ReceiveFile(ctx, ll, e2, logg, 2*time.Second, fileUUID, f.Name())
|
|
b, _ := ioutil.ReadFile(f.Name())
|
|
return string(b)
|
|
}, 190*time.Second, 1*time.Second).Should(Equal("testfile"))
|
|
})
|
|
})
|
|
})
|