lib/cucumber/pro.rb in cucumber-pro-0.0.2 vs lib/cucumber/pro.rb in cucumber-pro-0.0.4

@@ -1,15 +1,15 @@
 require 'logger'
 require 'cucumber/pro/formatter'
 require 'cucumber/pro/web_socket_session'
+require 'cucumber/pro/version'
 
 module Cucumber
   module Pro
 
     class << self
       def new(*)
-        url = config.url + "?token=#{config.token}"
         session = WebSocketSession.new(url, config.logger)
         Formatter.new(session)
       end
 
       def configure
@@ -19,10 +19,15 @@
       private
 
       def config
         @config ||= Config.new
       end
+
+      def url
+        token = config.token || raise(Error::MissingToken.new)
+        config.url + "?token=#{token}"
+      end
     end
 
     class Config
       attr_accessor :url, :logger, :token
     end
@@ -30,9 +35,23 @@
     # Default config
     configure do |config|
       config.url    = 'wss://results.cucumber.pro/ws'
       config.logger = Logger.new(ENV['cucumber_pro_log_path'] || STDOUT)
       config.token  = ENV['CUCUMBER_PRO_TOKEN']
+    end
+
+    module Error
+      AccessDenied = Class.new(StandardError) {
+        def initialize
+          super "Access denied."
+        end
+      }
+
+      MissingToken = Class.new(StandardError) {
+        def initialize
+          super "Missing access token. Please visit https://app.cucumber.pro/api-token for instructions."
+        end
+      }
     end
 
   end
 end