require File.join(File.dirname(__FILE__), '..' , 'rakelib' , 'extensions')
require 'tmpdir'
require 'test/unit'

include RakeFileUtils

class RakeFileUtilsTests < Test::Unit::TestCase
  def test_which()    
    assert File.executable?( which('ls') )
    assert which('boguscommand').nil?
  end

  # A helper for `test_zip_01()`
  def test_zip_01_fmt(fmt)
    Dir.mktmpdir do | tmp |
      chdir File.join(File.dirname(__FILE__), '..') do
        archive = File.join(tmp, "rakelib#{fmt}")
        zip "rakelib", archive: archive
        assert File.exist? archive
        unzip archive
        assert File.exist? File.join(tmp, 'rakelib')
        assert File.exist? File.join(tmp, 'rakelib', 'extensions.rb')
        assert File.exist? File.join(tmp, 'rakelib', 'support.rb')
      end
    end
  end  

  def test_zip_01()    
    test_zip_01_fmt('.zip') if which 'zip'
    test_zip_01_fmt('.tar.bz2') if which 'tar' and which 'bzip2'
    test_zip_01_fmt('.tar.gz') if which 'tar' and which 'gzip'
  end

  def test_zip_02()     
    chdir File.join(File.dirname(__FILE__), '..') do        
      fmt = win32? ? '.zip' : '.tar.bz2'
      begin
        zip "rakelib"        
        assert(File.exist? "rakelib#{fmt}")
      ensure
        rm_f "rakelib#{fmt}" 
      end
    end
  end

  # A helper for `test_zip_03()`
  def test_zip_03_fmt(fmt)
    Dir.mktmpdir do | tmp |
      chdir File.join(File.dirname(__FILE__), '..') do
        archive = File.join(tmp, "rakelib#{fmt}")
        zip "rakelib", archive: archive, include: [ 'extensions.rb' ]
        assert File.exist? archive
        unzip archive
        assert File.exist? File.join(tmp, 'rakelib')
        assert File.exist? File.join(tmp, 'rakelib', 'extensions.rb')
        assert (not File.exist? File.join(tmp, 'rakelib', 'support.rb'))
      end
    end
  end  

  def test_zip_03()    
    test_zip_03_fmt('.zip') if which 'zip'
    test_zip_03_fmt('.tar.bz2') if which 'tar' and which 'bzip2'
    test_zip_03_fmt('.tar.gz') if which 'tar' and which 'gzip'
  end

  def test_rm_rf_ex() 
    def with_test_dir(&block) 
      Dir.mktmpdir do | tmp |
        chdir tmp do
          mkdir_p File.join("a", "b", "c")
          touch File.join("a", "aaa1.txt")
          touch File.join("a", "aaa2.txt")
          touch File.join("a", "aaa3.txt")
          touch File.join("a", "b", "bbb1.txt")
          touch File.join("a", "b", "bbb2.txt")
          touch File.join("a", "b", "bbb3.txt")
          touch File.join("a", "b", "c", "ccc1.txt")
          touch File.join("a", "b", "c", "ccc2.txt")
          touch File.join("a", "b", "c", "ccc3.txt")          
          yield block
        end
      end
    end    

    with_test_dir do
      rm_rf_ex "a"
      sh 'ls -lr'
      assert (not File.exist? "a")
    end

    with_test_dir do
      rm_rf_ex "a", exceptions: ['aaa1.txt']
      sh 'ls -lr'
      assert (    File.exist? "a")
      assert (    File.exist? (File.join("a", "aaa1.txt")))
      assert (not File.exist? (File.join("a", "aaa2.txt")))
      assert (not File.exist? (File.join("a", "aaa3.txt")))
      assert (not File.exist? (File.join("a", "b")))
    end

    with_test_dir do
      rm_rf_ex "a", exceptions: ['bogus']
      sh 'ls -lr'
      assert (not File.exist? "a")      
    end

    with_test_dir do
      rm_rf_ex "a", exceptions: [File.join('b' 'bogus')]
      sh 'ls -lr'
      assert (not File.exist? "a")      
    end

    with_test_dir do
      rm_rf_ex "a", exceptions: ['b']
      sh 'ls -lr'
      assert (    File.exist? "a")
      assert (not File.exist? (File.join("a", "aaa1.txt")))
      assert (not File.exist? (File.join("a", "aaa2.txt")))
      assert (not File.exist? (File.join("a", "aaa3.txt")))
      assert (    File.exist? (File.join("a", "b")))
      assert (    File.exist? (File.join("a", "b", "bbb1.txt")))
      assert (    File.exist? (File.join("a", "b", "bbb2.txt")))
      assert (    File.exist? (File.join("a", "b", "bbb3.txt")))
      assert (    File.exist? (File.join("a", "b", "c")))
      assert (    File.exist? (File.join("a", "b", "c", "ccc1.txt")))
      assert (    File.exist? (File.join("a", "b", "c", "ccc2.txt")))
      assert (    File.exist? (File.join("a", "b", "c", "ccc3.txt")))
    end

    with_test_dir do
      rm_rf_ex "a", exceptions: [File.join('b', 'bbb1.txt'), File.join('b', 'c', 'ccc3.txt') ]
      sh 'ls -lr'
      assert (    File.exist? "a")
      assert (not File.exist? (File.join("a", "aaa1.txt")))
      assert (not File.exist? (File.join("a", "aaa2.txt")))
      assert (not File.exist? (File.join("a", "aaa3.txt")))
      assert (    File.exist? (File.join("a", "b")))
      assert (    File.exist? (File.join("a", "b", "bbb1.txt")))
      assert (not File.exist? (File.join("a", "b", "bbb2.txt")))
      assert (not File.exist? (File.join("a", "b", "bbb3.txt")))
      assert (    File.exist? (File.join("a", "b", "c")))
      assert (not File.exist? (File.join("a", "b", "c", "ccc1.txt")))
      assert (not File.exist? (File.join("a", "b", "c", "ccc2.txt")))
      assert (    File.exist? (File.join("a", "b", "c", "ccc3.txt")))
    end
  end
end

if __FILE__ == $0
  require 'test/unit/ui/console/testrunner'
  Test::Unit::UI::Console::TestRunner.run(RakeFileUtilsTests)
end