mirror of
https://github.com/github/semantic.git
synced 2024-12-21 05:41:54 +03:00
64 lines
2.2 KiB
Haskell
64 lines
2.2 KiB
Haskell
{-# LANGUAGE OverloadedLists, TypeOperators #-}
|
|
|
|
module Reprinting.Spec where
|
|
|
|
import SpecHelpers hiding (project, inject)
|
|
|
|
import Data.Functor.Foldable (embed, cata)
|
|
import qualified Data.Language as Language
|
|
import qualified Data.Syntax.Literal as Literal
|
|
import Data.Algebra
|
|
import Reprinting.Tokenize
|
|
import Reprinting.Pipeline
|
|
import Data.Sum
|
|
import Data.Foldable
|
|
import Semantic.IO
|
|
import Semantic.Util.Rewriting hiding (parseFile)
|
|
import Data.Blob
|
|
import Language.JSON.PrettyPrint
|
|
import Language.Ruby.PrettyPrint
|
|
import Language.Python.PrettyPrint
|
|
import qualified Data.Machine as Machine
|
|
|
|
spec :: Spec
|
|
spec = describe "reprinting" $ do
|
|
context "JSON" $ do
|
|
let path = "test/fixtures/javascript/reprinting/map.json"
|
|
(src, tree) <- runIO $ do
|
|
src <- blobSource <$> readBlobFromPath (File path Language.JSON)
|
|
tree <- parseFile jsonParser path
|
|
pure (src, tree)
|
|
|
|
describe "tokenization" $ do
|
|
|
|
it "should pass over a pristine tree" $ do
|
|
let tagged = mark Unmodified tree
|
|
let toks = Machine.run $ tokenizing src tagged
|
|
toks `shouldSatisfy` not . null
|
|
head toks `shouldSatisfy` isControl
|
|
last toks `shouldSatisfy` isChunk
|
|
|
|
it "should emit control tokens but only 1 chunk for a wholly-modified tree" $ do
|
|
let toks = Machine.run $ tokenizing src (mark Refactored tree)
|
|
for_ @[] [TList, THash] $ \t -> do
|
|
toks `shouldSatisfy` elem (TControl (Enter t))
|
|
toks `shouldSatisfy` elem (TControl (Exit t))
|
|
|
|
describe "pipeline" $ do
|
|
|
|
it "should roundtrip exactly over a pristine tree" $ do
|
|
let tagged = mark Unmodified tree
|
|
let printed = runReprinter src defaultJSONPipeline tagged
|
|
printed `shouldBe` Right src
|
|
|
|
it "should roundtrip exactly over a wholly-modified tree" $ do
|
|
let tagged = mark Refactored tree
|
|
let printed = runReprinter src defaultJSONPipeline tagged
|
|
printed `shouldBe` Right src
|
|
|
|
it "should be able to parse the output of a refactor" $ do
|
|
let tagged = increaseNumbers (mark Refactored tree)
|
|
let (Right printed) = runReprinter src defaultJSONPipeline tagged
|
|
tree' <- runTask (parse jsonParser (Blob printed path Language.JSON))
|
|
length tree' `shouldSatisfy` (/= 0)
|