lnd: replace defer cleanup with t.Cleanup

Signed-off-by: Eng Zer Jun <engzerjun@gmail.com>

server_test.go

@@ -75,7 +75,7 @@ func TestTLSAutoRegeneration(t *testing.T) {
 	if err != nil {
 		t.Fatalf("couldn't retrieve TLS config")
 	}
-	defer cleanUp()
+	t.Cleanup(cleanUp)
 
 	// Grab the certificate to test that getTLSConfig did its job correctly
 	// and generated a new cert.

witness_beacon_test.go

@@ -37,8 +37,7 @@ func TestWitnessBeaconIntercept(t *testing.T) {
 		[]byte{2},
 	)
 	require.NoError(t, err)
-
+	t.Cleanup(subscription.CancelSubscription)
-	defer subscription.CancelSubscription()
 
 	require.NoError(t, interceptedFwd.Settle(preimage))