minio-hs/test/Network/Minio/XmlGenerator/Test.hs

--
-- MinIO Haskell SDK, (C) 2017 MinIO, Inc.
--
-- Licensed under the Apache License, Version 2.0 (the "License");
-- you may not use this file except in compliance with the License.
-- You may obtain a copy of the License at
--
--     http://www.apache.org/licenses/LICENSE-2.0
--
-- Unless required by applicable law or agreed to in writing, software
-- distributed under the License is distributed on an "AS IS" BASIS,
-- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-- See the License for the specific language governing permissions and
-- limitations under the License.
--
{-# LANGUAGE QuasiQuotes #-}

module Network.Minio.XmlGenerator.Test
  ( xmlGeneratorTests,
  )
where

import qualified Data.ByteString.Lazy as LBS
import Lib.Prelude
import Network.Minio.Data
import Network.Minio.TestHelpers
import Network.Minio.XmlGenerator
import Network.Minio.XmlParser (parseNotification)
import Test.Tasty
import Test.Tasty.HUnit
import Text.RawString.QQ (r)
import Text.XML (def, parseLBS)

xmlGeneratorTests :: TestTree
xmlGeneratorTests =
  testGroup
    "XML Generator Tests"
    [ testCase "Test mkCreateBucketConfig" testMkCreateBucketConfig,
      testCase "Test mkCompleteMultipartUploadRequest" testMkCompleteMultipartUploadRequest,
      testCase "Test mkPutNotificationRequest" testMkPutNotificationRequest,
      testCase "Test mkSelectRequest" testMkSelectRequest
    ]

testMkCreateBucketConfig :: Assertion
testMkCreateBucketConfig = do
  let ns = "http://s3.amazonaws.com/doc/2006-03-01/"
  assertEqual "CreateBucketConfiguration xml should match: " expected $
    mkCreateBucketConfig ns "EU"
  where
    expected =
      "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\
      \<CreateBucketConfiguration xmlns=\"http://s3.amazonaws.com/doc/2006-03-01/\">\
      \<LocationConstraint>EU</LocationConstraint>\
      \</CreateBucketConfiguration>"

testMkCompleteMultipartUploadRequest :: Assertion
testMkCompleteMultipartUploadRequest =
  assertEqual "completeMultipartUpload xml should match: " expected $
    mkCompleteMultipartUploadRequest [(1, "abc")]
  where
    expected =
      "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\
      \<CompleteMultipartUpload>\
      \<Part>\
      \<PartNumber>1</PartNumber><ETag>abc</ETag>\
      \</Part>\
      \</CompleteMultipartUpload>"

testMkPutNotificationRequest :: Assertion
testMkPutNotificationRequest =
  forM_ cases $ \val -> do
    let ns = "http://s3.amazonaws.com/doc/2006-03-01/"
        result = fromStrictBS $ mkPutNotificationRequest ns val
    ntf <- runExceptT $ runTestNS $ parseNotification result
    either
      (\_ -> assertFailure "XML Parse Error!")
      (@?= val)
      ntf
  where
    cases =
      [ Notification
          []
          [ NotificationConfig
              "YjVkM2Y0YmUtNGI3NC00ZjQyLWEwNGItNDIyYWUxY2I0N2M4"
              "arn:aws:sns:us-east-1:account-id:s3notificationtopic2"
              [ReducedRedundancyLostObject, ObjectCreated]
              defaultFilter
          ]
          [],
        Notification
          [ NotificationConfig
              "1"
              "arn:aws:sqs:us-west-2:444455556666:s3notificationqueue"
              [ObjectCreatedPut]
              ( Filter $
                  FilterKey $
                    FilterRules
                      [ FilterRule "prefix" "images/",
                        FilterRule "suffix" ".jpg"
                      ]
              ),
            NotificationConfig
              ""
              "arn:aws:sqs:us-east-1:356671443308:s3notificationqueue"
              [ObjectCreated]
              defaultFilter
          ]
          [ NotificationConfig
              ""
              "arn:aws:sns:us-east-1:356671443308:s3notificationtopic2"
              [ReducedRedundancyLostObject]
              defaultFilter
          ]
          [ NotificationConfig
              "ObjectCreatedEvents"
              "arn:aws:lambda:us-west-2:35667example:function:CreateThumbnail"
              [ObjectCreated]
              defaultFilter
          ]
      ]

testMkSelectRequest :: Assertion
testMkSelectRequest = mapM_ assertFn cases
  where
    assertFn (a, b) =
      let generatedReqDoc = parseLBS def $ LBS.fromStrict $ mkSelectRequest a
          expectedReqDoc = parseLBS def $ LBS.fromStrict b
       in case (generatedReqDoc, expectedReqDoc) of
            (Right genDoc, Right expDoc) -> assertEqual "selectRequest XML should match: " expDoc genDoc
            (Left err, _) -> assertFailure $ "Generated selectRequest failed to parse as XML" ++ show err
            (_, Left err) -> assertFailure $ "Expected selectRequest failed to parse as XML" ++ show err
    cases =
      [ ( SelectRequest
            "Select * from S3Object"
            SQL
            ( InputSerialization
                (Just CompressionTypeGzip)
                ( InputFormatCSV $
                    fileHeaderInfo FileHeaderIgnore
                      <> recordDelimiter "\n"
                      <> fieldDelimiter ","
                      <> quoteCharacter "\""
                      <> quoteEscapeCharacter "\""
                )
            )
            ( OutputSerializationCSV $
                quoteFields QuoteFieldsAsNeeded
                  <> recordDelimiter "\n"
                  <> fieldDelimiter ","
                  <> quoteCharacter "\""
                  <> quoteEscapeCharacter "\""
            )
            (Just False),
          [r|<?xml version="1.0" encoding="UTF-8"?><SelectRequest><Expression>Select * from S3Object</Expression><ExpressionType>SQL</ExpressionType><InputSerialization><CompressionType>GZIP</CompressionType><CSV><FieldDelimiter>,</FieldDelimiter><FileHeaderInfo>IGNORE</FileHeaderInfo><QuoteCharacter>"</QuoteCharacter><QuoteEscapeCharacter>"</QuoteEscapeCharacter><RecordDelimiter>
</RecordDelimiter></CSV></InputSerialization><OutputSerialization><CSV><FieldDelimiter>,</FieldDelimiter><QuoteCharacter>"</QuoteCharacter><QuoteEscapeCharacter>"</QuoteEscapeCharacter><QuoteFields>ASNEEDED</QuoteFields><RecordDelimiter>
</RecordDelimiter></CSV></OutputSerialization><RequestProgress><Enabled>FALSE</Enabled></RequestProgress></SelectRequest>|]
        ),
        ( setRequestProgressEnabled False $
            setInputCompressionType CompressionTypeGzip $
              selectRequest
                "Select * from S3Object"
                documentJsonInput
                (outputJSONFromRecordDelimiter "\n"),
          [r|<?xml version="1.0" encoding="UTF-8"?><SelectRequest><Expression>Select * from S3Object</Expression><ExpressionType>SQL</ExpressionType><InputSerialization><CompressionType>GZIP</CompressionType><JSON><Type>DOCUMENT</Type></JSON></InputSerialization><OutputSerialization><JSON><RecordDelimiter>
</RecordDelimiter></JSON></OutputSerialization><RequestProgress><Enabled>FALSE</Enabled></RequestProgress></SelectRequest>|]
        ),
        ( setRequestProgressEnabled False $
            setInputCompressionType CompressionTypeNone $
              selectRequest
                "Select * from S3Object"
                defaultParquetInput
                ( outputCSVFromProps $
                    quoteFields QuoteFieldsAsNeeded
                      <> recordDelimiter "\n"
                      <> fieldDelimiter ","
                      <> quoteCharacter "\""
                      <> quoteEscapeCharacter "\""
                ),
          [r|<?xml version="1.0" encoding="UTF-8"?><SelectRequest><Expression>Select * from S3Object</Expression><ExpressionType>SQL</ExpressionType><InputSerialization><CompressionType>NONE</CompressionType><Parquet/></InputSerialization><OutputSerialization><CSV><FieldDelimiter>,</FieldDelimiter><QuoteCharacter>"</QuoteCharacter><QuoteEscapeCharacter>"</QuoteEscapeCharacter><QuoteFields>ASNEEDED</QuoteFields><RecordDelimiter>
</RecordDelimiter></CSV></OutputSerialization><RequestProgress><Enabled>FALSE</Enabled></RequestProgress></SelectRequest>|]
        )
      ]