hpython-0.2: test/Optics.hs

{-# language OverloadedStrings, TemplateHaskell #-}
module Optics (opticsTests) where

import Hedgehog

import Control.Lens.Plated (transformOn)
import Control.Lens.Setter ((.~), (<>~))
import Control.Monad.IO.Class (liftIO)
import Data.Function ((&))

import qualified Data.Text.IO as Text

import Language.Python.Parse (parseModule)
import Language.Python.Render (showModule)
import Language.Python.Syntax
import Language.Python.Optics

import Helpers (shouldBeParseSuccess)

opticsTests :: Group
opticsTests = $$discover

prop_optics_1 :: Property
prop_optics_1 =
  withTests 1 . property $ do
    str <- liftIO $ Text.readFile "test/files/indent_optics_in.py"

    tree <- shouldBeParseSuccess parseModule str

    str' <- liftIO $ Text.readFile "test/files/indent_optics_out.py"
    showModule
      (transformOn _Statements (_Indent .~ [Space, Space, Space, Space]) tree) === str'

prop_optics_2 :: Property
prop_optics_2 =
  withTests 1 . property $ do
    str <- liftIO $ Text.readFile "test/files/indent_optics_in2.py"

    tree <- shouldBeParseSuccess parseModule str
    -- annotateShow $! tree

    str' <- liftIO $ Text.readFile "test/files/indent_optics_out2.py"
    showModule
      (transformOn _Statements (_Indent .~ [Space, Space, Space, Space]) tree) === str'

prop_optics_3 :: Property
prop_optics_3 =
  withTests 1 . property $ do
    str <- liftIO $ Text.readFile "test/files/exprs_optics_in1.py"

    tree <- shouldBeParseSuccess parseModule str

    str' <- liftIO $ Text.readFile "test/files/exprs_optics_out1.py"
    showModule
      (transformOn (_Statements._Exprs) (_Ident.identValue <>~ "_") tree) === str'

prop_optics_4 :: Property
prop_optics_4 =
  withTests 1 . property $ do
    str <- liftIO $ Text.readFile "test/files/idents_optics_in1.py"

    tree <- shouldBeParseSuccess parseModule str

    str' <- liftIO $ Text.readFile "test/files/idents_optics_out1.py"
    showModule (tree & _Idents.identValue .~ "b") === str'