Return ID when datasource is created (#5406)

mindersec · Feb 5, 2025 · bc651b1 · bc651b1
1 parent 15f65ba
commit bc651b1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/internal/datasources/service/service.go b/internal/datasources/service/service.go
@@ -235,6 +235,8 @@ func (d *dataSourceService) Create(
 		return nil, fmt.Errorf("failed to commit transaction: %w", err)
 	}
 
+	ds.Id = dsRecord.ID.String()
+
 	return ds, nil
 }
 

diff --git a/internal/datasources/service/service_test.go b/internal/datasources/service/service_test.go
@@ -563,6 +563,7 @@ func TestCreate(t *testing.T) {
 
 			require.NoError(t, err)
 			assert.Equal(t, tt.want.Name, got.Name)
+			assert.NotEmpty(t, got.Id)
 		})
 	}
 }
-Original file line number
+Diff line change
@@ Expand Up / @@ -235,6 +235,8 @@ func (d *dataSourceService) Create( @@
     		return nil, fmt.Errorf("failed to commit transaction: %w", err)
     	}
+    	ds.Id = dsRecord.ID.String()
     	return ds, nil
     }
@@ Expand Down @@