-
Notifications
You must be signed in to change notification settings - Fork 378
/
testing_non_mocked.go
74 lines (58 loc) · 2.13 KB
/
testing_non_mocked.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
package bertymessengertesting
import (
"flag"
"fmt"
"io/ioutil"
"os"
"testing"
"github.com/stretchr/testify/assert"
"berty.tech/berty/v2/go/internal/initutil"
"berty.tech/berty/v2/go/pkg/bertyprotocol"
"berty.tech/berty/v2/go/pkg/messengertypes"
"berty.tech/berty/v2/go/pkg/protocoltypes"
)
func NonMockedTestingInfra(t testing.TB, accountAmount int) ([]messengertypes.MessengerServiceClient, []*bertyprotocol.TestingProtocol, func()) {
messengers := make([]messengertypes.MessengerServiceClient, accountAmount)
tps := make([]*bertyprotocol.TestingProtocol, accountAmount)
closeFuncs := ([]func())(nil)
for i := 0; i < accountAmount; i++ {
tempDir, err := ioutil.TempDir("", fmt.Sprintf("berty-main-%d", i))
assert.NoError(t, err)
closeFuncs = append(closeFuncs, func() {
_ = os.RemoveAll(tempDir)
})
man, err := initutil.New(nil)
assert.NoError(t, err)
closeFuncs = append(closeFuncs, func() { _ = man.Close(nil) })
fs := flag.NewFlagSet("man1", flag.ExitOnError)
man.SetupLoggingFlags(fs) // also available at root level
man.SetupMetricsFlags(fs) // add flags to enable metrics
man.SetupLocalMessengerServerFlags(fs) // add flags to allow creating a full node in the same process
man.SetupEmptyGRPCListenersFlags(fs) // by default, we don't want to expose gRPC server for mini
man.SetupRemoteNodeFlags(fs) // mini can be run against an already running server
man.SetupInitTimeout(fs)
err = fs.Parse([]string{"-store.dir", tempDir})
assert.NoError(t, err)
ipfs, _, err := man.GetLocalIPFS()
assert.NoError(t, err)
closeFuncs = append(closeFuncs, func() { _ = ipfs.Close() })
protocolClient, err := man.GetProtocolClient()
assert.NoError(t, err)
messengers[i], err = man.GetMessengerClient()
assert.NoError(t, err)
tps[i] = &bertyprotocol.TestingProtocol{
Client: &WrappedMessengerClient{protocolClient},
}
}
return messengers, tps, func() {
for i := len(closeFuncs) - 1; i >= 0; i-- {
closeFuncs[i]()
}
}
}
type WrappedMessengerClient struct {
protocoltypes.ProtocolServiceClient
}
func (*WrappedMessengerClient) Close() error {
return nil
}