MCPcopy
hub / github.com/wavetermdev/waveterm / initTestHarness

Function initTestHarness

cmd/test-conn/testutil.go:63–105  ·  view source on GitHub ↗
(autoAccept bool)

Source from the content-addressed store, hash-verified

61}
62
63func initTestHarness(autoAccept bool) error {
64 log.Printf("Initializing test harness...")
65
66 err := setupWaveEnvVars()
67 if err != nil {
68 return fmt.Errorf("failed to setup wave env vars: %w", err)
69 }
70
71 err = wavebase.CacheAndRemoveEnvVars()
72 if err != nil {
73 return fmt.Errorf("failed to cache env vars: %w", err)
74 }
75
76 wshutil.DefaultRouter = wshutil.NewWshRouter()
77 wshutil.DefaultRouter.SetAsRootRouter()
78
79 wstore.SetClientId("test-client-" + fmt.Sprintf("%d", time.Now().Unix()))
80
81 userinput.SetUserInputProvider(&CLIProvider{AutoAccept: autoAccept})
82
83 keyPair, err := wavejwt.GenerateKeyPair()
84 if err != nil {
85 return fmt.Errorf("failed to generate JWT key pair: %w", err)
86 }
87
88 err = wavejwt.SetPrivateKey(keyPair.PrivateKey)
89 if err != nil {
90 return fmt.Errorf("failed to set JWT private key: %w", err)
91 }
92
93 err = wavejwt.SetPublicKey(keyPair.PublicKey)
94 if err != nil {
95 return fmt.Errorf("failed to set JWT public key: %w", err)
96 }
97
98 rpc := wshserver.GetMainRpcClient()
99 wshutil.DefaultRouter.RegisterTrustedLeaf(rpc, wshutil.DefaultRoute)
100
101 wconfig.GetWatcher().Start()
102
103 log.Printf("Test harness initialized")
104 return nil
105}
106
107func testBasicConnect(connName string, timeout time.Duration) error {
108 opts, err := remote.ParseOpts(connName)

Callers 1

mainFunction · 0.85

Calls 13

CacheAndRemoveEnvVarsFunction · 0.92
NewWshRouterFunction · 0.92
SetClientIdFunction · 0.92
SetUserInputProviderFunction · 0.92
GenerateKeyPairFunction · 0.92
SetPrivateKeyFunction · 0.92
SetPublicKeyFunction · 0.92
GetMainRpcClientFunction · 0.92
GetWatcherFunction · 0.92
setupWaveEnvVarsFunction · 0.85
SetAsRootRouterMethod · 0.80
RegisterTrustedLeafMethod · 0.80

Tested by

no test coverage detected