watchtower: simplify the newHarness test function
This commit is contained in:
1 file changed
+2
-9
@@ -449,8 +449,6 @@ func newHarness(t *testing.T, cfg harnessCfg) *testHarness {
|
|||||||
MaxBackoff: time.Second,
|
MaxBackoff: time.Second,
|
||||||
ForceQuitDelay: 10 * time.Second,
|
ForceQuitDelay: 10 * time.Second,
|
||||||
}
|
}
|
||||||
client, err := wtclient.New(clientCfg)
|
|
||||||
require.NoError(t, err, "Unable to create wtclient")
|
|
||||||
|
|
||||||
h := &testHarness{
|
h := &testHarness{
|
||||||
t: t,
|
t: t,
|
||||||
@@ -459,7 +457,6 @@ func newHarness(t *testing.T, cfg harnessCfg) *testHarness {
|
|||||||
capacity: cfg.localBalance + cfg.remoteBalance,
|
capacity: cfg.localBalance + cfg.remoteBalance,
|
||||||
clientDB: clientDB,
|
clientDB: clientDB,
|
||||||
clientCfg: clientCfg,
|
clientCfg: clientCfg,
|
||||||
client: client,
|
|
||||||
serverAddr: towerAddr,
|
serverAddr: towerAddr,
|
||||||
serverDB: serverDB,
|
serverDB: serverDB,
|
||||||
serverCfg: serverCfg,
|
serverCfg: serverCfg,
|
||||||
@@ -470,12 +467,8 @@ func newHarness(t *testing.T, cfg harnessCfg) *testHarness {
|
|||||||
h.startServer()
|
h.startServer()
|
||||||
t.Cleanup(h.stopServer)
|
t.Cleanup(h.stopServer)
|
||||||
|
|
||||||
err = client.Start()
|
h.startClient()
|
||||||
require.NoError(t, err)
|
t.Cleanup(h.client.ForceQuit)
|
||||||
t.Cleanup(client.ForceQuit)
|
|
||||||
|
|
||||||
err = client.AddTower(towerAddr)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
h.makeChannel(0, h.cfg.localBalance, h.cfg.remoteBalance)
|
h.makeChannel(0, h.cfg.localBalance, h.cfg.remoteBalance)
|
||||||
if !cfg.noRegisterChan0 {
|
if !cfg.noRegisterChan0 {
|
||||||
|
|||||||
Reference in new issue
Block a user