test/client_test.rb in suo-0.3.1 vs test/client_test.rb in suo-0.3.2

@@ -29,9 +29,15 @@
     locked = @client.locked?
 
     assert_equal false, locked
   end
 
+  def test_lock_with_custom_token
+    token = 'foo-bar'
+    lock  = @client.lock token
+    assert_equal lock, token
+  end
+
   def test_empty_lock_on_invalid_data
     @client.send(:initial_set, "bad value")
     assert_equal false, @client.locked?
   end