From 4997b47fe11ac45e60e3301a308793867de18e71 Mon Sep 17 00:00:00 2001
From: Adithya Kumar <adithya@composewell.com>
Date: Thu, 17 Aug 2023 19:21:51 +0530
Subject: [PATCH] Add a recursive ADT to the Serialize test-suite

---
 test/Streamly/Test/Data/Serialize.hs | 29 +++++++++++++++++++++-------
 1 file changed, 22 insertions(+), 7 deletions(-)

diff --git a/test/Streamly/Test/Data/Serialize.hs b/test/Streamly/Test/Data/Serialize.hs
index 2c183ece18..199834f7b6 100644
--- a/test/Streamly/Test/Data/Serialize.hs
+++ b/test/Streamly/Test/Data/Serialize.hs
@@ -15,17 +15,15 @@ module Streamly.Test.Data.Serialize (main) where
 --------------------------------------------------------------------------------
 
 import Streamly.Internal.Data.Unbox (newBytes)
-
-import Test.Hspec as H
-
+import GHC.Generics (Generic)
+import Streamly.Test.Data.Serialize.TH (genDatatype)
 import Streamly.Internal.Data.Serialize.TH (deriveSerialize)
--- import Streamly.Internal.Data.Serialize (Serialize(..))
 
 import qualified Streamly.Internal.Data.Serialize as Serialize
 
-import Streamly.Test.Data.Serialize.TH (genDatatype)
-
 import Test.Hspec.QuickCheck
+import Test.QuickCheck
+import Test.Hspec as H
 
 --------------------------------------------------------------------------------
 -- Generated types
@@ -34,6 +32,21 @@ import Test.Hspec.QuickCheck
 $(genDatatype "CustomDatatype" 15)
 $(deriveSerialize ''CustomDatatype)
 
+--------------------------------------------------------------------------------
+-- Recursive type
+--------------------------------------------------------------------------------
+
+-- Recursive ADT
+data BinTree a
+  = Tree (BinTree a) (BinTree a)
+  | Leaf a
+  deriving (Show, Read, Eq, Generic)
+
+$(deriveSerialize ''BinTree)
+
+instance Arbitrary a => Arbitrary (BinTree a) where
+  arbitrary = oneof [Leaf <$> arbitrary, Tree <$> arbitrary <*> arbitrary]
+
 --------------------------------------------------------------------------------
 -- Test helpers
 --------------------------------------------------------------------------------
@@ -91,12 +104,14 @@ testCases = do
     prop "[CustomDatatype]"
         $ \(x :: [CustomDatatype]) -> roundtrip x
 
+    prop "BinTree" $ \(x :: BinTree Int) -> roundtrip x
+
 --------------------------------------------------------------------------------
 -- Main function
 --------------------------------------------------------------------------------
 
 moduleName :: String
-moduleName = "Data.Unboxed"
+moduleName = "Data.Serialize"
 
 main :: IO ()
 main = hspec $ H.parallel $ describe moduleName testCases