split github branch protection

pkg/remote/github/github_branch_protection_enumerator.go

@@ -0,0 +1,45 @@
+package github
+
+import (
+	remoteerror "github.com/cloudskiff/driftctl/pkg/remote/error"
+	"github.com/cloudskiff/driftctl/pkg/resource"
+	"github.com/cloudskiff/driftctl/pkg/resource/github"
+)
+
+type GithubBranchProtectionEnumerator struct {
+	repository GithubRepository
+	factory    resource.ResourceFactory
+}
+
+func NewGithubBranchProtectionEnumerator(repo GithubRepository, factory resource.ResourceFactory) *GithubBranchProtectionEnumerator {
+	return &GithubBranchProtectionEnumerator{
+		repository: repo,
+		factory:    factory,
+	}
+}
+
+func (g *GithubBranchProtectionEnumerator) SupportedType() resource.ResourceType {
+	return github.GithubBranchProtectionResourceType
+}
+
+func (g *GithubBranchProtectionEnumerator) Enumerate() ([]resource.Resource, error) {
+	ids, err := g.repository.ListBranchProtection()
+	if err != nil {
+		return nil, remoteerror.NewResourceEnumerationError(err, string(g.SupportedType()))
+	}
+
+	results := make([]resource.Resource, len(ids))
+
+	for _, id := range ids {
+		results = append(
+			results,
+			g.factory.CreateAbstractResource(
+				string(g.SupportedType()),
+				id,
+				map[string]interface{}{},
+			),
+		)
+	}
+
+	return results, err
+}

pkg/remote/github/github_branch_protection_supplier.go

@@ -1,57 +0,0 @@
-package github
-
-import (
-	remoteerror "github.com/cloudskiff/driftctl/pkg/remote/error"
-	"github.com/cloudskiff/driftctl/pkg/resource"
-	resourcegithub "github.com/cloudskiff/driftctl/pkg/resource/github"
-	"github.com/cloudskiff/driftctl/pkg/terraform"
-	"github.com/sirupsen/logrus"
-	"github.com/zclconf/go-cty/cty"
-)
-
-type GithubBranchProtectionSupplier struct {
-	reader       terraform.ResourceReader
-	deserializer *resource.Deserializer
-	repository   GithubRepository
-	runner       *terraform.ParallelResourceReader
-}
-
-func NewGithubBranchProtectionSupplier(provider *GithubTerraformProvider, repository GithubRepository, deserializer *resource.Deserializer) *GithubBranchProtectionSupplier {
-	return &GithubBranchProtectionSupplier{
-		provider,
-		deserializer,
-		repository,
-		terraform.NewParallelResourceReader(provider.Runner().SubRunner()),
-	}
-}
-
-func (s GithubBranchProtectionSupplier) Resources() ([]resource.Resource, error) {
-
-	resourceList, err := s.repository.ListBranchProtection()
-
-	if err != nil {
-		return nil, remoteerror.NewResourceEnumerationError(err, resourcegithub.GithubBranchProtectionResourceType)
-	}
-
-	for _, id := range resourceList {
-		id := id
-		s.runner.Run(func() (cty.Value, error) {
-			completeResource, err := s.reader.ReadResource(terraform.ReadResourceArgs{
-				Ty: resourcegithub.GithubBranchProtectionResourceType,
-				ID: id,
-			})
-			if err != nil {
-				logrus.Warnf("Error reading %s[%s]: %+v", id, resourcegithub.GithubBranchProtectionResourceType, err)
-				return cty.NilVal, err
-			}
-			return *completeResource, nil
-		})
-	}
-
-	results, err := s.runner.Wait()
-	if err != nil {
-		return nil, err
-	}
-
-	return s.deserializer.Deserialize(resourcegithub.GithubBranchProtectionResourceType, results)
-}

pkg/remote/github/github_branch_protection_supplier_test.go

@@ -1,89 +0,0 @@
-package github
-
-import (
-	"context"
-	"testing"
-
-	"github.com/cloudskiff/driftctl/pkg/parallel"
-	"github.com/cloudskiff/driftctl/pkg/resource"
-	resourcegithub "github.com/cloudskiff/driftctl/pkg/resource/github"
-	"github.com/cloudskiff/driftctl/pkg/terraform"
-	"github.com/cloudskiff/driftctl/test"
-	"github.com/cloudskiff/driftctl/test/goldenfile"
-	dritftctlmocks "github.com/cloudskiff/driftctl/test/mocks"
-	testresource "github.com/cloudskiff/driftctl/test/resource"
-	"github.com/stretchr/testify/assert"
-	"github.com/stretchr/testify/mock"
-)
-
-func TestGithubBranchProtectionSupplier_Resources(t *testing.T) {
-	cases := []struct {
-		test    string
-		dirName string
-		mocks   func(client *MockGithubRepository)
-		err     error
-	}{
-		{
-			test:    "no branch protection",
-			dirName: "github_branch_protection_empty",
-			mocks: func(client *MockGithubRepository) {
-				client.On("ListBranchProtection").Return([]string{}, nil)
-			},
-			err: nil,
-		},
-		{
-			test:    "Multiple branch protections",
-			dirName: "github_branch_protection_multiples",
-			mocks: func(client *MockGithubRepository) {
-				client.On("ListBranchProtection").Return([]string{
-					"MDIwOkJyYW5jaFByb3RlY3Rpb25SdWxlMTk1NDg0NzI=", //"repo0:main"
-					"MDIwOkJyYW5jaFByb3RlY3Rpb25SdWxlMTk1NDg0Nzg=", //"repo0:toto"
-					"MDIwOkJyYW5jaFByb3RlY3Rpb25SdWxlMTk1NDg0NzQ=", //"repo1:main"
-					"MDIwOkJyYW5jaFByb3RlY3Rpb25SdWxlMTk1NDg0ODA=", //"repo1:toto"
-					"MDIwOkJyYW5jaFByb3RlY3Rpb25SdWxlMTk1NDg0NzE=", //"repo2:main"
-					"MDIwOkJyYW5jaFByb3RlY3Rpb25SdWxlMTk1NDg0Nzc=", //"repo2:toto"
-				}, nil)
-			},
-			err: nil,
-		},
-	}
-	for _, c := range cases {
-		shouldUpdate := c.dirName == *goldenfile.Update
-
-		providerLibrary := terraform.NewProviderLibrary()
-		supplierLibrary := resource.NewSupplierLibrary()
-
-		repo := testresource.InitFakeSchemaRepository(terraform.GITHUB, "4.4.0")
-		resourcegithub.InitResourcesMetadata(repo)
-		factory := terraform.NewTerraformResourceFactory(repo)
-
-		deserializer := resource.NewDeserializer(factory)
-
-		mockedRepo := MockGithubRepository{}
-		c.mocks(&mockedRepo)
-
-		if shouldUpdate {
-			provider, err := InitTestGithubProvider(providerLibrary)
-			if err != nil {
-				t.Fatal(err)
-			}
-
-			supplierLibrary.AddSupplier(NewGithubBranchProtectionSupplier(provider, &mockedRepo, deserializer))
-		}
-
-		t.Run(c.test, func(tt *testing.T) {
-			provider := dritftctlmocks.NewMockedGoldenTFProvider(c.dirName, providerLibrary.Provider(terraform.GITHUB), shouldUpdate)
-			s := &GithubBranchProtectionSupplier{
-				provider,
-				deserializer,
-				&mockedRepo,
-				terraform.NewParallelResourceReader(parallel.NewParallelRunner(context.TODO(), 10)),
-			}
-			got, err := s.Resources()
-			assert.Equal(tt, c.err, err)
-
-			mock.AssertExpectationsForObjects(tt)
-			test.CtyTestDiff(got, c.dirName, provider, deserializer, shouldUpdate, tt)
-		})
-	}
-}

pkg/remote/github/init.go

@@ -56,7 +56,8 @@ func Init(version string, alerter *alerter.Alerter,
 	remoteLibrary.AddEnumerator(NewGithubTeamMembershipEnumerator(repository, factory))
 	remoteLibrary.AddDetailsFetcher(github.GithubTeamMembershipResourceType, common.NewGenericDetailsFetcher(github.GithubTeamMembershipResourceType, provider, deserializer))
 
-	supplierLibrary.AddSupplier(NewGithubBranchProtectionSupplier(provider, repository, deserializer))
+	remoteLibrary.AddEnumerator(NewGithubBranchProtectionEnumerator(repository, factory))
+	remoteLibrary.AddDetailsFetcher(github.GithubBranchProtectionResourceType, common.NewGenericDetailsFetcher(github.GithubBranchProtectionResourceType, provider, deserializer))
 
 	err = resourceSchemaRepository.Init(version, provider.Schema())
 	if err != nil {

pkg/remote/github/init_test.go

@@ -1,20 +0,0 @@
-package github
-
-import (
-	"github.com/cloudskiff/driftctl/pkg/output"
-	"github.com/cloudskiff/driftctl/pkg/terraform"
-)
-
-func InitTestGithubProvider(providerLibrary *terraform.ProviderLibrary) (*GithubTerraformProvider, error) {
-	provider, err := NewGithubTerraformProvider("", &output.MockProgress{}, "")
-	if err != nil {
-		return nil, err
-	}
-	err = provider.Init()
-	if err != nil {
-		return nil, err
-	}
-	providerLibrary.AddProvider(terraform.GITHUB, provider)
-
-	return provider, nil
-}

pkg/remote/github_branch_protection_scanner_test.go

@@ -0,0 +1,101 @@
+package remote
+
+import (
+	"testing"
+
+	"github.com/cloudskiff/driftctl/mocks"
+	"github.com/cloudskiff/driftctl/pkg/remote/cache"
+	"github.com/cloudskiff/driftctl/pkg/remote/common"
+	"github.com/cloudskiff/driftctl/pkg/remote/github"
+	githubres "github.com/cloudskiff/driftctl/pkg/resource/github"
+	"github.com/cloudskiff/driftctl/pkg/terraform"
+	testresource "github.com/cloudskiff/driftctl/test/resource"
+	tftest "github.com/cloudskiff/driftctl/test/terraform"
+
+	"github.com/cloudskiff/driftctl/pkg/resource"
+	"github.com/cloudskiff/driftctl/test"
+	"github.com/cloudskiff/driftctl/test/goldenfile"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestScanGithubBranchProtection(t *testing.T) {
+
+	cases := []struct {
+		test    string
+		dirName string
+		mocks   func(client *github.MockGithubRepository)
+		err     error
+	}{
+		{
+			test:    "no branch protection",
+			dirName: "github_branch_protection_empty",
+			mocks: func(client *github.MockGithubRepository) {
+				client.On("ListBranchProtection").Return([]string{}, nil)
+			},
+			err: nil,
+		},
+		{
+			test:    "Multiple branch protections",
+			dirName: "github_branch_protection_multiples",
+			mocks: func(client *github.MockGithubRepository) {
+				client.On("ListBranchProtection").Return([]string{
+					"MDIwOkJyYW5jaFByb3RlY3Rpb25SdWxlMTk1NDg0NzI=", //"repo0:main"
+					"MDIwOkJyYW5jaFByb3RlY3Rpb25SdWxlMTk1NDg0Nzg=", //"repo0:toto"
+					"MDIwOkJyYW5jaFByb3RlY3Rpb25SdWxlMTk1NDg0NzQ=", //"repo1:main"
+					"MDIwOkJyYW5jaFByb3RlY3Rpb25SdWxlMTk1NDg0ODA=", //"repo1:toto"
+					"MDIwOkJyYW5jaFByb3RlY3Rpb25SdWxlMTk1NDg0NzE=", //"repo2:main"
+					"MDIwOkJyYW5jaFByb3RlY3Rpb25SdWxlMTk1NDg0Nzc=", //"repo2:toto"
+				}, nil)
+			},
+			err: nil,
+		},
+	}
+
+	schemaRepository := testresource.InitFakeSchemaRepository("github", "4.4.0")
+	githubres.InitResourcesMetadata(schemaRepository)
+	factory := terraform.NewTerraformResourceFactory(schemaRepository)
+	deserializer := resource.NewDeserializer(factory)
+	alerter := &mocks.AlerterInterface{}
+
+	for _, c := range cases {
+		t.Run(c.test, func(tt *testing.T) {
+			shouldUpdate := c.dirName == *goldenfile.Update
+
+			scanOptions := ScannerOptions{Deep: true}
+
+			providerLibrary := terraform.NewProviderLibrary()
+			remoteLibrary := common.NewRemoteLibrary()
+
+			mockedRepo := github.MockGithubRepository{}
+			c.mocks(&mockedRepo)
+			var repo github.GithubRepository = &mockedRepo
+
+			realProvider, err := tftest.InitTestGithubProvider(providerLibrary, "4.4.0")
+			if err != nil {
+				t.Fatal(err)
+			}
+			provider := tftest.NewFakeTerraformProvider(realProvider)
+			provider.WithResponse(c.dirName)
+
+			if shouldUpdate {
+				err := realProvider.Init()
+				if err != nil {
+					t.Fatal(err)
+				}
+				provider.ShouldUpdate()
+				repo = github.NewGithubRepository(realProvider.GetConfig(), cache.New(0))
+			}
+
+			remoteLibrary.AddEnumerator(github.NewGithubBranchProtectionEnumerator(repo, factory))
+			remoteLibrary.AddDetailsFetcher(githubres.GithubBranchProtectionResourceType, common.NewGenericDetailsFetcher(githubres.GithubBranchProtectionResourceType, provider, deserializer))
+
+			s := NewScanner(nil, remoteLibrary, alerter, scanOptions)
+			got, err := s.Resources()
+			assert.Equal(tt, err, c.err)
+			if err != nil {
+				return
+			}
+			test.TestAgainstGoldenFile(got, githubres.GithubBranchProtectionResourceType, c.dirName, provider, deserializer, shouldUpdate, tt)
+		})
+	}
+}