{-# LANGUAGE DeriveAnyClass #-}
{-# LANGUAGE DeriveGeneric #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE ScopedTypeVariables #-}
{-# LANGUAGE QuasiQuotes #-}

module Spark.Core.Internal.OpFunctionsSpec where

import Data.Aeson
import Test.Hspec
import Text.RawString.QQ

import Spark.Core.Functions
import Spark.Core.Internal.OpFunctions
import Spark.Core.Internal.DatasetFunctions


spec :: Spec
spec = do
  describe "extraNodeOpData" $ do
    it "should have the content of a constant dataset" $ do
      let l = [1,2,3] :: [Int]
      let res :: Maybe Value
          res = decode
              ([r|{"content": [1,2,3],
                    "cellType" : {
                      "dt": "integer",
                      "nullable": false
                    }
                  }|])
      let ds = dataset l
      let d = extraNodeOpData . nodeOp $ ds
      Just d `shouldBe` res