package aws

import (
	"testing"

	"github.com/cloudskiff/driftctl/pkg/resource"
	tf "github.com/cloudskiff/driftctl/pkg/terraform"
	testresource "github.com/cloudskiff/driftctl/test/resource"
	"github.com/stretchr/testify/assert"
)

func TestAWS_Metadata_Flags(t *testing.T) {
	testcases := map[string][]resource.Flags{
		AwsAmiResourceType:                            {resource.FlagDeepMode},
		AwsApiGatewayAccountResourceType:              {},
		AwsApiGatewayApiKeyResourceType:               {},
		AwsApiGatewayAuthorizerResourceType:           {},
		AwsApiGatewayBasePathMappingResourceType:      {},
		AwsApiGatewayDeploymentResourceType:           {},
		AwsApiGatewayDomainNameResourceType:           {},
		AwsApiGatewayGatewayResponseResourceType:      {},
		AwsApiGatewayIntegrationResourceType:          {},
		AwsApiGatewayIntegrationResponseResourceType:  {},
		AwsApiGatewayMethodResourceType:               {},
		AwsApiGatewayMethodResponseResourceType:       {},
		AwsApiGatewayMethodSettingsResourceType:       {},
		AwsApiGatewayModelResourceType:                {},
		AwsApiGatewayRequestValidatorResourceType:     {},
		AwsApiGatewayResourceResourceType:             {},
		AwsApiGatewayRestApiResourceType:              {},
		AwsApiGatewayRestApiPolicyResourceType:        {},
		AwsApiGatewayStageResourceType:                {},
		AwsApiGatewayVpcLinkResourceType:              {},
		AwsApiGatewayV2ApiResourceType:                {},
		AwsApiGatewayV2VpcLinkResourceType:            {},
		AwsAppAutoscalingPolicyResourceType:           {resource.FlagDeepMode},
		AwsAppAutoscalingScheduledActionResourceType:  {},
		AwsAppAutoscalingTargetResourceType:           {resource.FlagDeepMode},
		AwsCloudformationStackResourceType:            {resource.FlagDeepMode},
		AwsCloudfrontDistributionResourceType:         {resource.FlagDeepMode},
		AwsDbInstanceResourceType:                     {resource.FlagDeepMode},
		AwsDbSubnetGroupResourceType:                  {resource.FlagDeepMode},
		AwsDefaultNetworkACLResourceType:              {resource.FlagDeepMode},
		AwsDefaultRouteTableResourceType:              {resource.FlagDeepMode},
		AwsDefaultSecurityGroupResourceType:           {resource.FlagDeepMode},
		AwsDefaultSubnetResourceType:                  {resource.FlagDeepMode},
		AwsDefaultVpcResourceType:                     {resource.FlagDeepMode},
		AwsDynamodbTableResourceType:                  {resource.FlagDeepMode},
		AwsEbsSnapshotResourceType:                    {resource.FlagDeepMode},
		AwsEbsVolumeResourceType:                      {resource.FlagDeepMode},
		AwsEcrRepositoryResourceType:                  {resource.FlagDeepMode},
		AwsEipResourceType:                            {resource.FlagDeepMode},
		AwsEipAssociationResourceType:                 {resource.FlagDeepMode},
		AwsIamAccessKeyResourceType:                   {resource.FlagDeepMode},
		AwsIamPolicyResourceType:                      {resource.FlagDeepMode},
		AwsIamPolicyAttachmentResourceType:            {resource.FlagDeepMode},
		AwsIamRoleResourceType:                        {resource.FlagDeepMode},
		AwsIamRolePolicyResourceType:                  {resource.FlagDeepMode},
		AwsIamRolePolicyAttachmentResourceType:        {resource.FlagDeepMode},
		AwsIamUserResourceType:                        {resource.FlagDeepMode},
		AwsIamUserPolicyResourceType:                  {resource.FlagDeepMode},
		AwsIamUserPolicyAttachmentResourceType:        {resource.FlagDeepMode},
		AwsInstanceResourceType:                       {resource.FlagDeepMode},
		AwsInternetGatewayResourceType:                {resource.FlagDeepMode},
		AwsKeyPairResourceType:                        {resource.FlagDeepMode},
		AwsKmsAliasResourceType:                       {resource.FlagDeepMode},
		AwsKmsKeyResourceType:                         {resource.FlagDeepMode},
		AwsLambdaEventSourceMappingResourceType:       {resource.FlagDeepMode},
		AwsLambdaFunctionResourceType:                 {resource.FlagDeepMode},
		AwsNatGatewayResourceType:                     {resource.FlagDeepMode},
		AwsNetworkACLResourceType:                     {resource.FlagDeepMode},
		AwsRDSClusterResourceType:                     {resource.FlagDeepMode},
		AwsRDSClusterInstanceResourceType:             {},
		AwsRouteResourceType:                          {resource.FlagDeepMode},
		AwsRoute53HealthCheckResourceType:             {resource.FlagDeepMode},
		AwsRoute53RecordResourceType:                  {resource.FlagDeepMode},
		AwsRoute53ZoneResourceType:                    {resource.FlagDeepMode},
		AwsRouteTableResourceType:                     {resource.FlagDeepMode},
		AwsRouteTableAssociationResourceType:          {resource.FlagDeepMode},
		AwsS3BucketResourceType:                       {resource.FlagDeepMode},
		AwsS3BucketAnalyticsConfigurationResourceType: {resource.FlagDeepMode},
		AwsS3BucketInventoryResourceType:              {resource.FlagDeepMode},
		AwsS3BucketMetricResourceType:                 {resource.FlagDeepMode},
		AwsS3BucketNotificationResourceType:           {resource.FlagDeepMode},
		AwsS3BucketPolicyResourceType:                 {resource.FlagDeepMode},
		AwsSecurityGroupResourceType:                  {resource.FlagDeepMode},
		AwsSnsTopicResourceType:                       {resource.FlagDeepMode},
		AwsSnsTopicPolicyResourceType:                 {resource.FlagDeepMode},
		AwsSnsTopicSubscriptionResourceType:           {resource.FlagDeepMode},
		AwsSqsQueueResourceType:                       {resource.FlagDeepMode},
		AwsSqsQueuePolicyResourceType:                 {resource.FlagDeepMode},
		AwsSubnetResourceType:                         {resource.FlagDeepMode},
		AwsVpcResourceType:                            {resource.FlagDeepMode},
		AwsSecurityGroupRuleResourceType:              {resource.FlagDeepMode},
		AwsNetworkACLRuleResourceType:                 {resource.FlagDeepMode},
	}

	schemaRepository := testresource.InitFakeSchemaRepository(tf.AWS, "3.19.0")
	InitResourcesMetadata(schemaRepository)

	for ty, flags := range testcases {
		t.Run(ty, func(tt *testing.T) {
			sch, exist := schemaRepository.GetSchema(ty)
			assert.True(tt, exist)

			if len(flags) == 0 {
				assert.Equal(tt, resource.Flags(0x0), sch.Flags, "should not have any flag")
				return
			}

			for _, flag := range flags {
				assert.Truef(tt, sch.Flags.HasFlag(flag), "should have given flag %d", flag)
			}
		})
	}
}