Add tests

Author: millicentachieng

test/Microsoft.OpenAPI.OData.Reader.Tests/Operation/EdmActionOperationHandlerTests.cs

@@ -54,7 +54,7 @@ public void CreateOperationForEdmActionReturnsCorrectOperation()
         }
 
         [Fact]
-        public void CreateOperationForEdmActionReturnsCorrectOperationHierarhicalClass()
+        public void CreateOperationForEdmActionReturnsCorrectOperationHierarchicalClass()
         {
             // Arrange
             IEdmModel model = EdmModelHelper.ContractServiceModel;
@@ -296,5 +296,44 @@ public void OperationRestrictionsTermWorksToCreateOperationForEdmAction(bool ena
                 Assert.Empty(operation.Security);
             }
         }
+
+        [Theory]
+        [InlineData("getMailTips", true)] // returns collection
+        [InlineData("getMailTips", false)] // returns collection
+        [InlineData("findMeetingTimes", true)] // does not return collection
+        public void CreateOperationForEdmActionWithCollectionReturnTypeContainsXMsPageableExtension(string actionName, bool enablePagination)
+        {
+            // Arrange
+            IEdmModel model = EdmModelHelper.GraphBetaModel;
+            OpenApiConvertSettings settings = new OpenApiConvertSettings
+            {
+                EnableOperationId = true,
+                EnablePagination = enablePagination
+            };
+            ODataContext context = new ODataContext(model, settings);
+            IEdmAction action = model.SchemaElements.OfType<IEdmAction>()
+                .First(x => x.Name == actionName &&
+                    x.FindParameter("bindingParameter").Type.Definition.ToString() == "microsoft.graph.user");
+            IEdmEntityContainer container = model.SchemaElements.OfType<IEdmEntityContainer>().First();
+            IEdmEntitySet users = container.FindEntitySet("users");
+            IEdmEntityType user = model.SchemaElements.OfType<IEdmEntityType>().First(x => x.Name == "user");
+
+            ODataPath path = new ODataPath(new ODataNavigationSourceSegment(users),
+                new ODataKeySegment(user),
+                new ODataOperationSegment(action));
+
+            // Act
+            var operation = _operationHandler.CreateOperation(context, path);
+
+            // Assert
+            if (enablePagination && action.ReturnType.IsCollection())
+            {
+                Assert.True(operation.Extensions.ContainsKey(Common.Constants.xMsPageable));
+            }
+            else
+            {
+                Assert.False(operation.Extensions.ContainsKey(Common.Constants.xMsPageable));
+            }
+        }
     }
 }

test/Microsoft.OpenAPI.OData.Reader.Tests/Operation/EdmFunctionOperationHandlerTests.cs

@@ -112,7 +112,7 @@ public void CreateOperationForEdmFunctionReturnsCorrectOperation(bool useHTTPSta
         }
 
         [Fact]
-        public void CreateOperationForEdmFunctionReturnsCorrectOperationHierarhicalClass()
+        public void CreateOperationForEdmFunctionReturnsCorrectOperationHierarchicalClass()
         {
             // Arrange
             IEdmModel model = EdmModelHelper.ContractServiceModel;
@@ -408,5 +408,42 @@ public void OperationRestrictionsTermWorksToCreateOperationForEdmFunction(bool e
                 Assert.Empty(operation.Security);
             }
         }
+
+        [Theory]
+        [InlineData("getUserArchivedPrintJobs", true)] // returns collection
+        [InlineData("getUserArchivedPrintJobs", false)] // returns collection
+        [InlineData("managedDeviceEnrollmentAbandonmentSummary", true)] // does not return collection
+        public void CreateOperationForEdmFunctionWithCollectionReturnTypeContainsXMsPageableExtension(string functionName, bool enablePagination)
+        {
+            // Arrange
+            IEdmModel model = EdmModelHelper.GraphBetaModel;
+            OpenApiConvertSettings settings = new OpenApiConvertSettings
+            {
+                EnableOperationId = true,
+                EnablePagination = enablePagination
+            };
+            ODataContext context = new ODataContext(model, settings);
+            IEdmFunction function = model.SchemaElements.OfType<IEdmFunction>()
+                .First(x => x.Name == functionName &&
+                    x.FindParameter("bindingParameter").Type.Definition.ToString() == "microsoft.graph.reportRoot");
+            IEdmEntityContainer container = model.SchemaElements.OfType<IEdmEntityContainer>().First();
+            IEdmSingleton reports = container.FindSingleton("reports");
+
+            ODataPath path = new ODataPath(new ODataNavigationSourceSegment(reports),
+                new ODataOperationSegment(function));
+
+            // Act
+            var operation = _operationHandler.CreateOperation(context, path);
+
+            // Assert
+            if (enablePagination && function.ReturnType.IsCollection())
+            {
+                Assert.True(operation.Extensions.ContainsKey(Common.Constants.xMsPageable));
+            }
+            else
+            {
+                Assert.False(operation.Extensions.ContainsKey(Common.Constants.xMsPageable));
+            }
+        }
     }
 }