adding graceful shutdown unit test

blob/service_test.go

@@ -734,14 +734,35 @@ func TestService_Subscribe(t *testing.T) {
 			t.Fatal("timeout waiting for first subscription response")
 		}
 
-		// check that the subscription channel is closed
 		select {
 		case _, ok := <-subCh:
 			assert.False(t, ok, "expected subscription channel to be closed")
 		case <-time.After(time.Second * 2):
 			t.Fatal("timeout waiting for subscription channel to close")
 		}
 	})
+
+	t.Run("graceful shutdown", func(t *testing.T) {
+		ns := blobs[0].Namespace()
+		subCh, err := service.Subscribe(ctx, ns)
+		require.NoError(t, err)
+
+		// cancel the subscription context after receiving the first response
+		select {
+		case <-subCh:
+			service.Stop(context.Background())
+		case <-time.After(time.Second * 2):
+			t.Fatal("timeout waiting for first subscription response")
+		}
+
+		select {
+		case _, ok := <-subCh:
+			assert.False(t, ok, "expected subscription channel to be closed")
+		case <-time.After(time.Second * 2):
+			t.Fatal("timeout waiting for subscription channel to close")
+		}
+
+	})
 }
 
 func TestService_Subscribe_MultipleNamespaces(t *testing.T) {

nodebuilder/blob/blob.go

@@ -47,7 +47,6 @@ type Module interface {
 
 type API struct {
 	Internal struct {
-<<<<<<< HEAD
 		Submit func(
 			context.Context,
 			[]*blob.Blob,