mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-15 17:31:56 +03:00
7e334e08a4
PR-URL: https://github.com/hasura/graphql-engine-mono/pull/8947 GitOrigin-RevId: 18e52c928e1df535579e2077b4af6c2ce92bdcef
81 lines
2.7 KiB
Haskell
81 lines
2.7 KiB
Haskell
-- | Tests for stuff under Hasura.Eventing hierarchy
|
|
module Hasura.EventingSpec (spec) where
|
|
|
|
import Control.Concurrent.STM.TVar
|
|
import Data.HashMap.Strict qualified as HashMap
|
|
import Data.Set qualified as Set
|
|
import Data.Time.Clock
|
|
import Hasura.Eventing.EventTrigger
|
|
import Hasura.Eventing.ScheduledTrigger
|
|
import Hasura.Prelude
|
|
import Hasura.RQL.Types.Common
|
|
import Hasura.RQL.Types.Eventing
|
|
import System.Cron.Parser
|
|
import Test.Hspec
|
|
|
|
spec :: Spec
|
|
spec = do
|
|
scheduleTriggersSpec
|
|
eventTriggersLockingUnlockingSpec
|
|
|
|
scheduleTriggersSpec :: Spec
|
|
scheduleTriggersSpec = do
|
|
-- https://hasura.io/docs/latest/graphql/core/scheduled-triggers/create-cron-trigger.html
|
|
--
|
|
-- FYI this is quite helpful for experimenting with cron expressions:
|
|
-- https://crontab.guru/
|
|
describe "cron" $ do
|
|
it "calculates future events sanely" $ do
|
|
cronTest
|
|
"* * * * *"
|
|
[ "2021-04-20 16:20:00 UTC",
|
|
"2021-04-20 16:21:00 UTC",
|
|
"2021-04-20 16:22:00 UTC"
|
|
]
|
|
|
|
cronTest
|
|
"5 0 * 8 *" -- “At 00:05 in August.”
|
|
[ "2021-08-01 00:05:00 UTC",
|
|
"2021-08-02 00:05:00 UTC",
|
|
"2021-08-03 00:05:00 UTC"
|
|
]
|
|
|
|
cronTest
|
|
"15 14 1 * *" -- “At 14:15 on day-of-month 1.”
|
|
[ "2021-05-01 14:15:00 UTC",
|
|
"2021-06-01 14:15:00 UTC",
|
|
"2021-07-01 14:15:00 UTC"
|
|
]
|
|
|
|
cronTest
|
|
"0 22 * * 1-5" -- “At 22:00 on every day-of-week from Monday through Friday.”
|
|
[ "2021-04-20 22:00:00 UTC",
|
|
"2021-04-21 22:00:00 UTC",
|
|
"2021-04-22 22:00:00 UTC"
|
|
]
|
|
where
|
|
-- A few unit tests for schedule projection into the future, from an
|
|
-- arbitrary time:
|
|
now = read "2021-04-20 16:19:19.450 UTC" :: UTCTime -- Tuesday
|
|
cronTest cronExpr expected = case parseCronSchedule cronExpr of
|
|
Left e -> error $ "Fix test: " <> show e
|
|
Right sched ->
|
|
generateScheduleTimes now 3 sched
|
|
`shouldBe` map read expected
|
|
|
|
eventTriggersLockingUnlockingSpec :: Spec
|
|
eventTriggersLockingUnlockingSpec = do
|
|
describe "check locking and unlocking of events" $ do
|
|
lockedEventsContainer <- runIO $ newTVarIO mempty
|
|
let eventId = EventId "a7aece90-4a6a-4a8c-ad9d-da5f25dacad9"
|
|
|
|
it "locks events correctly" $ do
|
|
saveLockedEventTriggerEvents SNDefault [eventId] lockedEventsContainer
|
|
currentLockedEvents <- readTVarIO lockedEventsContainer
|
|
currentLockedEvents `shouldBe` (HashMap.singleton SNDefault (Set.singleton eventId))
|
|
|
|
it "unlocks (removes) an event correctly from the locked events" $ do
|
|
removeEventTriggerEventFromLockedEvents SNDefault eventId lockedEventsContainer
|
|
currentLockedEvents <- readTVarIO lockedEventsContainer
|
|
currentLockedEvents `shouldBe` (HashMap.singleton SNDefault mempty)
|