raise exception when attempting to publish when server is not set

Author: ryanb

lib/private_pub.rb

@@ -30,6 +30,7 @@ def publish_to(channel, data)
 
     # Sends the given message hash to the Faye server using Net::HTTP.
     def publish_message(message)
+      raise Error, "No server specified, ensure private_pub.yml was loaded properly." unless config[:server]
       Net::HTTP.post_form(URI.parse(config[:server]), :message => message.to_json)
     end
 
@@ -45,7 +46,8 @@ def message(channel, data)
     end
 
     # Returns a subscription hash to pass to the PrivatePub.sign call in JavaScript.
-    def subscription
+    # Any options passed are merged to the hash.
+    def subscription(options = {})
       sub = {:server => config[:server], :timestamp => (Time.now.to_f * 1000).round}.merge(options)
       sub[:signature] = Digest::SHA1.hexdigest([config[:secret_token], sub[:channel], sub[:timestamp]].join)
       sub

spec/private_pub_spec.rb

@@ -77,6 +77,12 @@
     PrivatePub.publish_message(message).should == :result
   end
 
+  it "raises an exception if no server is specified when calling publish_message" do
+    lambda {
+      PrivatePub.publish_message("foo")
+    }.should raise_error(PrivatePub::Error)
+  end
+
   it "publish_to passes message to publish_message call" do
     PrivatePub.should_receive(:message).with("chan", "foo").and_return("message")
     PrivatePub.should_receive(:publish_message).with("message").and_return(:result)