collimator 0.0.1

data/test/progressbar_test.rb

@@ -0,0 +1,87 @@
+require 'test_helper'
+
+class TestProgressBar < Test::Unit::TestCase
+  include Collimator
+
+  def teardown
+    $stdout = STDOUT
+  end
+
+  def test_progress
+    out = capture_output do
+      ProgressBar.start({:min => 0, :max => 100, :method => :percent, :step_size => 1})
+    end
+
+    s = out.string.clone
+    assert_equal 102, s.count("\b")
+
+    s.gsub!("\b", "")
+    assert_equal "|                                                0.0%                                                |", s
+    assert_equal s.length, 102
+
+    out = capture_output do
+      ProgressBar.increment
+    end
+
+    s = out.string.clone
+    s.gsub!("\b", "")
+    assert_equal "|-                                               1.0%                                                |", s
+
+    5.times do
+      out = capture_output do
+        ProgressBar.increment
+      end
+    end
+
+    out = capture_output do
+      ProgressBar.increment
+    end
+
+    s = out.string.clone
+    s.gsub!("\b", "")
+    assert_equal "|-------                                         7.0%                                                |", s
+
+    75.times do
+      out = capture_output do
+        ProgressBar.increment
+      end
+    end
+
+    out = capture_output do
+      ProgressBar.increment
+    end
+
+    s = out.string.clone
+    s.gsub!("\b", "")
+    assert_equal "|-----------------------------------------------83.0% ------------------------------                 |", s
+
+  end
+
+  def test_completion
+    out = capture_output do
+      ProgressBar.start({:min => 0, :max => 100, :method => :percent, :step_size => 10})
+    end
+
+    5.times do
+      out = capture_output do
+        ProgressBar.increment
+      end
+    end
+
+    out = capture_output do
+      ProgressBar.increment
+    end
+    s = out.string.clone
+    s.gsub!("\b", "")
+    assert_equal "|-----------------------------------------------60.0% -------                                        |", s
+
+    out = capture_output do
+      ProgressBar.complete
+    end
+
+    s = out.string.clone
+    s.gsub!("\b", "")
+    assert_equal "                                                                                                      \n", s
+
+  end
+end

data/test/spinner_test.rb

@@ -0,0 +1,34 @@
+require 'test_helper'
+
+class TestProgressBar < Test::Unit::TestCase
+  include Collimator
+
+  def teardown
+    $stdout = STDOUT
+  end
+
+  def test_spinner
+    spin_time = 1
+    out = capture_output do
+      Spinner.spin
+      sleep spin_time
+      Spinner.stop
+    end
+
+    length_should_be = (spin_time / 0.1) * 2
+    s = out.string.clone
+    assert_equal length_should_be, s.length
+
+    spin_time = 0.2
+    out = capture_output do
+      Spinner.spin
+      sleep spin_time
+      Spinner.stop
+    end
+
+    length_should_be = (spin_time / 0.1) * 2
+    s = out.string.clone
+    assert_equal length_should_be, s.length
+  end
+
+end