Merge pull request #170 from sailpoint-oss/fix/transform-preview

Limit preview identity results to 250, limit fields to pull to just i…

Author: tyler-mairose-sp

assets/vhs/preview.tape

@@ -0,0 +1,27 @@
+Set FontSize 16
+Set Width 1200
+Set Height 1000
+
+Hide Sleep 2s Show
+
+Sleep 5s
+Type "sail transform preview -f transform_files/DeriveFirstInitialLastNameInUpper.json"
+Enter
+Set TypingSpeed 500ms 
+Sleep 5s
+Down
+Sleep 500ms
+Enter
+Sleep 4.5s
+Type "/"
+Sleep 3s
+Type "Brian"
+Sleep 5s
+Backspace 5
+Sleep 5s
+Type "Adam"
+Down
+Sleep 5s
+Enter
+Sleep 15s
+

cmd/connector/static/connector/src/index.spec.ts

@@ -1,5 +1,5 @@
 import { connector } from './index'
-import { Connector, RawResponse, ResponseType, StandardCommand } from '@sailpoint/connector-sdk'
+import { Connector, RawResponse, ResponseType, StandardCommand, AssumeAwsRoleRequest, AssumeAwsRoleResponse } from '@sailpoint/connector-sdk'
 import { PassThrough } from 'stream'
 
 const mockConfig: any = {
@@ -18,7 +18,11 @@ describe('connector unit tests', () => {
             StandardCommand.StdTestConnection,
             {reloadConfig() {
                 return Promise.resolve()
-            },},
+            },
+            assumeAwsRole(assumeAwsRoleRequest: AssumeAwsRoleRequest): Promise<AssumeAwsRoleResponse> {
+                return Promise.resolve(new AssumeAwsRoleResponse('accessKeyId', 'secretAccessKey', 'sessionToken', "123"))
+            }
+        },
             undefined,
             new PassThrough({ objectMode: true }).on('data', (chunk) => expect(chunk).toStrictEqual(new RawResponse ({}, ResponseType.Output)))
         )

cmd/connector/static/customizer/src/index.spec.ts

@@ -1,23 +1,29 @@
 import { connectorCustomizer } from './index'
-import { CustomizerType, StandardCommand } from '@sailpoint/connector-sdk'
+import { CustomizerType, StandardCommand, AssumeAwsRoleRequest, AssumeAwsRoleResponse } from '@sailpoint/connector-sdk'
 
 const mockConfig: any = {
-    token: 'xxx123'
+    token: 'xxx123',
 }
 process.env.CONNECTOR_CONFIG = Buffer.from(JSON.stringify(mockConfig)).toString('base64')
 
 describe('connector customizer unit tests', () => {
-
     it('should not change input from beforeStdAccountReadHandler', async () => {
         let customizer = await connectorCustomizer()
         let input = {
             identity: 'john.doe',
         }
         let updatedInput = await customizer._exec(
             customizer.handlerKey(CustomizerType.Before, StandardCommand.StdAccountRead),
-            {reloadConfig() {
-                return Promise.resolve()
-            },},
+            {
+                reloadConfig() {
+                    return Promise.resolve()
+                },
+                assumeAwsRole(assumeAwsRoleRequest: AssumeAwsRoleRequest): Promise<AssumeAwsRoleResponse> {
+                    return Promise.resolve(
+                        new AssumeAwsRoleResponse('ccessKeyId', 'secretAccessKey', 'sessionToken', '123')
+                    )
+                },
+            },
             input
         )
 
@@ -28,20 +34,27 @@ describe('connector customizer unit tests', () => {
         let customizer = await connectorCustomizer()
         let output = await customizer._exec(
             customizer.handlerKey(CustomizerType.After, StandardCommand.StdAccountRead),
-            {reloadConfig() {
-                return Promise.resolve()
-            },},
+            {
+                reloadConfig() {
+                    return Promise.resolve()
+                },
+                assumeAwsRole(assumeAwsRoleRequest: AssumeAwsRoleRequest): Promise<AssumeAwsRoleResponse> {
+                    return Promise.resolve(
+                        new AssumeAwsRoleResponse('ccessKeyId', 'secretAccessKey', 'sessionToken', '123')
+                    )
+                },
+            },
             {
                 identity: '',
                 attributes: {
                     username: 'john.doe',
                     firstName: 'john',
                     lastName: 'doe',
                     email: '[email protected]',
-                }
+                },
             }
         )
 
         expect(output.attributes.location).toStrictEqual('Austin')
     })
-})
+})

cmd/transform/preview.go

@@ -122,7 +122,9 @@ func newPreviewCommand() *cobra.Command {
 					return err
 				}
 
-				formattedResponse, err := search.PerformSearch(*apiClient, searchObj)
+				searchObj.QueryResultFilter.Includes = []string{"id", "displayName", "email"}
+
+				formattedResponse, err := search.PerformSearchWithLimit(*apiClient, searchObj, 250)
 				if err != nil {
 					return err
 				}

internal/search/search.go

@@ -120,6 +120,78 @@ func PerformSearch(apiClient sailpoint.APIClient, search sailpointsdk.Search) (S
 	return SearchResults, nil
 }
 
+func PerformSearchWithLimit(apiClient sailpoint.APIClient, search sailpointsdk.Search, limit int32) (SearchResults, error) {
+	var SearchResults SearchResults
+
+	if limit == 0 {
+		limit = 500
+	}
+
+	ctx := context.TODO()
+	resp, r, err := apiClient.V3.SearchAPI.SearchPost(ctx).Search(search).Limit(limit).Execute()
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "Error: %v\n", err)
+		fmt.Fprintf(os.Stderr, "Full HTTP response: %v\n", r)
+	}
+
+	log.Debug("Search complete")
+
+	for i := 0; i < len(resp); i++ {
+		entry := resp[i]
+		switch entry["_type"] {
+		case "accountactivity":
+			var AccountActivity AccountActivity
+			err := mapstructure.Decode(entry, &AccountActivity)
+			if err != nil {
+				return SearchResults, err
+			}
+			SearchResults.AccountActivities = append(SearchResults.AccountActivities, AccountActivity)
+
+		case "accessprofile":
+			var AccessProfile AccessProfile
+			err := mapstructure.Decode(entry, &AccessProfile)
+			if err != nil {
+				return SearchResults, err
+			}
+			SearchResults.AccessProfiles = append(SearchResults.AccessProfiles, AccessProfile)
+
+		case "entitlement":
+			var Entitlement Entitlement
+			err := mapstructure.Decode(entry, &Entitlement)
+			if err != nil {
+				return SearchResults, err
+			}
+			SearchResults.Entitlements = append(SearchResults.Entitlements, Entitlement)
+
+		case "event":
+			var Event Event
+			err := mapstructure.Decode(entry, &Event)
+			if err != nil {
+				return SearchResults, err
+			}
+			SearchResults.Events = append(SearchResults.Events, Event)
+
+		case "identity":
+			var Identity Identity
+			err := mapstructure.Decode(entry, &Identity)
+			if err != nil {
+				return SearchResults, err
+			}
+			SearchResults.Identities = append(SearchResults.Identities, Identity)
+
+		case "role":
+			var Role Role
+			err := mapstructure.Decode(entry, &Role)
+			if err != nil {
+				return SearchResults, err
+			}
+			SearchResults.Roles = append(SearchResults.Roles, Role)
+		}
+	}
+
+	return SearchResults, nil
+}
+
 func IterateIndices(SearchResults SearchResults, searchQuery string, folderPath string, outputTypes []string) error {
 	if len(SearchResults.AccountActivities) > 0 {
 		fileName := "query=" + searchQuery + "&indices=AccountActivities"