mirror of
https://github.com/github/semantic.git
synced 2024-12-18 20:31:55 +03:00
48 lines
1.7 KiB
Haskell
48 lines
1.7 KiB
Haskell
{-# LANGUAGE PackageImports #-}
|
|
|
|
module Graphing.Calls.Spec ( spec ) where
|
|
|
|
import Prelude hiding (readFile)
|
|
import Prologue
|
|
import SpecHelpers hiding (readFile)
|
|
|
|
import Algebra.Graph
|
|
import Data.List (uncons)
|
|
|
|
import "semantic" Data.Graph (Graph (..), topologicalSort)
|
|
import Data.Graph.Vertex
|
|
import qualified Data.Language as Language
|
|
import Semantic.Config (defaultOptions)
|
|
import Semantic.Graph
|
|
import Semantic.IO
|
|
|
|
callGraphPythonProject paths = runTaskWithOptions defaultOptions $ do
|
|
let proxy = Proxy @'Language.Python
|
|
let lang = Language.Python
|
|
blobs <- catMaybes <$> traverse readFile (flip File lang <$> paths)
|
|
package <- parsePackage pythonParser (Project (takeDirectory (maybe "/" fst (uncons paths))) blobs lang [])
|
|
modules <- topologicalSort <$> runImportGraph proxy package
|
|
runCallGraph proxy False modules package
|
|
|
|
spec :: Spec
|
|
spec = describe "call graphing" $ do
|
|
|
|
let needs r n = unGraph r `shouldSatisfy` hasVertex (Variable n)
|
|
|
|
it "should work for a simple example" $ do
|
|
res <- callGraphPythonProject ["test/fixtures/python/graphing/simple/simple.py"]
|
|
res `needs` "magnus"
|
|
|
|
it "should evaluate both sides of an if-statement" $ do
|
|
res <- callGraphPythonProject ["test/fixtures/python/graphing/conditional/conditional.py"]
|
|
res `needs` "merle"
|
|
res `needs` "taako"
|
|
|
|
it "should continue even when a type error is encountered" $ do
|
|
res <- callGraphPythonProject ["test/fixtures/python/graphing/typeerror/typeerror.py"]
|
|
res `needs` "lup"
|
|
|
|
it "should continue when an unbound variable is encountered" $ do
|
|
res <- callGraphPythonProject ["test/fixtures/python/graphing/unbound/unbound.py"]
|
|
res `needs` "lucretia"
|