--
-- 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 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)

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) = assertEqual "selectRequest XML should match: " b $ mkSelectRequest a
    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><QuoteCharacter>&#34;</QuoteCharacter><RecordDelimiter>
</RecordDelimiter><FileHeaderInfo>IGNORE</FileHeaderInfo><QuoteEscapeCharacter>&#34;</QuoteEscapeCharacter><FieldDelimiter>,</FieldDelimiter></CSV></InputSerialization><OutputSerialization><CSV><QuoteCharacter>&#34;</QuoteCharacter><QuoteFields>ASNEEDED</QuoteFields><RecordDelimiter>
</RecordDelimiter><QuoteEscapeCharacter>&#34;</QuoteEscapeCharacter><FieldDelimiter>,</FieldDelimiter></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><QuoteCharacter>&#34;</QuoteCharacter><QuoteFields>ASNEEDED</QuoteFields><RecordDelimiter>
</RecordDelimiter><QuoteEscapeCharacter>&#34;</QuoteEscapeCharacter><FieldDelimiter>,</FieldDelimiter></CSV></OutputSerialization><RequestProgress><Enabled>FALSE</Enabled></RequestProgress></SelectRequest>|]
        )
      ]