mirror of
https://github.com/github/semantic.git
synced 2024-11-28 10:15:55 +03:00
188 lines
7.8 KiB
Haskell
188 lines
7.8 KiB
Haskell
{-# LANGUAGE FlexibleContexts, RecordWildCards, TypeApplications #-}
|
|
{-# OPTIONS_GHC -O1 #-}
|
|
module Main (main) where
|
|
|
|
import Control.Carrier.Parse.Measured
|
|
import Control.Concurrent.Async (forConcurrently)
|
|
import Control.Effect
|
|
import Control.Effect.Reader
|
|
import Control.Exception (displayException)
|
|
import qualified Control.Foldl as Foldl
|
|
import Control.Lens
|
|
import Control.Monad
|
|
import Control.Monad.Trans.Resource (ResIO, runResourceT)
|
|
import Data.Blob
|
|
import qualified Data.ByteString.Lazy.Char8 as BLC
|
|
import qualified Data.ByteString.Streaming.Char8 as ByteStream
|
|
import Data.Foldable
|
|
import Data.Function ((&))
|
|
import Data.Language (LanguageMode (..), PerLanguageModes (..))
|
|
import Data.List
|
|
import Data.Set (Set)
|
|
import qualified Data.Set as Set
|
|
import qualified Streaming.Prelude as Stream
|
|
import System.FilePath.Glob
|
|
import System.Path ((</>))
|
|
import qualified System.Path as Path
|
|
import qualified System.Process as Process
|
|
|
|
import Data.Flag
|
|
import Proto.Semantic as P hiding (Blob, BlobPair)
|
|
import Proto.Semantic_Fields as P
|
|
import Semantic.Api.Symbols (parseSymbols)
|
|
import Semantic.Config as Config
|
|
import Semantic.Task
|
|
import Semantic.Task.Files
|
|
|
|
import qualified Test.Tasty as Tasty
|
|
import qualified Test.Tasty.HUnit as HUnit
|
|
|
|
data LanguageExample
|
|
= LanguageExample
|
|
{ languageName :: String
|
|
, languageExtension :: String
|
|
, languageExampleDir :: Path.RelDir
|
|
, languageKnownFailuresTxt :: Maybe Path.RelFile
|
|
} deriving (Eq, Show)
|
|
|
|
le :: String -> String -> Path.RelDir -> Maybe Path.RelFile -> LanguageExample
|
|
le = LanguageExample
|
|
|
|
examples :: [LanguageExample]
|
|
examples =
|
|
[ le "python" ".py" examples (Just $ Path.relFile "script/known_failures.txt")
|
|
-- , le "ruby" ".rb" examples (Just $ Path.relFile "script/known_failures.txt")
|
|
-- , le "typescript" ".ts" examples (Just $ Path.relFile "typescript/script/known_failures.txt")
|
|
-- , le "typescript" ".tsx" examples (Just $ Path.relFile "typescript/script/known_failures.txt")
|
|
-- , le "typescript" ".js" examples Nothing -- parse JavaScript with TypeScript parser.
|
|
-- , le "go" ".go" examples (Just $ Path.relFile "script/known-failures.txt")
|
|
|
|
-- TODO: Java assignment errors need to be investigated
|
|
-- , le "java" ".java" "examples/guava" (Just "script/known_failures_guava.txt")
|
|
-- , le "java" ".java" "examples/elasticsearch" (Just "script/known_failures_elasticsearch.txt")
|
|
-- , le "java" ".java" "examples/RxJava" (Just "script/known_failures_RxJava.txt")
|
|
|
|
-- TODO: Haskell assignment errors need to be investigated
|
|
-- , le "haskell" ".hs" "examples/effects" (Just "script/known-failures-effects.txt")
|
|
-- , le "haskell" ".hs" "examples/postgrest" (Just "script/known-failures-postgrest.txt")
|
|
-- , le "haskell" ".hs" "examples/ivory" (Just "script/known-failures-ivory.txt")
|
|
|
|
-- , ("php", ".php") -- TODO: No parse-examples in tree-sitter yet
|
|
] where examples = Path.relDir "examples"
|
|
|
|
buildExamples :: TaskSession -> LanguageExample -> Path.RelDir -> IO Tasty.TestTree
|
|
buildExamples session lang tsDir = do
|
|
knownFailures <- knownFailuresForPath tsDir (languageKnownFailuresTxt lang)
|
|
files <- globDir1 (compile ("**/*" <> languageExtension lang)) (Path.toString (tsDir </> languageExampleDir lang))
|
|
let paths = Path.relFile <$> files
|
|
trees <- forConcurrently paths $ \file -> do
|
|
let name = "[" <> languageName lang <> "] " <> Path.toString file
|
|
pure . HUnit.testCaseSteps name $ \step -> do
|
|
-- Use alacarte language mode
|
|
step "a la carte"
|
|
alacarte <- runTask session (runParse (parseSymbolsFilePath aLaCarteLanguageModes file))
|
|
assertOK "a la carte" alacarte
|
|
|
|
-- Test out precise language mode
|
|
step "precise"
|
|
precise <- runTask session (runParse (parseSymbolsFilePath preciseLanguageModes file))
|
|
assertOK "precise" precise
|
|
|
|
-- Compare the two
|
|
step "compare"
|
|
assertMatch file knownFailures alacarte precise
|
|
|
|
pure (Tasty.testGroup (languageName lang) (take 100 trees))
|
|
|
|
where
|
|
assertOK msg = either (\e -> HUnit.assertFailure (msg <> " failed to parse" <> show e)) (refuteErrors msg)
|
|
refuteErrors msg a = case toList (a^.files) of
|
|
[x] | (e:_) <- toList (x^.errors) -> HUnit.assertFailure (msg <> " parse errors " <> show e)
|
|
_ -> pure ()
|
|
|
|
assertMatch filePath knownFailures a b = case (a, b) of
|
|
(Right a, Right b) -> case (toList (a^.files), toList (b^.files)) of
|
|
([x], [y]) | (e1:_) <- toList (x^.errors)
|
|
, (e2:_) <- toList (y^.errors)
|
|
-> if Set.member filePath knownFailures
|
|
then pure ()
|
|
else HUnit.assertFailure ("Parse errors (both) " <> show e1 <> show e2)
|
|
(_, [y]) | (e:_) <- toList (y^.errors)
|
|
-> HUnit.assertFailure ("Parse errors (precise) " <> show e)
|
|
([x], _) | (_e:_) <- toList (x^.errors)
|
|
-> pure () -- HUnit.assertFailure ("Parse errors (a la carte) " <> show e)
|
|
([x], [y]) -> do
|
|
HUnit.assertEqual "Expected paths to be equal" (x^.path) (y^.path)
|
|
let xSymbols = sort $ toListOf (symbols . traverse . symbol) x
|
|
ySymbols = sort $ toListOf (symbols . traverse . symbol) y
|
|
delta = xSymbols \\ ySymbols
|
|
msg = "Found in a la carte, but not precise: "
|
|
<> show delta
|
|
<> "\n"
|
|
<> "Found in precise but not a la carte: "
|
|
<> show (ySymbols \\ xSymbols)
|
|
<> "\n"
|
|
<> "Expected: " <> show xSymbols <> "\n"
|
|
<> "But got:" <> show ySymbols
|
|
|
|
HUnit.assertBool ("Expect symbols to be equal.\n" <> msg) (null delta)
|
|
pure ()
|
|
_ -> HUnit.assertFailure "Expected 1 file in each response"
|
|
(Left e1, Left e2) -> HUnit.assertFailure ("Unable to parse (both)" <> show (displayException e1) <> show (displayException e2))
|
|
(_, Left e) -> HUnit.assertFailure ("Unable to parse (precise)" <> show (displayException e))
|
|
(Left _e, _) -> pure () -- HUnit.assertFailure ("Unable to parse (a la carte)" <> show (displayException e))
|
|
|
|
aLaCarteLanguageModes :: PerLanguageModes
|
|
aLaCarteLanguageModes = PerLanguageModes
|
|
{ pythonMode = ALaCarte
|
|
}
|
|
|
|
preciseLanguageModes :: PerLanguageModes
|
|
preciseLanguageModes = PerLanguageModes
|
|
{ pythonMode = Precise
|
|
}
|
|
|
|
testOptions :: Config.Options
|
|
testOptions = defaultOptions
|
|
{ optionsFailOnWarning = flag FailOnWarning True
|
|
, optionsLogLevel = Nothing
|
|
}
|
|
|
|
main :: IO ()
|
|
main = withOptions testOptions $ \ config logger statter -> do
|
|
void $ Process.system "script/clone-example-repos"
|
|
|
|
let session = TaskSession config "-" False logger statter
|
|
|
|
allTests <- forConcurrently examples $ \lang@LanguageExample{..} -> do
|
|
let tsLang = Path.relDir ("tree-sitter-" <> languageName)
|
|
let tsDir = Path.relDir "tmp/haskell-tree-sitter" </> tsLang </> Path.relDir "vendor" </> tsLang
|
|
buildExamples session lang tsDir
|
|
|
|
Tasty.defaultMain $ Tasty.testGroup "parse-examples" allTests
|
|
|
|
knownFailuresForPath :: Path.RelDir -> Maybe Path.RelFile -> IO (Set Path.RelFile)
|
|
knownFailuresForPath _ Nothing = pure mempty
|
|
knownFailuresForPath tsDir (Just path)
|
|
= runResourceT
|
|
( ByteStream.readFile @ResIO (Path.toString (tsDir </> path))
|
|
& ByteStream.lines
|
|
& ByteStream.denull
|
|
& Stream.mapped ByteStream.toLazy
|
|
& Stream.filter ((/= '#') . BLC.head)
|
|
& Stream.map (Path.relFile . BLC.unpack)
|
|
& Foldl.purely Stream.fold_ Foldl.set
|
|
)
|
|
|
|
parseSymbolsFilePath ::
|
|
( Member (Error SomeException) sig
|
|
, Member Distribute sig
|
|
, Member Parse sig
|
|
, Member Files sig
|
|
, Carrier sig m
|
|
)
|
|
=> PerLanguageModes
|
|
-> Path.RelFile
|
|
-> m ParseTreeSymbolResponse
|
|
parseSymbolsFilePath languageModes path = readBlob (fileForTypedPath path) >>= runReader languageModes . parseSymbols . pure @[]
|