StringProp.hs

-- | Hedgehog property tests for WasmEdge_String roundtrips
module Test.Property.StringProp (spec) where

import Data.Text (Text)
import Data.Text qualified as T
import Hedgehog.Gen qualified as Gen
import Hedgehog.Range qualified as Range
import Test.Hspec
import Test.Hspec.Hedgehog

import WasmEdge.Internal.Utils (peekWasmString, withWasmString)

spec :: Spec
spec = describe "WasmEdge_String property tests" $ do
  it "ASCII string roundtrip" $ hedgehog prop_ascii_roundtrip
  it "UTF-8 string roundtrip" $ hedgehog prop_utf8_roundtrip
  it "empty string roundtrip" $ hedgehog prop_empty_roundtrip

-- ---------------------------------------------------------------------------
-- Generators
-- ---------------------------------------------------------------------------

genAscii :: Gen Text
genAscii = T.pack <$> Gen.list (Range.linear 0 256) Gen.alpha

genUnicode :: Gen Text
genUnicode = Gen.text (Range.linear 0 128) Gen.unicode

-- ---------------------------------------------------------------------------
-- Properties
-- ---------------------------------------------------------------------------

prop_ascii_roundtrip :: PropertyT IO ()
prop_ascii_roundtrip = do
  txt <- forAll genAscii
  result <- evalIO $ withWasmString txt peekWasmString
  result === txt

prop_utf8_roundtrip :: PropertyT IO ()
prop_utf8_roundtrip = do
  txt <- forAll genUnicode
  result <- evalIO $ withWasmString txt peekWasmString
  result === txt

prop_empty_roundtrip :: PropertyT IO ()
prop_empty_roundtrip = do
  result <- evalIO $ withWasmString T.empty peekWasmString
  result === T.empty