[MOCK] More credible count.

src/Gargantext/API/Count.hs

@@ -32,7 +32,7 @@ import Test.QuickCheck.Arbitrary
 import Test.QuickCheck (elements)
 import Data.List (repeat,permutations)
 -----------------------------------------------------------------------
-type CountAPI = Post '[JSON] [Count]
+type CountAPI = Post '[JSON] Counts
 
 -----------------------------------------------------------------------
 data Scraper = Pubmed | Hal | IsTex | Isidore
@@ -78,60 +78,59 @@ instance Arbitrary Query where
 
 -----------------------------------------------------------------------
 -----------------------------------------------------------------------
-data ErrorMessage = ErrorMessage Text
-        deriving (Eq, Show, Generic)
-
-errorMessages :: [ErrorMessage]
-errorMessages = map (\m -> ErrorMessage (pack m)) $ [ "Ill formed query             "
-                                                    , "API connexion error          "
-                                                    , "Internal Gargantext Error    "
-                                                    , "Connexion to Gargantext Error"
-                                                   -- , "Token has expired            "
-                                                    ] <> take 100 ( repeat ("No Error"))
 
-instance Arbitrary ErrorMessage where
-    arbitrary = elements errorMessages
+type Error  = Text
+type Errors = [Error]
 
-instance FromJSON ErrorMessage
-instance ToJSON   ErrorMessage
-
------------------------------------------------------------------------
-data Error = Error { error_message :: ErrorMessage
-                   , error_code    :: Int
-                   }
+data Message = Message Integer Errors
         deriving (Eq, Show, Generic)
-instance FromJSON Error
-instance ToJSON   Error
 
-errorCodes :: [Int]
-errorCodes = [200,300,400,500]
+toMessage :: [(Integer, [Text])] -> [Message]
+toMessage = map (\(c,es) -> Message c es)
 
-errors :: [Error]
-errors =  [ Error m c | m <- errorMessages
-                      , c <- errorCodes
-                      ]
+messages :: [Message]
+messages =  toMessage $ [ (400, ["Ill formed query             "])
+                        , (300, ["API connexion error          "])
+                        , (300, ["Internal Gargantext Error    "])
+                        , (300, ["Connexion to Gargantext Error"])
+                        , (300, ["Token has expired            "])
+                        ] <> take 10 ( repeat (200, [""]))
+
+instance Arbitrary Message where
+    arbitrary = elements messages
 
-instance Arbitrary Error where
-    arbitrary = elements errors
+instance FromJSON Message
+instance ToJSON   Message
 
 -----------------------------------------------------------------------
 -----------------------------------------------------------------------
+data Counts = Counts [Count]
+                   deriving (Eq, Show, Generic)
+
+instance FromJSON Counts
+instance ToJSON   Counts
+
 data Count = Count { count_name    :: Scraper
                    , count_count   :: Maybe Int
-                   , count_errors :: Maybe [Error]
+                   , count_message :: Maybe Message
                    }
                    deriving (Eq, Show, Generic)
 
 instance FromJSON Count
 instance ToJSON   Count
 
-instance Arbitrary Count where
-    arbitrary = elements [ Count n (Just c) (Just [e]) | n <- scrapers
-                                                       , c <- [100..1000]
-                                                       , e <- errors
-                                                       ]
-
+instance Arbitrary Counts where
+    arbitrary = elements $ select
+                         $ map Counts 
+                         $ map (\xs -> zipWith (\s (c,m) -> Count s c m) scrapers xs) 
+                         $ chunkAlong (length scrapers) 1 $  (map filter' countOrErrors)
+        where
+            select xs = (take 10 xs) <> (take 10 $ drop 100 xs)
+            countOrErrors = [ (c,e) | c <- [500..1000], e <- reverse messages]
+            filter' (c,e) = case e of
+                              Message 200 _ -> (Just c , Nothing     )
+                              message       -> (Nothing, Just message)
 
 -----------------------------------------------------------------------
-count :: Query -> Handler [Count]
+count :: Query -> Handler Counts
 count _ = undefined