task: more test coverage

3 files changed, 44 insertions, 10 deletions

diff --git a/test/net_test.rb b/test/net_test.rb
index 3eec454..e00d16a 100644
--- a/test/net_test.rb
+++ b/test/net_test.rb
@@ -2,15 +2,6 @@ require 'test_helper'
 
 class NetTest < HCl::TestCase
 
-  def setup
-    FakeWeb.allow_net_connect = false
-    HCl::Net.configure \
-      'login' => 'bob',
-      'password' => 'secret',
-      'subdomain' => 'bobclock',
-      'ssl' => true
-  end
-
   def test_configure
     assert_equal 'bob', HCl::Net.login
     assert_equal 'secret', HCl::Net.password
diff --git a/test/task_test.rb b/test/task_test.rb
index bbdce89..47393e8 100644
--- a/test/task_test.rb
+++ b/test/task_test.rb
@@ -18,4 +18,34 @@ class TaskTest < HCl::TestCase
     assert_equal 1, HCl::Task.all.size
     assert_equal 'Security support', HCl::Task.all.first.name
   end
+
+  def test_add
+    task = HCl::Task.new(id:1, project:HCl::Project.new({id:2}))
+    FakeWeb.register_uri(:post, %r{/daily/add$},
+                         body: Yajl::Encoder.encode(note:'good stuff',
+                                hours:0.2, project_id:2, task_id:1, spent_at: Date.today))
+    entry = task.add(note:'good stuff', starting_time:0.2)
+    assert_equal 'good stuff', entry.note
+  end
+
+  def test_start_running
+    task = HCl::Task.new(id:1, project:HCl::Project.new({id:2}))
+    FakeWeb.register_uri(:post, %r{/daily/add$},
+                         body: Yajl::Encoder.encode(note:'good stuff',
+                                timer_started_at:DateTime.now,
+                                hours:0.2, project_id:2, task_id:1, spent_at: Date.today))
+    entry = task.start(note:'good stuff', starting_time:0.2)
+    assert_equal 'good stuff', entry.note
+  end
+
+  def test_start_then_toggle
+    task = HCl::Task.new(id:1, project:HCl::Project.new({id:2}))
+    FakeWeb.register_uri(:post, %r{/daily/add$},
+                         body: Yajl::Encoder.encode(id:123, note:'woot'))
+    FakeWeb.register_uri(:get, %r{/daily/timer/123$},
+                         body: Yajl::Encoder.encode(note:'good stuff',
+                                hours:0.2, project_id:2, task_id:1, spent_at: Date.today))
+    entry = task.start(note:'good stuff', starting_time:0.2)
+    assert_equal 'good stuff', entry.note
+  end
 end
diff --git a/test/test_helper.rb b/test/test_helper.rb
index 9067d91..98c0b35 100644
--- a/test/test_helper.rb
+++ b/test/test_helper.rb
@@ -31,5 +31,18 @@ require 'fakeweb'
 # require test extensions/helpers
 Dir[File.dirname(__FILE__) + '/ext/*.rb'].each { |ext| require ext }
 
-class HCl::TestCase < MiniTest::Unit::TestCase; end
+class HCl::TestCase < MiniTest::Unit::TestCase
+  def setup
+    FakeWeb.allow_net_connect = false
+    HCl::Net.configure \
+      'login' => 'bob',
+      'password' => 'secret',
+      'subdomain' => 'bobclock',
+      'ssl' => true
+  end
+  def teardown
+    FakeWeb.clean_registry
+  end
+end
+