mirror of
https://github.com/github/semantic.git
synced 2024-12-18 20:31:55 +03:00
50 lines
2.2 KiB
Haskell
50 lines
2.2 KiB
Haskell
{-# LANGUAGE GADTs, PackageImports #-}
|
|
|
|
module Graphing.Calls.Spec ( spec ) where
|
|
|
|
import Prelude hiding (readFile)
|
|
import SpecHelpers
|
|
|
|
import Algebra.Graph
|
|
|
|
import Control.Effect.Parse
|
|
import "semantic" Data.Graph (Graph (..), topologicalSort)
|
|
import Data.Graph.ControlFlowVertex
|
|
import qualified Data.Language as Language
|
|
import qualified Data.Map as Map
|
|
import Data.Maybe (fromJust)
|
|
import Semantic.Graph
|
|
import qualified System.Path as Path
|
|
|
|
callGraphPythonProject :: Path.RelFile -> IO (Semantic.Graph.Graph ControlFlowVertex)
|
|
callGraphPythonProject path = runTaskOrDie $ do
|
|
let proxy = Proxy @'Language.Python
|
|
lang = Language.Python
|
|
SomeParser parser <- pure . fromJust $! parserForLanguage analysisParsers Language.Python
|
|
blob <- readBlobFromFile' (fileForTypedPath path)
|
|
package <- fmap snd <$> parsePackage parser (Project (Path.toString (Path.takeDirectory path)) [blob] lang [])
|
|
modules <- topologicalSort <$> runImportGraphToModules proxy package
|
|
runCallGraph proxy False modules package
|
|
|
|
spec :: Spec
|
|
spec = describe "call graphing" $ do
|
|
|
|
let needs r v = unGraph r `shouldSatisfy` hasVertex v
|
|
|
|
it "should work for a simple example" $ do
|
|
res <- callGraphPythonProject (Path.relFile "test/fixtures/python/graphing/simple/simple.py")
|
|
res `needs` Variable "magnus" "simple.py" (Span (Pos 4 1) (Pos 4 7))
|
|
|
|
it "should evaluate both sides of an if-statement" $ do
|
|
res <- callGraphPythonProject (Path.relFile "test/fixtures/python/graphing/conditional/conditional.py")
|
|
res `needs` Variable "merle" "conditional.py" (Span (Pos 5 5) (Pos 5 10))
|
|
res `needs` Variable "taako" "conditional.py" (Span (Pos 8 5) (Pos 8 10))
|
|
|
|
it "should continue even when a type error is encountered" $ do
|
|
res <- callGraphPythonProject (Path.relFile "test/fixtures/python/graphing/typeerror/typeerror.py")
|
|
res `needs` Variable "lup" "typeerror.py" (Span (Pos 5 1) (Pos 5 4))
|
|
|
|
it "should continue when an unbound variable is encountered" $ do
|
|
res <- callGraphPythonProject (Path.relFile "test/fixtures/python/graphing/unbound/unbound.py")
|
|
res `needs` Variable "lucretia" "unbound.py" (Span (Pos 5 1) (Pos 5 9))
|