diff options
Diffstat (limited to 'specs/TestUtils.hs')
| -rw-r--r-- | specs/TestUtils.hs | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/specs/TestUtils.hs b/specs/TestUtils.hs index c98899d..76a185d 100644 --- a/specs/TestUtils.hs +++ b/specs/TestUtils.hs @@ -3,6 +3,7 @@ module TestUtils where import Chelleport.AppShell (MonadAppShell (..)) import Chelleport.Control (MonadControl (..)) import Chelleport.Draw (MonadDraw (..)) +import Chelleport.OCR (MonadOCR (..)) import Control.Monad (void) import Control.Monad.IO.Class (MonadIO) import Control.Monad.State (MonadState (state), StateT (runStateT)) @@ -11,7 +12,7 @@ import Foreign.C (CInt) import Mock import Test.Hspec -$(generateMock [''MonadDraw, ''MonadControl, ''MonadAppShell]) +$(generateMock [''MonadDraw, ''MonadControl, ''MonadAppShell, ''MonadOCR]) newtype MockCalls = MockCalls {calls :: [Call]} deriving (Show) @@ -53,6 +54,7 @@ instance (MonadIO m) => MonadControl (TestM m) where instance (MonadIO m) => MonadDraw (TestM m) where drawLine p1 p2 = registerMockCall $ Mock_drawLine p1 p2 + fillRect p size = registerMockCall $ Mock_fillRect p size drawText p color text = (fromIntegral $ mockTextWidth * Text.length text, 0) <$ registerMockCall (Mock_drawText p color text) drawCircle radius p = registerMockCall $ Mock_drawCircle radius p setDrawColor color = registerMockCall $ Mock_setDrawColor color @@ -63,3 +65,6 @@ instance (MonadIO m) => MonadAppShell (TestM m) where hideWindow = registerMockCall Mock_hideWindow showWindow = registerMockCall Mock_showWindow shutdownApp = registerMockCall Mock_shutdownApp + +instance (MonadIO m) => MonadOCR (TestM m) where + getWordsOnScreen = [] <$ registerMockCall Mock_getWordsOnScreen |