wasp/waspc/test/Generator/FileDraft/TemplateFileDraftTest.hs

module Generator.FileDraft.TemplateFileDraftTest where

import Data.Aeson (object, (.=))
import Data.Text (Text)
import Fixtures (systemSPRoot)
import qualified Generator.MockWriteableMonad as Mock
import qualified StrongPath as SP
import Test.Tasty.Hspec
import Wasp.Generator.FileDraft

spec_TemplateFileDraft :: Spec
spec_TemplateFileDraft = do
  describe "write" $ do
    it "Creates new file from existing template file" $ do
      let mock = write dstDir fileDraft
      let mockLogs = Mock.getMockLogs mock mockConfig
      Mock.compileAndRenderTemplate_calls mockLogs
        `shouldBe` [(templatePath, templateData)]
      Mock.createDirectoryIfMissing_calls mockLogs
        `shouldBe` [(True, SP.toFilePath $ SP.parent expectedDstPath)]
      Mock.writeFileFromText_calls mockLogs
        `shouldBe` [(SP.toFilePath expectedDstPath, mockTemplateContent)]
  where
    (dstDir, dstPath, templatePath) =
      ( systemSPRoot SP.</> [SP.reldir|a/b|],
        [SP.relfile|c/d/dst.txt|],
        [SP.relfile|e/tmpl.txt|]
      )
    templateData = object ["foo" .= ("bar" :: String)]
    fileDraft = createTemplateFileDraft dstPath templatePath (Just templateData)
    expectedDstPath = dstDir SP.</> dstPath
    mockTemplatesDirAbsPath = systemSPRoot SP.</> [SP.reldir|mock/templates/dir|]
    mockTemplateContent = "Mock template content" :: Text
    mockConfig =
      Mock.defaultMockConfig
        { Mock.getTemplatesDirAbsPath_impl = mockTemplatesDirAbsPath,
          Mock.compileAndRenderTemplate_impl = \_ _ -> mockTemplateContent
        }