-
Notifications
You must be signed in to change notification settings - Fork 384
/
testing_non_mocked.go
70 lines (54 loc) · 1.89 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
package bertymessengertesting
import (
"flag"
"fmt"
"os"
"testing"
"github.com/stretchr/testify/require"
"berty.tech/berty/v2/go/internal/initutil"
"berty.tech/berty/v2/go/pkg/messengertypes"
"berty.tech/weshnet"
"berty.tech/weshnet/pkg/protocoltypes"
)
func NonMockedTestingInfra(t testing.TB, accountAmount int) ([]messengertypes.MessengerServiceClient, []*weshnet.TestingProtocol) {
messengers := make([]messengertypes.MessengerServiceClient, accountAmount)
tps := make([]*weshnet.TestingProtocol, accountAmount)
for i := 0; i < accountAmount; i++ {
tempDir, err := os.MkdirTemp("", fmt.Sprintf("berty-main-%d", i))
require.NoError(t, err)
t.Cleanup(func() {
_ = os.RemoveAll(tempDir)
})
man, err := initutil.New(nil)
require.NoError(t, err)
t.Cleanup(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})
require.NoError(t, err)
_, _, err = man.GetLocalIPFS()
require.NoError(t, err)
protocolClient, err := man.GetProtocolClient()
require.NoError(t, err)
messengers[i], err = man.GetMessengerClient()
require.NotNil(t, messengers[i])
require.NoError(t, err)
tps[i] = &weshnet.TestingProtocol{
Client: &WrappedMessengerClient{protocolClient},
}
}
return messengers, tps
}
type WrappedMessengerClient struct {
protocoltypes.ProtocolServiceClient
}
func (*WrappedMessengerClient) Close() error {
return nil
}