ArtifactGeneratorImplSpec.hs

module B9.ArtifactGeneratorImplSpec
  ( spec
  )
where

import           B9.Artifact.Readable
import           B9.Artifact.Readable.Interpreter
import           B9.DiskImages
import           B9.ExecEnv
import           B9.ShellScript
import           B9.Vm
import           Data.Text                      ( )
import           Test.Hspec

spec :: Spec
spec = describe "assemble" $ do
  it "replaces '$...' variables in SourceImage Image file paths"
    $ let
        src = Let
          [("variable", "value")]
          [vmImagesArtifact "" [transientCOW "${variable}" ""] NoVmScript]
        expected = transientCOW "value" ""
        (Right [IG _ _ (VmImages [actual] _)]) =
          runArtifactGenerator mempty "" "" src
      in
        actual `shouldBe` expected
  it "replaces '$...' variables in SourceImage 'From' names"
    $ let
        src = Let
          [("variable", "value")]
          [vmImagesArtifact "" [transientShared "${variable}" ""] NoVmScript]
        expected = transientShared "value" ""
        (Right [IG _ _ (VmImages [actual] _)]) =
          runArtifactGenerator mempty "" "" src
      in
        actual `shouldBe` expected
  it "replaces '$...' variables in the name of a shared image"
    $ let
        src = Let
          [("variable", "value")]
          [vmImagesArtifact "" [shareCOW "${variable}" ""] NoVmScript]
        expected = shareCOW "value" ""
        (Right [IG _ _ (VmImages [actual] _)]) =
          runArtifactGenerator mempty "" "" src
      in
        actual `shouldBe` expected
  it "replaces '$...' variables in the name and path of a live installer image"
    $ let
        src = Let
          [("variable", "value")]
          [ vmImagesArtifact ""
                             [liveInstallerCOWImage "${variable}" ""]
                             NoVmScript
          ]
        expected = liveInstallerCOWImage "value" ""
        (Right [IG _ _ (VmImages [actual] _)]) =
          runArtifactGenerator mempty "" "" src
      in
        actual `shouldBe` expected
  it
      "replaces '$...' variables in the file name of an image exported as LocalFile"
    $ let
        src = Let
          [("variable", "value")]
          [vmImagesArtifact "" [localCOWImage "${variable}" ""] NoVmScript]
        expected = localCOWImage "value" ""
        (Right [IG _ _ (VmImages [actual] _)]) =
          runArtifactGenerator mempty "" "" src
      in
        actual `shouldBe` expected
  it "replaces '$...' variables in mount point of an image"
    $ let
        src = Let
          [("variable", "value")]
          [vmImagesArtifact "" [localCOWImage "" "${variable}"] NoVmScript]
        expected = localCOWImage "" "value"
        (Right [IG _ _ (VmImages [actual] _)]) =
          runArtifactGenerator mempty "" "" src
      in
        actual `shouldBe` expected
  it "replaces '$...' variables in shared directory source and mount point (RO)"
    $ let
        src = Let
          [("variable", "value")]
          [vmImagesArtifact "" [] (emptyScriptWithSharedDirRO "${variable}")]
        expected = emptyScriptWithSharedDirRO "value"
        (Right [IG _ _ (VmImages [] actual)]) =
          runArtifactGenerator mempty "" "" src
      in
        actual `shouldBe` expected
  it "replaces '$...' variables in shared directory source and mount point (RW)"
    $ let
        src = Let
          [("variable", "value")]
          [vmImagesArtifact "" [] (emptyScriptWithSharedDirRW "${variable}")]
        expected = emptyScriptWithSharedDirRW "value"
        (Right [IG _ _ (VmImages [] actual)]) =
          runArtifactGenerator mempty "" "" src
      in
        actual `shouldBe` expected
  it "replaces '$...' variables in VmImages build script instructions"
    $ let
        src = Let [("variable", "value")]
                  [vmImagesArtifact "" [] (buildScript "${variable}")]
        expected = buildScript "value"
        (Right [IG _ _ (VmImages [] actual)]) =
          runArtifactGenerator mempty "" "" src
      in
        actual `shouldBe` expected

transientCOW :: FilePath -> FilePath -> ImageTarget
transientCOW fileName mountPoint = ImageTarget
  Transient
  (CopyOnWrite (Image fileName QCow2 Ext4))
  (MountPoint mountPoint)

transientShared :: FilePath -> FilePath -> ImageTarget
transientShared name mountPoint =
  ImageTarget Transient (From name KeepSize) (MountPoint mountPoint)

shareCOW :: FilePath -> FilePath -> ImageTarget
shareCOW destName mountPoint = ImageTarget
  (Share destName QCow2 KeepSize)
  (CopyOnWrite (Image "cowSource" QCow2 Ext4))
  (MountPoint mountPoint)

liveInstallerCOWImage :: FilePath -> FilePath -> ImageTarget
liveInstallerCOWImage destName mountPoint = ImageTarget
  (LiveInstallerImage destName destName KeepSize)
  (CopyOnWrite (Image "cowSource" QCow2 Ext4))
  (MountPoint mountPoint)

localCOWImage :: FilePath -> FilePath -> ImageTarget
localCOWImage destName mountPoint = ImageTarget
  (LocalFile (Image destName QCow2 Ext4) KeepSize)
  (CopyOnWrite (Image "cowSource" QCow2 Ext4))
  (MountPoint mountPoint)

vmImagesArtifact :: String -> [ImageTarget] -> VmScript -> ArtifactGenerator
vmImagesArtifact iid imgs script = Artifact (IID iid) (VmImages imgs script)

emptyScriptWithSharedDirRO :: String -> VmScript
emptyScriptWithSharedDirRO arg =
  VmScript X86_64 [SharedDirectoryRO arg (MountPoint arg)] (Run "" [])

emptyScriptWithSharedDirRW :: String -> VmScript
emptyScriptWithSharedDirRW arg =
  VmScript X86_64 [SharedDirectory arg (MountPoint arg)] (Run "" [])

buildScript :: String -> VmScript
buildScript arg = VmScript
  X86_64
  [SharedDirectory arg (MountPoint arg), SharedDirectoryRO arg NotMounted]
  (As arg [In arg [Run arg [arg]]])