wimlib.net Git - wimlib/blob - tests/test-imagex-capture_and_apply

   1 #!/usr/bin/env bash
   2
   3 # Test capturing and applying a WIM image in the normal (non-NTFS) capture mode
   4 #
   5 # Add in some tests with WIM splitting, joining, and exporting as well.
   6 #
   7 # Test all three compression modes (None, XPRESS, and LZX).
   8 #
   9 # Also, test if the capture configuration file works correctly.
  10
  11 set -e
  12 cd tests
  13 srcdir="${srcdir:-.}/.."
  14 srcdir="$(cd $srcdir; pwd)"
  15 . "$srcdir/tests/tests-common.sh"
  16
  17 TEST_SUBDIR=tmpdir_test-imagex-capture_and_apply
  18
  19 do_tree_cmp() {
  20         if ! ../tree-cmp in.dir out.dir; then
  21                 if [ -x /usr/bin/tree ]; then
  22                         echo "Dumping tree of applied image"
  23                         echo "(Note: compression type was $ctype)"
  24                         tree out.dir --inodes -F -s --noreport
  25                         error 'Information was lost or corrupted while capturing
  26                                 and then applying a directory tree'
  27                 fi
  28         fi
  29 }
  30
  31 image_name=0
  32 do_test() {
  33         for ctype in None LZX XPRESS; do
  34
  35                 # Can we capture the WIM, apply it, and get the same result?
  36                 cd in.dir
  37                 eval "$1"
  38                 cd ..
  39                 if [ -x /usr/bin/tree -a "$ctype" = "None" ]; then
  40                         tree in.dir --inodes -F -s --noreport
  41                 fi
  42                 if ! imagex capture in.dir test.wim --compress=$ctype; then
  43                         error "Failed to capture directory tree into a WIM"
  44                 fi
  45                 if ! imagex apply test.wim 1 out.dir; then
  46                         error "Failed to apply WIM to directory"
  47                 fi
  48                 if [ `wim_ctype test.wim` != $ctype ]; then
  49                         error "'imagex info' didn't report the compression type on the captured WIM correctly"
  50                 fi
  51                 do_tree_cmp
  52                 rm -rf out.dir/*
  53
  54                 # Can we split the WIM, apply the split WIM, join the split WIM,
  55                 # and apply the joined WIM, and get the same results every time?
  56                 if ! imagex split test.wim test.swm 0.01; then
  57                         error "Failed to split WIM"
  58                 fi
  59                 if ! imagex apply test.swm 1 out.dir --ref "test*.swm" ; then
  60                         error "Failed to apply split WIM"
  61                 fi
  62                 do_tree_cmp
  63                 rm -rf out.dir/* test.wim
  64                 if ! imagex join test.wim test*.swm; then
  65                         error "Failed to join split WIM"
  66                 fi
  67                 if ! imagex apply test.wim out.dir; then
  68                         error "Failed to apply joined WIM"
  69                 fi
  70                 do_tree_cmp
  71                 rm -rf out.dir/*
  72
  73                 # Can we export the image to another WIM, apply it, and get the
  74                 # same results?
  75                 (( image_name++ )) || true
  76                 if ! imagex export test.wim 1 test2.wim "$image_name"; then
  77                         error "Failed to export WIM image"
  78                 fi
  79
  80                 if ! imagex apply test2.wim "$image_name" out.dir; then
  81                         error "Failed to apply exported WIM image"
  82                 fi
  83                 do_tree_cmp
  84
  85                 rm -rf out.dir/* in.dir/* test.wim test*.swm
  86
  87         done
  88 }
  89
  90 __msg() {
  91         echo "--------------------------------------------------------------------"
  92         echo $1
  93         echo "--------------------------------------------------------------------"
  94 }
  95
  96 msg() {
  97         __msg "Testing image capture and application of directory containing $1"
  98 }
  99
 100 default_cleanup
 101 mkdir $TEST_SUBDIR
 102 cd $TEST_SUBDIR
 103 mkdir in.dir out.dir
 104
 105 . $srcdir/tests/common_tests.sh
 106
 107 # Make sure exclusion list works
 108 __msg "Testing default capture configuration file"
 109 touch in.dir/hiberfil.sys
 110 mkdir -p "in.dir/System Volume Information/subdir"
 111 imagex capture in.dir test.wim
 112 imagex apply test.wim out.dir
 113 if [ -e out.dir/hiberfil.sys -o -e "out.dir/System Volume Information" ]; then
 114         error "Files were not excluded from capture as expected"
 115 fi
 116
 117 echo "**********************************************************"
 118 echo "          imagex capture/apply tests passed               "
 119 echo "**********************************************************"
 120
 121 cd ..
 122 default_cleanup