Fix a few require paths.

[dead/whatever-dl.git] / lib / vendor / ruby-progressbar / test.rb

diff --git a/lib/vendor/ruby-progressbar/test.rb b/lib/vendor/ruby-progressbar/test.rb
new file mode 100644 (file)
index 0000000..06cd571
--- /dev/null
+++ b/lib/vendor/ruby-progressbar/test.rb
@@ -0,0 +1,110 @@
+require 'test/unit'
+
+# Modified a little to run from the whatever-dl test suite.
+require (File.dirname(__FILE__) + '/progressbar')
+
+class ProgressBarTest < Test::Unit::TestCase
+  SleepUnit = 0.01
+
+  def do_make_progress_bar (title, total)
+    ProgressBar.new(title, total)
+  end
+
+  def test_bytes
+    total = 1024 * 1024
+    pbar = do_make_progress_bar("test(bytes)", total)
+    pbar.file_transfer_mode
+    0.step(total, 2**14) {|x|
+      pbar.set(x)
+      sleep(SleepUnit)
+    }
+    pbar.finish
+  end
+
+  def test_clear
+    total = 100
+    pbar = do_make_progress_bar("test(clear)", total)
+    total.times {
+      sleep(SleepUnit)
+      pbar.inc
+    }
+    pbar.clear
+    puts
+  end
+
+  def test_halt
+    total = 100
+    pbar = do_make_progress_bar("test(halt)", total)
+    (total / 2).times {
+      sleep(SleepUnit)
+      pbar.inc
+    }
+    pbar.halt
+  end
+
+  def test_inc
+    total = 100
+    pbar = do_make_progress_bar("test(inc)", total)
+    total.times {
+      sleep(SleepUnit)
+      pbar.inc
+    }
+    pbar.finish
+  end
+
+  def test_inc_x
+    # Modified a little to run from the whatever-dl test suite.
+    pbar_file_path = (File.dirname(__FILE__) + '/progressbar.rb')
+    
+    total = File.size(pbar_file_path)
+    pbar = do_make_progress_bar("test(inc(x))", total)
+    File.new(pbar_file_path).each {|line|
+      sleep(SleepUnit)
+      pbar.inc(line.length)
+    }
+    pbar.finish
+  end
+
+  def test_invalid_set
+    total = 100
+    pbar = do_make_progress_bar("test(invalid set)", total)
+    begin
+      pbar.set(200)
+    rescue RuntimeError => e
+      puts e.message
+    end
+  end
+
+  def test_set
+    total = 1000
+    pbar = do_make_progress_bar("test(set)", total)
+    (1..total).find_all {|x| x % 10 == 0}.each {|x|
+      sleep(SleepUnit)
+      pbar.set(x)
+    }
+    pbar.finish
+  end
+
+  def test_slow
+    total = 100000
+    pbar = do_make_progress_bar("test(slow)", total)
+    0.step(500, 1) {|x|
+      pbar.set(x)
+      sleep(SleepUnit)
+    }
+    pbar.halt
+  end
+
+  def test_total_zero
+    total = 0
+    pbar = do_make_progress_bar("test(total=0)", total)
+    pbar.finish
+  end
+end
+
+class ReversedProgressBarTest < ProgressBarTest
+  def do_make_progress_bar (title, total)
+    ReversedProgressBar.new(title, total)
+  end
+end
+