improve readability of test code
This commit is contained in:
Hongrui Fang
parent
3b15fedc26
commit
2d3f8f0463
2 changed files with 116 additions and 118 deletions
|
|
@ -36,7 +36,6 @@ import Agora.Stake (
|
|||
StakeRedeemer (PermitVote, WitnessStake),
|
||||
)
|
||||
import Agora.Stake.Scripts (stakeValidator)
|
||||
import Control.Monad (join)
|
||||
import Data.Default.Class (Default (def))
|
||||
import Data.Tagged (Tagged (Tagged), untag)
|
||||
import PlutusLedgerApi.V1 (ScriptContext (..), ScriptPurpose (..))
|
||||
|
|
@ -341,112 +340,110 @@ specs =
|
|||
(Spending Proposal.proposalRef)
|
||||
)
|
||||
]
|
||||
, group
|
||||
"unlocking"
|
||||
$ map
|
||||
( \pc ->
|
||||
group
|
||||
(show pc <> " proposals")
|
||||
[ group
|
||||
"legal"
|
||||
[ group
|
||||
"retract votes and unlock stake while voting"
|
||||
[ UnlockStake.mkProposalValidatorTestCase
|
||||
( UnlockStake.UnlockStakeParameters
|
||||
pc
|
||||
UnlockStake.Voter
|
||||
True
|
||||
VotingReady
|
||||
)
|
||||
True
|
||||
]
|
||||
, group
|
||||
"unlock the stake that has been used to create the proposal"
|
||||
[ UnlockStake.mkProposalValidatorTestCase
|
||||
( UnlockStake.UnlockStakeParameters
|
||||
pc
|
||||
UnlockStake.Creator
|
||||
False
|
||||
Finished
|
||||
)
|
||||
True
|
||||
]
|
||||
, group "unlock stake after voting" $
|
||||
map
|
||||
( \ps ->
|
||||
, group "unlocking" $ do
|
||||
proposalCount <- [1, 42]
|
||||
|
||||
let legalGroup = group "legal" $ do
|
||||
let voterRetractVotesAndUnlockStakeWhileVoting =
|
||||
UnlockStake.mkProposalValidatorTestCase
|
||||
( UnlockStake.UnlockStakeParameters
|
||||
pc
|
||||
UnlockStake.Voter
|
||||
False
|
||||
ps
|
||||
)
|
||||
UnlockStake.UnlockStakeParameters
|
||||
{ UnlockStake.proposalCount = proposalCount
|
||||
, UnlockStake.stakeUsage = UnlockStake.Voter
|
||||
, UnlockStake.retractVotes = True
|
||||
, UnlockStake.proposalStatus = VotingReady
|
||||
}
|
||||
True
|
||||
)
|
||||
[Finished, Locked]
|
||||
]
|
||||
, group
|
||||
"illegal"
|
||||
[ group "retract votes while the proposal is not voting ready" $
|
||||
map
|
||||
( \ps ->
|
||||
creatorUnlockStakeWhileFinished =
|
||||
UnlockStake.mkProposalValidatorTestCase
|
||||
( UnlockStake.UnlockStakeParameters
|
||||
pc
|
||||
UnlockStake.Voter
|
||||
UnlockStake.UnlockStakeParameters
|
||||
{ UnlockStake.proposalCount = proposalCount
|
||||
, UnlockStake.stakeUsage = UnlockStake.Creator
|
||||
, UnlockStake.retractVotes = False
|
||||
, UnlockStake.proposalStatus = Finished
|
||||
}
|
||||
True
|
||||
ps
|
||||
)
|
||||
False
|
||||
)
|
||||
[Draft, Locked, Finished]
|
||||
, group "irrelevant stake" $
|
||||
join $
|
||||
map
|
||||
( \rv ->
|
||||
map
|
||||
( \ps ->
|
||||
|
||||
let voterUnlockStakeAfterVoting = group "voter unlocks stake after voting" $ do
|
||||
status <- [Finished, Locked]
|
||||
|
||||
pure $
|
||||
UnlockStake.mkProposalValidatorTestCase
|
||||
( UnlockStake.UnlockStakeParameters
|
||||
pc
|
||||
UnlockStake.Irrelevant
|
||||
rv
|
||||
ps
|
||||
)
|
||||
False
|
||||
)
|
||||
[Draft, VotingReady, Locked, Finished]
|
||||
)
|
||||
[True, False]
|
||||
, group "unlock stake that has been used to create the proposal before finished" $
|
||||
map
|
||||
( \ps ->
|
||||
UnlockStake.mkProposalValidatorTestCase
|
||||
( UnlockStake.UnlockStakeParameters
|
||||
pc
|
||||
UnlockStake.Creator
|
||||
False
|
||||
ps
|
||||
)
|
||||
False
|
||||
)
|
||||
[Draft, VotingReady, Locked]
|
||||
, group "creator stake retract votes" $
|
||||
map
|
||||
( \ps ->
|
||||
UnlockStake.mkProposalValidatorTestCase
|
||||
( UnlockStake.UnlockStakeParameters
|
||||
pc
|
||||
UnlockStake.Creator
|
||||
UnlockStake.UnlockStakeParameters
|
||||
{ UnlockStake.proposalCount = proposalCount
|
||||
, UnlockStake.stakeUsage = UnlockStake.Voter
|
||||
, UnlockStake.retractVotes = False
|
||||
, UnlockStake.proposalStatus = status
|
||||
}
|
||||
True
|
||||
ps
|
||||
)
|
||||
|
||||
[ voterRetractVotesAndUnlockStakeWhileVoting
|
||||
, creatorUnlockStakeWhileFinished
|
||||
, voterUnlockStakeAfterVoting
|
||||
]
|
||||
|
||||
let illegalGroup = group "illegal" $ do
|
||||
let retractsVotesWhileNotVotingReady =
|
||||
group "voter retracts votes while not voting" $ do
|
||||
status <- [Draft, Locked, Finished]
|
||||
|
||||
pure $
|
||||
UnlockStake.mkProposalValidatorTestCase
|
||||
UnlockStake.UnlockStakeParameters
|
||||
{ UnlockStake.proposalCount = proposalCount
|
||||
, UnlockStake.stakeUsage = UnlockStake.Voter
|
||||
, UnlockStake.retractVotes = True
|
||||
, UnlockStake.proposalStatus = status
|
||||
}
|
||||
False
|
||||
)
|
||||
[Draft, VotingReady, Locked, Finished]
|
||||
]
|
||||
]
|
||||
)
|
||||
[1, 25]
|
||||
|
||||
unlockIrrelevantStake =
|
||||
group "unlock an irrelevant stake" $ do
|
||||
status <- [Draft, VotingReady, Locked, Finished]
|
||||
shouldRetractVotes <- [True, False]
|
||||
|
||||
pure $
|
||||
UnlockStake.mkProposalValidatorTestCase
|
||||
UnlockStake.UnlockStakeParameters
|
||||
{ UnlockStake.proposalCount = proposalCount
|
||||
, UnlockStake.stakeUsage = UnlockStake.Irrelevant
|
||||
, UnlockStake.retractVotes = shouldRetractVotes
|
||||
, UnlockStake.proposalStatus = status
|
||||
}
|
||||
False
|
||||
|
||||
unlockCreatorStakeBeforeFinished =
|
||||
group "unlock creator stake before finished" $ do
|
||||
status <- [Draft, VotingReady, Locked]
|
||||
|
||||
pure $
|
||||
UnlockStake.mkProposalValidatorTestCase
|
||||
UnlockStake.UnlockStakeParameters
|
||||
{ UnlockStake.proposalCount = proposalCount
|
||||
, UnlockStake.stakeUsage = UnlockStake.Creator
|
||||
, UnlockStake.retractVotes = False
|
||||
, UnlockStake.proposalStatus = status
|
||||
}
|
||||
False
|
||||
retractVotesWithCreatorStake =
|
||||
group "creator stake retracts votes" $ do
|
||||
status <- [Draft, VotingReady, Locked, Finished]
|
||||
|
||||
pure $
|
||||
UnlockStake.mkProposalValidatorTestCase
|
||||
UnlockStake.UnlockStakeParameters
|
||||
{ UnlockStake.proposalCount = proposalCount
|
||||
, UnlockStake.stakeUsage = UnlockStake.Creator
|
||||
, UnlockStake.retractVotes = True
|
||||
, UnlockStake.proposalStatus = status
|
||||
}
|
||||
False
|
||||
|
||||
[ retractsVotesWhileNotVotingReady
|
||||
, unlockIrrelevantStake
|
||||
, unlockCreatorStakeBeforeFinished
|
||||
, retractVotesWithCreatorStake
|
||||
]
|
||||
|
||||
[legalGroup, illegalGroup]
|
||||
]
|
||||
]
|
||||
|
|
|
|||
|
|
@ -27,7 +27,7 @@ import Data.ByteString.Lazy qualified as ByteString.Lazy
|
|||
--------------------------------------------------------------------------------
|
||||
|
||||
import Data.Bifunctor (second)
|
||||
import Data.List (sortBy)
|
||||
import Data.List (sortOn)
|
||||
import Plutarch.Crypto (pblake2b_256)
|
||||
import PlutusLedgerApi.V1.Interval qualified as PlutusTx
|
||||
import PlutusLedgerApi.V1.Scripts (Datum (Datum), DatumHash (DatumHash))
|
||||
|
|
@ -93,15 +93,16 @@ updateMap f k =
|
|||
--------------------------------------------------------------------------------
|
||||
|
||||
sortMap :: forall k v. Ord k => AssocMap.Map k v -> AssocMap.Map k v
|
||||
sortMap (AssocMap.toList -> l) =
|
||||
AssocMap.fromList $
|
||||
sortBy
|
||||
( \(k1, _)
|
||||
(k2, _) -> compare k1 k2
|
||||
)
|
||||
l
|
||||
sortMap =
|
||||
AssocMap.fromList
|
||||
. sortOn fst
|
||||
. AssocMap.toList
|
||||
|
||||
sortValue :: Value -> Value
|
||||
sortValue (AssocMap.toList . getValue -> l) =
|
||||
let innerSorted = second sortMap <$> l
|
||||
in Value $ sortMap $ AssocMap.fromList innerSorted
|
||||
sortValue =
|
||||
Value
|
||||
. sortMap
|
||||
. AssocMap.fromList
|
||||
. fmap (second sortMap)
|
||||
. AssocMap.toList
|
||||
. getValue
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue