Restructure websub test cases to fix tests hanging issue

ayeshLK · Apr 2, 2024 · a0e3ffa · a0e3ffa
1 parent 2b58ac7
commit a0e3ffa
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/ballerina/tests/unsubscription_test.bal b/ballerina/tests/unsubscription_test.bal
@@ -65,8 +65,6 @@ isolated function notifySubscriber(string mode) returns error? {
  }
 }
 
-listener Listener unsubscriptionTestListener = new (9102);
-
 SubscriberService unsubscriptionTestSubscriber = @SubscriberServiceConfig {
  target: ["http://127.0.0.1:9197/common/hub", "test"], 
  leaseSeconds: 36000,
@@ -87,12 +85,13 @@ service object {
  groups: ["unsubscriptionViaGracefulstop"]
 }
 function testUnsubscriptionOnGracefulStop() returns error? {
- check unsubscriptionTestListener.attach(unsubscriptionTestSubscriber, "sub");
- check unsubscriptionTestListener.'start();
+ Listener ls = check new (9102);
+ check ls.attach(unsubscriptionTestSubscriber, "sub");
+ check ls.'start();
  log:printInfo("[UNSUB_VER] Starting Subscriber");
  runtime:sleep(5);
  log:printInfo("[UNSUB_VER] Invoking graceful stop");
- check unsubscriptionTestListener.gracefulStop();
+ check ls.gracefulStop();
  runtime:sleep(5);
  log:printInfo("[UNSUB_VER] Verifying shutdown");
  test:assertTrue(isVerified());