test: add unit test

goto · Dec 2, 2024 · b18752e · b18752e
1 parent 78ee4c8
commit b18752e
Showing 1 changed file with 60 additions and 0 deletions.
diff --git a/api/handler/v1beta1/provider_test.go b/api/handler/v1beta1/provider_test.go
@@ -48,6 +48,12 @@ func (s *GrpcHandlersSuite) TestListProvider() {
 							},
 						},
 					},
+					Policies: []*domain.ProviderPolicy{
+						{
+							When:   "test-when",
+							Policy: "test-policy",
+						},
+					},
 				},
 				CreatedAt: timeNow,
 				UpdatedAt: timeNow,
@@ -82,6 +88,12 @@ func (s *GrpcHandlersSuite) TestListProvider() {
 								},
 							},
 						},
+						Policies: []*guardianv1beta1.ProviderPolicy{
+							{
+								When:   "test-when",
+								Policy: "test-policy",
+							},
+						},
 					},
 					CreatedAt: timestamppb.New(timeNow),
 					UpdatedAt: timestamppb.New(timeNow),
@@ -172,6 +184,12 @@ func (s *GrpcHandlersSuite) TestGetProvider() {
 						},
 					},
 				},
+				Policies: []*domain.ProviderPolicy{
+					{
+						When:   "test-when",
+						Policy: "test-policy",
+					},
+				},
 			},
 			CreatedAt: timeNow,
 			UpdatedAt: timeNow,
@@ -200,6 +218,12 @@ func (s *GrpcHandlersSuite) TestGetProvider() {
 							},
 						},
 					},
+					Policies: []*guardianv1beta1.ProviderPolicy{
+						{
+							When:   "test-when",
+							Policy: "test-policy",
+						},
+					},
 				},
 				CreatedAt: timestamppb.New(timeNow),
 				UpdatedAt: timestamppb.New(timeNow),
@@ -352,6 +376,12 @@ func (s *GrpcHandlersSuite) TestCreateProvider() {
 						Description: "Please enter your username",
 					},
 				},
+				Policies: []*domain.ProviderPolicy{
+					{
+						When:   "test-when",
+						Policy: "test-policy",
+					},
+				},
 			},
 		}
 		expectedResponse := &guardianv1beta1.CreateProviderResponse{
@@ -390,6 +420,12 @@ func (s *GrpcHandlersSuite) TestCreateProvider() {
 							Description: "Please enter your username",
 						},
 					},
+					Policies: []*guardianv1beta1.ProviderPolicy{
+						{
+							When:   "test-when",
+							Policy: "test-policy",
+						},
+					},
 				},
 				CreatedAt: timestamppb.New(timeNow),
 				UpdatedAt: timestamppb.New(timeNow),
@@ -434,6 +470,12 @@ func (s *GrpcHandlersSuite) TestCreateProvider() {
 						Description: "Please enter your username",
 					},
 				},
+				Policies: []*guardianv1beta1.ProviderPolicy{
+					{
+						When:   "test-when",
+						Policy: "test-policy",
+					},
+				},
 			},
 		}
 		res, err := s.grpcServer.CreateProvider(context.Background(), req)
@@ -508,6 +550,12 @@ func (s *GrpcHandlersSuite) TestUpdatedProvider() {
 						},
 					},
 				},
+				Policies: []*domain.ProviderPolicy{
+					{
+						When:   "test-when",
+						Policy: "test-policy",
+					},
+				},
 			},
 		}
 		expectedResponse := &guardianv1beta1.UpdateProviderResponse{
@@ -534,6 +582,12 @@ func (s *GrpcHandlersSuite) TestUpdatedProvider() {
 							},
 						},
 					},
+					Policies: []*guardianv1beta1.ProviderPolicy{
+						{
+							When:   "test-when",
+							Policy: "test-policy",
+						},
+					},
 				},
 				CreatedAt: timestamppb.New(timeNow),
 				UpdatedAt: timestamppb.New(timeNow),
@@ -566,6 +620,12 @@ func (s *GrpcHandlersSuite) TestUpdatedProvider() {
 						},
 					},
 				},
+				Policies: []*guardianv1beta1.ProviderPolicy{
+					{
+						When:   "test-when",
+						Policy: "test-policy",
+					},
+				},
 			},
 		}
 		res, err := s.grpcServer.UpdateProvider(context.Background(), req)