esqueleto-postgis-1.2.0: test/Wkb/PolygonSpec.hs

{-# LANGUAGE OverloadedStrings #-}

module Wkb.PolygonSpec where

import qualified Data.ByteString.Builder as ByteStringBuilder
import qualified Data.SpecHelper as SpecHelper
import qualified Database.Esqueleto.Postgis.Wkb as Wkb
import Test.Hspec (Spec, describe, it, shouldBe)

spec :: Spec
spec = do
  testWkbPolygonParsing
  testWkbMultiPolygonParsing

-- Test Wkb Polygon Parsing

testWkbPolygonParsing :: Spec
testWkbPolygonParsing =
  describe "Test wkb polygon" $ do
    testValidWkbPolyonParsing
    testInvalidWkbPolyonParsing

testValidWkbPolyonParsing :: Spec
testValidWkbPolyonParsing =
  SpecHelper.testRoundTripWkbGeometryParsing "polygon" SpecHelper.genPolygon

testInvalidWkbPolyonParsing :: Spec
testInvalidWkbPolyonParsing =
  it "does not parse bad wkb polygon" $
    Wkb.parseByteString exampleBadWkbPolygon `shouldBe` Left "Could not parse wkb: First and last points of linear ring are different: first=GeoPointXY (PointXY {_xyX = 1.0, _xyY = 2.0}) last=GeoPointXY (PointXY {_xyX = 7.0, _xyY = 8.0})"
  where
    exampleBadWkbPolygon =
      ByteStringBuilder.toLazyByteString $
        ByteStringBuilder.word8 0
          <> ByteStringBuilder.int32BE 3
          <> ByteStringBuilder.int32BE 1
          <> ByteStringBuilder.int32BE 4
          <> ByteStringBuilder.doubleBE 1.0
          <> ByteStringBuilder.doubleBE 2.0
          <> ByteStringBuilder.doubleBE 3.0
          <> ByteStringBuilder.doubleBE 4.0
          <> ByteStringBuilder.doubleBE 5.0
          <> ByteStringBuilder.doubleBE 6.0
          <> ByteStringBuilder.doubleBE 7.0
          <> ByteStringBuilder.doubleBE 8.0

-- Test Wkb MultiPolygon Parsing

testWkbMultiPolygonParsing :: Spec
testWkbMultiPolygonParsing =
  describe "Test wkb multipolygon parsing" $
    SpecHelper.testRoundTripWkbGeometryParsing "multipolygon" SpecHelper.genMultiPolygon