Example #1
0
        public void TestSchema(bool usePrintSchema)
        {
            var requestsSchema = new StructType(new List<StructField>
            {
                new StructField("test", new StringType(), false),
            });
            var jsonValue = requestsSchema.JsonValue.ToString();
            Mock<IStructTypeProxy> mockStructTypeProxy = new Mock<IStructTypeProxy>();
            mockStructTypeProxy.Setup(m => m.ToJson()).Returns(jsonValue);
            Mock<IDataFrameProxy> mockDataFrameProxy = new Mock<IDataFrameProxy>();
            mockDataFrameProxy.Setup(m => m.GetSchema()).Returns(mockStructTypeProxy.Object);
            mockDatasetProxy.Setup(m => m.ToDF()).Returns(mockDataFrameProxy.Object);

            var dataset = new Dataset(mockDatasetProxy.Object);

            if (usePrintSchema)
                dataset.PrintSchema();
            else
                dataset.ShowSchema();

            mockDataFrameProxy.Verify(m => m.GetSchema(), Times.Once);
            mockStructTypeProxy.Verify(m => m.ToJson(), Times.Once());
        }