(.require
 [library
  [lux (.except)
   ["_" test (.only Test)]
   [abstract
    [monad (.only do)]
    [\\specification
     ["$[0]" functor (.only Injection Comparison)]
     ["$[0]" monad]]]
   [control
    ["[0]" pipe]
    ["[0]" try (.use "[1]#[0]" functor)]
    ["[0]" exception (.only exception)]]
   [data
    ["[0]" text (.use "[1]#[0]" equivalence)
     ["%" \\format (.only format)]]]
   [math
    ["[0]" random]
    [number
     ["n" nat]
     ["i" int]]]]]
 [\\library
  ["[0]" / (.only)
   [//
    [meta
     ["[0]" archive]]]]])

(def (injection value)
  (All (_ s) (Injection (/.Operation s)))
  (function (_ state)
    {try.#Success [state value]}))

(def (comparison init)
  (All (_ s) (-> s (Comparison (/.Operation s))))
  (function (_ == left right)
    (case [(/.result init left)
           (/.result init right)]
      [{try.#Success left}
       {try.#Success right}]
      (== left right)

      _
      false)))

(exception oops)

(def test|error
  Test
  (do [! random.monad]
    [state random.nat
     expected random.int
     expected_error (random.lower_case 1)]
    (all _.and
         (_.coverage [/.failure]
           (|> (/.failure expected_error)
               (/.result state)
               (pipe.case {try.#Failure actual_error}
                 (same? expected_error actual_error)

                 _
                 false)))
         (_.coverage [/.lifted]
           (and (|> (/.lifted {try.#Failure expected_error})
                    (/.result state)
                    (pipe.case {try.#Failure actual_error}
                      (same? expected_error actual_error)

                      _
                      false))
                (|> (/.lifted {try.#Success expected})
                    (at /.functor each (same? expected))
                    (/.result state)
                    (try.else false))))
         (_.coverage [/.except]
           (|> (/.except ..oops [])
               (/.result state)
               (pipe.case {try.#Failure error}
                 (exception.match? ..oops error)

                 _
                 false)))
         (_.coverage [/.assertion]
           (and (|> (/.assertion ..oops [] false)
                    (/.result state)
                    (pipe.case {try.#Failure error}
                      (exception.match? ..oops error)

                      _
                      false))
                (|> (/.assertion ..oops [] true)
                    (/.result state)
                    (pipe.case {try.#Success _}
                      true

                      _
                      false))))
         )))

(def test|state
  Test
  (do [! random.monad]
    [state random.nat
     dummy random.nat
     expected random.int]
    (all _.and
         (_.coverage [/.state]
           (|> /.state
               (at /.functor each (same? state))
               (/.result state)
               (try.else false)))
         (_.coverage [/.with]
           (|> (do /.monad
                 [_ (/.with state)]
                 /.state)
               (at /.functor each (same? state))
               (/.result dummy)
               (try.else false)))
         (_.coverage [/.sub]
           (|> (/.sub [(at n.hex encoded)
                       (function (_ new old)
                         (|> new (at n.hex decoded) (try.else dummy)))]
                      (do /.monad
                        [state/hex /.state]
                        (in (|> state
                                (at n.hex encoded)
                                (text#= state/hex)))))
               (/.result' state)
               (pipe.case {try.#Success [state' verdict]}
                 (and verdict
                      (n.= state state'))

                 _
                 false)))
         )))

(def test|operation
  Test
  (do [! random.monad]
    [state random.nat
     expected random.int]
    (all _.and
         (_.for [/.functor]
                ($functor.spec ..injection (..comparison state) /.functor))
         (_.for [/.monad]
                ($monad.spec ..injection (..comparison state) /.monad))

         (_.coverage [/.result]
           (|> (at /.monad in expected)
               (/.result state)
               (pipe.case {try.#Success actual}
                 (same? expected actual)

                 _
                 false)))
         (_.coverage [/.result']
           (|> (at /.monad in expected)
               (/.result' state)
               (pipe.case {try.#Success [state' actual]}
                 (and (same? state state')
                      (same? expected actual))

                 _
                 false)))
         ..test|state
         ..test|error
         )))

(def test|phase
  Test
  (do [! random.monad]
    [state/0 random.nat
     state/1 random.rev
     expected random.int]
    (all _.and
         (_.coverage [/.identity]
           (|> (/.identity archive.empty expected)
               (/.result state/0)
               (try#each (same? expected))
               (try.else false)))
         (_.coverage [/.composite]
           (let [phase (/.composite (is (/.Phase Nat Int Frac)
                                        (function (_ archive input)
                                          (at /.monad in (i.frac input))))
                                    (is (/.Phase Rev Frac Text)
                                        (function (_ archive input)
                                          (at /.monad in (%.frac input)))))]
             (|> (phase archive.empty expected)
                 (/.result' [state/0 state/1])
                 (pipe.case {try.#Success [[state/0' state/1'] actual]}
                   (and (text#= (%.frac (i.frac expected)) actual)
                        (same? state/0 state/0')
                        (same? state/1 state/1'))

                   _
                   false))))
         )))

(def .public test
  Test
  (<| (_.covering /._)
      (all _.and
           (_.for [/.Operation]
                  ..test|operation)
           (_.for [/.Phase]
                  ..test|phase)
           )))