semantic/test/Matching/Go/Spec.hs

{-# LANGUAGE TypeOperators #-}

module Matching.Go.Spec (spec) where

import           Control.Abstract.Matching
import           Data.Abstract.Module
import           Data.List
import           Data.Sum
import qualified Data.Syntax.Declaration as Decl
import qualified Data.Syntax.Literal as Lit
import qualified Data.Syntax.Statement as Stmt
import           Data.Text (Text)
import           SpecHelpers

-- This gets the Text contents of all integers
integerMatcher :: (Lit.Integer :< fs) => Matcher (Term (Sum fs) ann) Text
integerMatcher = match Lit.integerContent target

-- This matches all for-loops with its index variable new variable bound to 0,
-- e.g. `for i := 0; i < 10; i++`
loopMatcher :: ( Stmt.For :< fs
               , Stmt.Assignment :< fs
               , Lit.Integer :< fs)
            => TermMatcher fs ann
loopMatcher = target <* go where
  go = match Stmt.forBefore $
         match Stmt.assignmentValue $
            match Lit.integerContent $
               ensure (== "0")

spec :: Spec
spec = describe "matching/go" $ do
  it "extracts integers" $ do
    parsed <- parseFile goParser "test/fixtures/go/matching/integers.go"
    let matched = runMatcher integerMatcher parsed
    sort matched `shouldBe` ["1", "2", "3"]

  it "counts for loops" $ do
    parsed <- parseFile goParser "test/fixtures/go/matching/for.go"
    let matched = runMatcher @[] loopMatcher parsed
    length matched `shouldBe` 2