{-# LANGUAGE TypeApplications #-}
{-# LANGUAGE DeriveGeneric #-}
{-# LANGUAGE OverloadedStrings #-}

module Test.Validity.OpticsSpec where

import GHC.Generics (Generic)

import Test.Hspec

import Test.Validity
import Test.Validity.Optics

import Lens.Micro

spec :: Spec
spec = do
    describe "lensSpecOnValid" $
        lensSpecOnValid
            ((_2) :: Lens (Rational, Rational) (Rational, Rational) Rational Rational)
    describe "lensSpec" $ lensSpec ((_2) :: Lens (Int, Int) (Int, Int) Int Int)
    describe "lensSpecOnArbitrary" $
        lensSpecOnArbitrary
            ((_2) :: Lens (Rational, Rational) (Rational, Rational) Rational Rational)
    describe "lensSpecOnGen" $
        lensSpecOnGen
            ((_2) :: Lens (Rational, Rational) (Rational, Rational) Rational Rational)
            (abs <$> genValid)
            "positive valid doubles"
            (filter (0.0 >=) . shrinkValid)
            ((,) <$> (negate . abs <$> genValid) <*> (negate . abs <$> genValid))
            "tuples of negative valid doubles"
            (const [])
    describe "lensGettingProducesValidOnValid" $
        it "holds for (_2) for doubles" $
        lensGettingProducesValidOnValid
            ((_2) :: Lens (Rational, Rational) (Rational, Rational) Rational Rational)
    describe "lensGettingProducesValid" $
        it "holds for (_2) for ints" $
        lensGettingProducesValid ((_2) :: Lens (Int, Int) (Int, Int) Int Int)
    describe "lensGettingProducesValidOnArbitrary" $
        it "holds for (_2) for doubles" $
        lensGettingProducesValidOnArbitrary
            ((_2) :: Lens (Rational, Rational) (Rational, Rational) Rational Rational)
    describe "lensGettingProducesValidOnGen" $
        it "holds for (_2) for special generators" $
        lensGettingProducesValidOnGen
            ((_2) :: Lens (Rational, Rational) (Rational, Rational) Rational Rational)
            ((,) <$> (negate . abs <$> genValid) <*> (negate . abs <$> genValid))
            (const [])
    describe "lensSettingProducesValidOnValid" $
        it "holds for (_2) for doubles" $
        lensSettingProducesValidOnValid
            ((_2) :: Lens (Rational, Rational) (Rational, Rational) Rational Rational)
    describe "lensSettingProducesValid" $
        it "holds for (_2) for ints" $
        lensSettingProducesValid ((_2) :: Lens (Int, Int) (Int, Int) Int Int)
    describe "lensSettingProducesValidOnArbitrary" $
        it "holds for (_2) for doubles" $
        lensSettingProducesValidOnArbitrary
            ((_2) :: Lens (Rational, Rational) (Rational, Rational) Rational Rational)
    describe "lensSettingProducesValidOnGen" $
        it "holds for (_2) for special generators" $
        lensSettingProducesValidOnGen
            ((_2) :: Lens (Rational, Rational) (Rational, Rational) Rational Rational)
            (abs <$> genValid)
            (filter (0.0 >=) . shrinkValid)
            ((,) <$> (negate . abs <$> genValid) <*> (negate . abs <$> genValid))
            (const [])
    describe "myBoolLens" $
        lensSpecOnValid myBoolLens
    describe "myRationalLens" $
        lensSpecOnValid myRationalLens

data MyRecord = MyRecord
    { myBool :: Bool
    , myRational :: Rational
    } deriving (Show, Eq, Generic)

instance Validity MyRecord

instance GenUnchecked MyRecord

instance GenValid MyRecord

myBoolLens :: Lens' MyRecord Bool
myBoolLens = lens myBool $ \mr b -> mr {myBool = b}

myRationalLens :: Lens' MyRecord Rational
myRationalLens = lens myRational $ \mr r -> mr {myRational = r}