require-0.4.6: test-suite/Main.hs

import qualified Data.Text as Text
import Relude
import qualified Require
import qualified Test.Tasty
import Test.Tasty.Hspec

main :: IO ()
main = do
  test <- testSpec "require" spec
  Test.Tasty.defaultMain test

spec :: Spec
spec = parallel $ do
  it "transforms the 'require' keyword into a properly qualified import" $ do
    let input = "require Data.Text"
    let expected = "import qualified Data.Text as Text"
    let actual = Require.transform False (Require.FileName "Foo.hs") "" input
    expected `Text.isInfixOf` actual
  it "imports the type based on the module" $ do
    let input = "require Data.Text"
    let expected = "import Data.Text (Text)"
    let actual = Require.transform False (Require.FileName "Foo.hs") "" input
    expected `Text.isInfixOf` actual
  it "keeps the rest of the content intact" $ do
    let input = "module Foo where\nrequire Data.Text\nfoo = 42"
    let expectedStart = "{-# LINE 1"
    let expectedModule = "module Foo where"
    let expectedTypeImport = "import Data.Text (Text)"
    let expectedQualifiedImport = "import qualified Data.Text as Text"
    let expectedContent = "foo = 42\n"
    let actual = toString $ Require.transform False (Require.FileName "Foo.hs") "" input
    actual `shouldStartWith` expectedStart
    actual `shouldContain` expectedModule
    actual `shouldContain` expectedTypeImport
    actual `shouldContain` expectedQualifiedImport
    actual `shouldEndWith` expectedContent
  it "aliases the modules properly" $ do
    let input = "require Data.Text as Foo"
    let expectedTypeImport = "import Data.Text (Text)"
    let expectedQualifiedImport = "import qualified Data.Text as Foo"
    let actual = toString $ Require.transform False (Require.FileName "Foo.hs") "" input
    actual `shouldContain` expectedTypeImport
    actual `shouldContain` expectedQualifiedImport
  it "imports the types properly" $ do
    let input = "require Data.Text (Foo)"
    let expectedTypeImport = "import Data.Text (Foo)"
    let expectedQualifiedImport = "import qualified Data.Text as Text"
    let actual = toString $ Require.transform False (Require.FileName "Foo.hs") "" input
    actual `shouldContain` expectedTypeImport
    actual `shouldContain` expectedQualifiedImport
  it "imports the types and aliases the modules properly" $ do
    let input = "require Data.Text as Quux (Foo)"
    let expectedTypeImport = "import Data.Text (Foo)"
    let expectedQualifiedImport = "import qualified Data.Text as Quux"
    let actual = toString $ Require.transform False (Require.FileName "Foo.hs") "" input
    actual `shouldContain` expectedTypeImport
    actual `shouldContain` expectedQualifiedImport
  it "skips comments" $ do
    let input = "require Data.Text -- test of comments"
    let expected = "import Data.Text (Text)"
    let actual = Require.transform False (Require.FileName "Foo.hs") "" input
    expected `Text.isInfixOf` actual
  it "allows empty parentheses" $ do
    let input = "require Data.Text ()"
    let expected1 = "import Data.Text ()"
    let expected2 = "import qualified Data.Text as Text"
    let actual = lines $ Require.transform False (Require.FileName "Foo.hs") "" input
    actual `shouldSatisfy` elem expected1
    actual `shouldSatisfy` elem expected2