From 60f58b7812be3b313043a6f74bb9d5f9eb514eb0 Mon Sep 17 00:00:00 2001 From: Elle Mouton Date: Wed, 12 Oct 2022 09:15:13 +0200 Subject: [PATCH] watchtower: simplify the newHarness test function --- watchtower/wtclient/client_test.go | 11 ++--------- 1 file changed, 2 insertions(+), 9 deletions(-) diff --git a/watchtower/wtclient/client_test.go b/watchtower/wtclient/client_test.go index 312a7bc1b..a680f0a6b 100644 --- a/watchtower/wtclient/client_test.go +++ b/watchtower/wtclient/client_test.go @@ -449,8 +449,6 @@ func newHarness(t *testing.T, cfg harnessCfg) *testHarness { MaxBackoff: time.Second, ForceQuitDelay: 10 * time.Second, } - client, err := wtclient.New(clientCfg) - require.NoError(t, err, "Unable to create wtclient") h := &testHarness{ t: t, @@ -459,7 +457,6 @@ func newHarness(t *testing.T, cfg harnessCfg) *testHarness { capacity: cfg.localBalance + cfg.remoteBalance, clientDB: clientDB, clientCfg: clientCfg, - client: client, serverAddr: towerAddr, serverDB: serverDB, serverCfg: serverCfg, @@ -470,12 +467,8 @@ func newHarness(t *testing.T, cfg harnessCfg) *testHarness { h.startServer() t.Cleanup(h.stopServer) - err = client.Start() - require.NoError(t, err) - t.Cleanup(client.ForceQuit) - - err = client.AddTower(towerAddr) - require.NoError(t, err) + h.startClient() + t.Cleanup(h.client.ForceQuit) h.makeChannel(0, h.cfg.localBalance, h.cfg.remoteBalance) if !cfg.noRegisterChan0 {