{-# LANGUAGE OverloadedStrings #-}

module GraphingSpec (spec) where

import HomeAssistant.Runtime.Graphing
  ( GraphSpec (..)
  , Range (..)
  , buildGraphArgs
  , graphs
  , lookupGraph
  , parseNameParam
  , parseRange
  , rangeStart
  )
import Test.Hspec

traffic :: GraphSpec
traffic = case lookupGraph graphs "traffic" of
  Just s  -> s
  Nothing -> error "traffic graph is missing from the table"

spec :: Spec
spec = do
  describe "parseRange" $ do
    it "accepts every suffix in the closed set" $
      map parseRange ["10m", "1h", "24h", "7d", "30d"]
        `shouldBe` [ Just TenMinutes
                   , Just OneHour
                   , Just OneDay
                   , Just OneWeek
                   , Just OneMonth
                   ]

    it "rejects everything outside the closed set" $
      map parseRange
        [ ""
        , "1h "
        , " 1h"
        , "1h; rm -rf /"
        , "end-1h"
        , "../../etc/passwd"
        , "1y"
        , "30m"
        , "1H"
        ]
        `shouldBe` replicate 9 Nothing

  describe "rangeStart" $ do
    it "renders rrdtool end- relative starts" $
      map rangeStart [TenMinutes, OneHour, OneDay, OneWeek, OneMonth]
        `shouldBe` ["end-10m", "end-1h", "end-24h", "end-7d", "end-30d"]

  describe "parseNameParam" $ do
    it "resolves the traffic graph only with a .png suffix" $ do
      fmap gsName (parseNameParam graphs "traffic.png") `shouldBe` Just "traffic"
      parseNameParam graphs "traffic" `shouldBe` Nothing

    it "rejects unknown, malformed, and traversing names" $
      map (parseNameParam graphs)
        [ "unknown.png"
        , "../traffic.png"
        , "traffic.png.png"
        , ""
        , "nope"
        ]
        `shouldBe` replicate 5 Nothing

  describe "buildGraphArgs" $ do
    it "builds the traffic argv for a range" $
      buildGraphArgs "hass-controller.rrd" traffic OneHour
        `shouldBe`
          [ "graph"
          , "-"
          , "--start", "end-1h"
          , "--title", "Hass triggers in / services out"
          , "--width", "800"
          , "--height", "250"
          , "--imgformat", "PNG"
          , "DEF:trig=hass-controller.rrd:hass_trigger_in:AVERAGE"
          , "CDEF:trig_min=trig,60,*"
          , "DEF:svc=hass-controller.rrd:hass_service_out:AVERAGE"
          , "CDEF:svc_min=svc,60,*"
          , "LINE2:trig_min#0066cc:triggers in/min"
          , "LINE2:svc_min#cc0000:services out/min"
          ]