Add unit tests

Author: mbaijal

pkg/generate/sagemaker_test.go

@@ -140,3 +140,89 @@ func TestSageMaker_Error_Suffix_Message(t *testing.T) {
 	// Validation Exception has suffix ModelPackageGroup arn:aws:sagemaker:/ does not exist
 	assert.Equal("&& strings.HasSuffix(awsErr.Message(), \"does not exist.\") ", code.CheckExceptionMessage(crd.Config(), crd, 404))
 }
+
+func TestSageMaker_RequeueOnSuccessSeconds(t *testing.T) {
+	assert := assert.New(t)
+	require := require.New(t)
+
+	g := testutil.NewGeneratorForService(t, "sagemaker")
+
+	crds, err := g.GetCRDs()
+	require.Nil(err)
+
+	crd := getCRDByName("Endpoint", crds)
+	require.NotNil(crd)
+
+	// The CreateEndpoint has the following definition:
+	//
+	// "CreateEndpoint":{
+	// 	"name":"CreateEndpoint",
+	// 	"http":{
+	// 	  "method":"POST",
+	// 	  "requestUri":"/"
+	// 	},
+	// 	"input":{"shape":"CreateEndpointInput"},
+	// 	"output":{"shape":"CreateEndpointOutput"},
+	// 	"errors":[
+	// 	  {"shape":"ResourceLimitExceeded"}
+	// 	]
+	//   }
+	//
+	// Where the CreateEndpointOutput shape looks like this:
+	//
+	// "CreateEndpointOutput":{
+	// 	"type":"structure",
+	// 	"required":["EndpointArn"],
+	// 	"members":{
+	// 	  "EndpointArn":{"shape":"EndpointArn"}
+	// 	}
+	//   }
+	//
+	// So, we expect that crd.ReconcileRequeuOnSuccessSeconds() returns the requeue 
+	// duration specified in the config file
+	assert.Equal(10, crd.ReconcileRequeuOnSuccessSeconds())
+}
+
+func TestSageMaker_RequeueOnSuccessSeconds_Default(t *testing.T) {
+	assert := assert.New(t)
+	require := require.New(t)
+
+	g := testutil.NewGeneratorForService(t, "sagemaker")
+
+	crds, err := g.GetCRDs()
+	require.Nil(err)
+
+	crd := getCRDByName("DataQualityJobDefinition", crds)
+	require.NotNil(crd)
+
+	// The CreateDataQualityJobDefinition has the following definition:
+	//
+	//   "CreateDataQualityJobDefinition":{
+	//   "name":"CreateDataQualityJobDefinition",
+	//   "http":{
+	//     "method":"POST",
+	//     "requestUri":"/"
+	//   },
+	//   "input":{"shape":"CreateDataQualityJobDefinitionRequest"},
+	//   "output":{"shape":"CreateDataQualityJobDefinitionResponse"},
+	//   "errors":[
+	//     {"shape":"ResourceLimitExceeded"},
+	//     {"shape":"ResourceInUse"}
+	//   ]
+	// }
+	//
+	// Where the CreateDataQualityJobDefinitionResponse shape looks like this:
+	//
+	// "CreateDataQualityJobDefinitionResponse":{
+	// 	"type":"structure",
+	// 	"required":["JobDefinitionArn"],
+	// 	"members":{
+	// 	  "JobDefinitionArn":{"shape":"MonitoringJobDefinitionArn"}
+	// 	}
+	// }
+	//
+	// So, we expect that crd.ReconcileRequeuOnSuccessSeconds() returns the default  
+	// requeue duration of 0 because it is not specified in the config file
+	assert.Equal(0, crd.ReconcileRequeuOnSuccessSeconds())
+
+}

pkg/generate/testdata/models/apis/sagemaker/0000-00-00/generator.yaml

@@ -19,6 +19,9 @@ resources:
             404:
               code: ValidationException
               message_suffix: does not exist.
+  Endpoint:
+    reconcile: 
+      requeue_on_success_seconds: 10
 ignore:
     resource_names:
       - Algorithm
@@ -35,7 +38,7 @@ ignore:
       - Domain
       - EdgePackagingJob
       - EndpointConfig
-      - Endpoint
+      # - Endpoint
       - Experiment
       - FeatureGroup
       - FlowDefinition