From 4e618a9b1aecab7dc0cece3437a34e70066f269d Mon Sep 17 00:00:00 2001
From: Matt Curtis <matt@obscu.ro>
Date: Thu, 9 May 2024 22:11:26 +0100
Subject: [PATCH] tidy

---
 integration/networktest/actions/subscription_actions.go   | 2 +-
 integration/networktest/tests/helpful/smoke_test.go       | 2 +-
 .../networktest/tests/subscription/subscriptions_test.go  | 8 ++++----
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/integration/networktest/actions/subscription_actions.go b/integration/networktest/actions/subscription_actions.go
index ecce2b1caa..99ab7a27af 100644
--- a/integration/networktest/actions/subscription_actions.go
+++ b/integration/networktest/actions/subscription_actions.go
@@ -60,7 +60,7 @@ func (r *recordNewHeadsSubscriptionAction) Run(ctx context.Context, network netw
 	return ctx, nil
 }
 
-func (r *recordNewHeadsSubscriptionAction) Verify(ctx context.Context, network networktest.NetworkConnector) error {
+func (r *recordNewHeadsSubscriptionAction) Verify(_ context.Context, _ networktest.NetworkConnector) error {
 	if len(r.recordedHeads) == 0 {
 		return fmt.Errorf("no new heads received during the %s period", r.duration)
 	}
diff --git a/integration/networktest/tests/helpful/smoke_test.go b/integration/networktest/tests/helpful/smoke_test.go
index f34eb6aa22..a97bbf54a5 100644
--- a/integration/networktest/tests/helpful/smoke_test.go
+++ b/integration/networktest/tests/helpful/smoke_test.go
@@ -18,7 +18,7 @@ func TestExecuteNativeFundsTransfer(t *testing.T) {
 	networktest.Run(
 		"native-funds-smoketest",
 		t,
-		env.UATTestnet(),
+		env.LocalDevNetwork(),
 		actions.Series(
 			&actions.CreateTestUser{UserID: 0},
 			&actions.CreateTestUser{UserID: 1},
diff --git a/integration/networktest/tests/subscription/subscriptions_test.go b/integration/networktest/tests/subscription/subscriptions_test.go
index 453cf5a541..1f86243b9e 100644
--- a/integration/networktest/tests/subscription/subscriptions_test.go
+++ b/integration/networktest/tests/subscription/subscriptions_test.go
@@ -7,22 +7,22 @@ import (
 	"github.com/ten-protocol/go-ten/integration/networktest"
 	"github.com/ten-protocol/go-ten/integration/networktest/actions"
 	"github.com/ten-protocol/go-ten/integration/networktest/env"
+	"github.com/ten-protocol/go-ten/integration/simulation/devnetwork"
 )
 
-// this test goes via the gateway only for now
 func TestGatewayNewHeadsSubscription(t *testing.T) {
 	networktest.TestOnlyRunsInIDE(t)
 	networktest.Run(
 		"gateway-new-heads-subscription",
 		t,
-		env.UATTestnet(), //env.LocalDevNetwork(devnetwork.WithGateway()),
+		env.LocalDevNetwork(devnetwork.WithGateway()),
 		actions.Series(
 			// user not technically needed, but we need a gateway address to use
 			&actions.CreateTestUser{UserID: 0, UseGateway: true},
 			actions.SetContextValue(actions.KeyNumberOfTestUsers, 1),
 
-			// Record new heads for 30 seconds, verify that the subscription is working
-			actions.RecordNewHeadsSubscription(30*time.Second),
+			// Record new heads for specified duration, verify that the subscription is working
+			actions.RecordNewHeadsSubscription(20*time.Second),
 		),
 	)
 }