Add CDDL testing to golden and roundtrip tests

Author: jasagredo

ouroboros-consensus/ouroboros-consensus.cabal

@@ -408,6 +408,7 @@ library unstable-consensus-testlib
     Test.Util.SOP
     Test.Util.SanityCheck
     Test.Util.Schedule
+    Test.Util.Serialisation.CDDL
     Test.Util.Serialisation.Examples
     Test.Util.Serialisation.Golden
     Test.Util.Serialisation.Roundtrip
@@ -459,6 +460,7 @@ library unstable-consensus-testlib
     ouroboros-network-api,
     ouroboros-network-mock,
     pretty-simple,
+    process,
     quickcheck-instances,
     quickcheck-state-machine:no-vendored-treediff ^>=0.10,
     quiet,
@@ -477,6 +479,7 @@ library unstable-consensus-testlib
     tasty-quickcheck >=0.11,
     tasty-rerun,
     template-haskell,
+    temporary,
     text,
     time,
     transformers-base,

ouroboros-consensus/src/unstable-consensus-testlib/Test/Util/Serialisation/CDDL.hs

@@ -0,0 +1,45 @@
+{-# LANGUAGE LambdaCase #-}
+
+module Test.Util.Serialisation.CDDL (cddlTestCase, cddlTest, CDDLsForNodeToNode (..)) where
+
+import qualified Data.ByteString as BS
+import qualified Data.Text as T
+import System.Exit
+import System.IO
+import System.IO.Temp
+import System.Process
+import Test.Tasty
+import Test.Tasty.HUnit
+
+-- | A Tasty test case running the @cuddle@
+cddlTestCase :: IO BS.ByteString -> FilePath -> T.Text -> TestTree
+cddlTestCase cborM cddl rule =
+  testCase "CDDL compliance" $
+    cddlTest cborM cddl rule >>= \case
+      Left err -> assertFailure err
+      Right _ -> pure ()
+
+-- | Test the CDDL conformance of the given bytestring
+cddlTest ::
+  IO BS.ByteString ->
+  String ->
+  T.Text ->
+  IO (Either String ())
+cddlTest cborM cddl rule =
+  withTempFile "." "testcase.cbor" $ \fp h -> do
+    bs <- cborM
+    BS.hPutStr h bs
+    hClose h
+    (code, _out, err) <-
+      readProcessWithExitCode "cuddle" ["validate-cbor", "-c", fp, "-r", T.unpack rule, cddl] mempty
+    case code of
+      ExitFailure _ -> do
+        BS.writeFile "failing.cbor" bs
+        pure (Left err)
+      ExitSuccess -> pure (Right ())
+
+-- | A collection of CDDL spec and the relevant rule to use
+data CDDLsForNodeToNode = CDDLsForNodeToNode
+  { blockCDDL :: (FilePath, T.Text)
+  , headerCDDL :: (FilePath, T.Text)
+  }

ouroboros-consensus/src/unstable-consensus-testlib/Test/Util/Serialisation/Golden.hs

@@ -47,6 +47,7 @@ import qualified Data.ByteString.UTF8 as BS.UTF8
 import Data.List (nub)
 import qualified Data.Map.Strict as Map
 import Data.Proxy (Proxy (..))
+import qualified Data.Text as T
 import Data.TreeDiff
 import GHC.Stack (HasCallStack)
 import Ouroboros.Consensus.Block (CodecConfig)
@@ -78,6 +79,7 @@ import System.FilePath (takeDirectory, (</>))
 import Test.Cardano.Binary.TreeDiff (CBORBytes (..))
 import Test.Tasty
 import Test.Tasty.Golden.Advanced (goldenTest)
+import Test.Util.Serialisation.CDDL
 import Test.Util.Serialisation.Examples (Examples (..), Labelled)
 import Test.Util.Serialisation.SomeResult (SomeResult (..))
 
@@ -96,14 +98,27 @@ goldenTestCBOR ::
   (a -> Encoding) ->
   -- | Path to the file containing the golden output
   FilePath ->
+  -- | Path to the CDDL file that defines this CBOR, and the rule name
+  Maybe (FilePath, T.Text) ->
   TestTree
-goldenTestCBOR testName example enc goldenFile =
-  goldenTest
-    testName
-    (Strict.readFile goldenFile)
-    (either exceptionToByteString id <$> try (evaluate actualValue))
-    diff
-    updateGoldenFile
+goldenTestCBOR testName example enc goldenFile mCddlPath =
+  testGroup testName $
+    [ goldenTest
+        "Golden == actual"
+        (Strict.readFile goldenFile)
+        (either exceptionToByteString id <$> try (evaluate actualValue))
+        diff
+        updateGoldenFile
+    ]
+      ++ ( case mCddlPath of
+             Nothing -> []
+             Just (cddlPath, rule) ->
+               [ cddlTestCase
+                   (Strict.readFile goldenFile)
+                   cddlPath
+                   rule
+               ]
+         )
  where
   -- Copied from tasty-golden because it isn't exported
   updateGoldenFile :: Strict.ByteString -> IO ()
@@ -185,18 +200,19 @@ goldenTests ::
   (a -> Encoding) ->
   -- | Folder containing the golden files
   FilePath ->
+  Maybe (FilePath, T.Text) ->
   TestTree
-goldenTests testName examples enc goldenFolder
+goldenTests testName examples enc goldenFolder mCDDL
   | nub labels /= labels =
       error $ "Examples with the same label for " <> testName
   | [(Nothing, example)] <- examples =
       -- If there's just a single unlabelled example, no need for grouping,
       -- which makes the output more verbose.
-      goldenTestCBOR testName example enc (goldenFolder </> testName)
+      goldenTestCBOR testName example enc (goldenFolder </> testName) mCDDL
   | otherwise =
       testGroup
         testName
-        [ goldenTestCBOR testName' example enc (goldenFolder </> testName')
+        [ goldenTestCBOR testName' example enc (goldenFolder </> testName') mCDDL
         | (mbLabel, example) <- examples
         , let testName' = case mbLabel of
                 Nothing -> testName
@@ -212,18 +228,19 @@ goldenTests' ::
   Labelled (a, a -> Encoding) ->
   -- | Folder containing the golden files
   FilePath ->
+  Maybe (FilePath, T.Text) ->
   TestTree
-goldenTests' testName examples goldenFolder
+goldenTests' testName examples goldenFolder mCDDL
   | nub labels /= labels =
       error $ "Examples with the same label for " <> testName
   | [(Nothing, (example, exampleEncoder))] <- examples =
       -- If there's just a single unlabelled example, no need for grouping,
       -- which makes the output more verbose.
-      goldenTestCBOR testName example exampleEncoder (goldenFolder </> testName)
+      goldenTestCBOR testName example exampleEncoder (goldenFolder </> testName) mCDDL
   | otherwise =
       testGroup
         testName
-        [ goldenTestCBOR testName' example exampleEncoder (goldenFolder </> testName')
+        [ goldenTestCBOR testName' example exampleEncoder (goldenFolder </> testName') mCDDL
         | (mbLabel, (example, exampleEncoder)) <- examples
         , let testName' = case mbLabel of
                 Nothing -> testName
@@ -272,13 +289,14 @@ goldenTest_all ::
   -- | Path relative to the root of the repository that contains the golden
   -- files
   FilePath ->
+  Maybe CDDLsForNodeToNode ->
   Examples blk ->
   TestTree
-goldenTest_all codecConfig goldenDir examples =
+goldenTest_all codecConfig goldenDir mCDDLs examples =
   testGroup
     "Golden tests"
     [ goldenTest_SerialiseDisk codecConfig goldenDir examples
-    , goldenTest_SerialiseNodeToNode codecConfig goldenDir examples
+    , goldenTest_SerialiseNodeToNode codecConfig goldenDir mCDDLs examples
     , goldenTest_SerialiseNodeToClient codecConfig goldenDir examples
     ]
 
@@ -312,6 +330,7 @@ goldenTest_SerialiseDisk codecConfig goldenDir Examples{..} =
       exampleValues
       enc
       (goldenDir </> "disk")
+      Nothing
 
   testLedgerTables :: TestTree
   testLedgerTables =
@@ -323,6 +342,7 @@ goldenTest_SerialiseDisk codecConfig goldenDir Examples{..} =
           exampleLedgerState
       )
       (goldenDir </> "disk")
+      Nothing
 
   encodeExt = encodeDiskExtLedgerState codecConfig
 
@@ -337,9 +357,10 @@ goldenTest_SerialiseNodeToNode ::
   ) =>
   CodecConfig blk ->
   FilePath ->
+  Maybe CDDLsForNodeToNode ->
   Examples blk ->
   TestTree
-goldenTest_SerialiseNodeToNode codecConfig goldenDir Examples{..} =
+goldenTest_SerialiseNodeToNode codecConfig goldenDir mCDDLs Examples{..} =
   testGroup
     "SerialiseNodeToNode"
     [ testVersion version
@@ -350,15 +371,15 @@ goldenTest_SerialiseNodeToNode codecConfig goldenDir Examples{..} =
   testVersion version =
     testGroup
       (toGoldenDirectory version)
-      [ test "Block" exampleBlock
-      , test "Header" exampleHeader
-      , test "SerialisedBlock" exampleSerialisedBlock
-      , test "SerialisedHeader" exampleSerialisedHeader
-      , test "GenTx" exampleGenTx
-      , test "GenTxId" exampleGenTxId
+      [ test "Block" exampleBlock $ fmap blockCDDL mCDDLs
+      , test "Header" exampleHeader $ fmap headerCDDL mCDDLs
+      , test "SerialisedBlock" exampleSerialisedBlock Nothing
+      , test "SerialisedHeader" exampleSerialisedHeader Nothing
+      , test "GenTx" exampleGenTx Nothing
+      , test "GenTxId" exampleGenTxId Nothing
       ]
    where
-    test :: SerialiseNodeToNode blk a => TestName -> Labelled a -> TestTree
+    test :: SerialiseNodeToNode blk a => TestName -> Labelled a -> Maybe (FilePath, T.Text) -> TestTree
     test testName exampleValues =
       goldenTests
         testName
@@ -416,6 +437,7 @@ goldenTest_SerialiseNodeToClient codecConfig goldenDir Examples{..} =
         exampleValues
         enc
         (goldenDir </> toGoldenDirectory versions)
+        Nothing
 
 {-------------------------------------------------------------------------------
   FlatTerm