DXCDT-509: Adding auth0_connections export support (#797)

* Implementing connections according to existing pattern

* Adding unit tests

---------

Co-authored-by: Will Vedder <[email protected]>
Co-authored-by: Rita Zerrizuela <[email protected]>

.gitignore

@@ -34,3 +34,6 @@ vendor
 
 # Binary output folder
 out/
+
+# Terraform export command artifacts
+*.tf

internal/cli/terraform.go

@@ -44,6 +44,9 @@ func (i *terraformInputs) parseResourceFetchers(api *auth0.API) []resourceDataFe
 		&clientResourceFetcher{
 			api: api,
 		},
+		&connectionResourceFetcher{
+			api: api,
+		},
 	}
 }
 

internal/cli/terraform_fetcher.go

@@ -24,6 +24,10 @@ type (
 	clientResourceFetcher struct {
 		api *auth0.API
 	}
+
+	connectionResourceFetcher struct {
+		api *auth0.API
+	}
 )
 
 func (f *clientResourceFetcher) FetchData(ctx context.Context) (importDataList, error) {
@@ -58,6 +62,37 @@ func (f *clientResourceFetcher) FetchData(ctx context.Context) (importDataList,
 	return data, nil
 }
 
+func (f *connectionResourceFetcher) FetchData(ctx context.Context) (importDataList, error) {
+	var data importDataList
+
+	var page int
+	for {
+		connections, err := f.api.Connection.List(
+			ctx,
+			management.Page(page),
+			management.IncludeFields("id", "name"),
+		)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, connection := range connections.Connections {
+			data = append(data, importDataItem{
+				ResourceName: "auth0_connection." + sanitizeResourceName(connection.GetName()),
+				ImportID:     connection.GetID(),
+			})
+		}
+
+		if !connections.HasNext() {
+			break
+		}
+
+		page++
+	}
+
+	return data, nil
+}
+
 // sanitizeResourceName will return a valid terraform resource name.
 //
 // A name must start with a letter or underscore and may

internal/cli/terraform_fetcher_test.go

@@ -140,3 +140,104 @@ func TestClientResourceFetcher_FetchData(t *testing.T) {
 		assert.EqualError(t, err, "failed to list clients")
 	})
 }
+
+func TestConnectionResourceFetcher_FetchData(t *testing.T) {
+	t.Run("it successfully retrieves connections data", func(t *testing.T) {
+		ctrl := gomock.NewController(t)
+		defer ctrl.Finish()
+
+		connAPI := mock.NewMockConnectionAPI(ctrl)
+		connAPI.EXPECT().
+			List(gomock.Any(), gomock.Any(), gomock.Any()).
+			Return(
+				&management.ConnectionList{
+					List: management.List{
+						Start: 0,
+						Limit: 2,
+						Total: 4,
+					},
+					Connections: []*management.Connection{
+						{
+							ID:   auth0.String("con_id1"),
+							Name: auth0.String("Connection 1"),
+						},
+						{
+							ID:   auth0.String("con_id2"),
+							Name: auth0.String("Connection 2"),
+						},
+					},
+				},
+				nil,
+			)
+		connAPI.EXPECT().
+			List(gomock.Any(), gomock.Any(), gomock.Any()).
+			Return(
+				&management.ConnectionList{
+					List: management.List{
+						Start: 2,
+						Limit: 4,
+						Total: 4,
+					},
+					Connections: []*management.Connection{
+						{
+							ID:   auth0.String("con_id3"),
+							Name: auth0.String("Connection 3"),
+						},
+						{
+							ID:   auth0.String("con_id4"),
+							Name: auth0.String("Connection 4"),
+						},
+					},
+				},
+				nil,
+			)
+
+		fetcher := connectionResourceFetcher{
+			api: &auth0.API{
+				Connection: connAPI,
+			},
+		}
+
+		expectedData := importDataList{
+			{
+				ResourceName: "auth0_connection.Connection1",
+				ImportID:     "con_id1",
+			},
+			{
+				ResourceName: "auth0_connection.Connection2",
+				ImportID:     "con_id2",
+			},
+			{
+				ResourceName: "auth0_connection.Connection3",
+				ImportID:     "con_id3",
+			},
+			{
+				ResourceName: "auth0_connection.Connection4",
+				ImportID:     "con_id4",
+			},
+		}
+
+		data, err := fetcher.FetchData(context.Background())
+		assert.NoError(t, err)
+		assert.Equal(t, expectedData, data)
+	})
+
+	t.Run("it returns an error if api call fails", func(t *testing.T) {
+		ctrl := gomock.NewController(t)
+		defer ctrl.Finish()
+
+		connAPI := mock.NewMockConnectionAPI(ctrl)
+		connAPI.EXPECT().
+			List(gomock.Any(), gomock.Any(), gomock.Any()).
+			Return(nil, fmt.Errorf("failed to list connections"))
+
+		fetcher := connectionResourceFetcher{
+			api: &auth0.API{
+				Connection: connAPI,
+			},
+		}
+
+		_, err := fetcher.FetchData(context.Background())
+		assert.EqualError(t, err, "failed to list connections")
+	})
+}