Use environment instead of test attribute on commerce initialization. #3

lib/tbk/commerce.rb

@@ -4,18 +4,22 @@ module TBK
   class Commerce
 
     # The registered commerce id
-    attr_accessor :id
+    attr_reader :id
 
     # The commerce secret RSA key
-    attr_accessor :key
+    attr_reader :key
+
+    # The commerce environment
+    attr_reader :environment
 
     # Initialyzes a new commerce
     # @param [Hash] attributes The commerce attributes
     # @option attributes [Integer] :id The commerce ID
     # @option attributes [String|OpenSSL::PKey::RSA] :key The commerce RSA private key
     # @option attributes [Boolean] :test flag to set commerce in test mode
     def initialize(attributes)
-      @test = attributes[:test]
+      @environment = (attributes[:environment] || :production).to_sym
+      raise TBK::CommerceError, "Invalid commerce environment" unless [:production,:test].include? @environment
 
       self.id = attributes[:id]
       raise TBK::CommerceError, "Missing commerce id" if self.id.nil?
@@ -35,12 +39,12 @@ def initialize(attributes)
 
     # @return [Boolean] wether or not the commerce is in test mode
     def test?
-      @test || false
+      self.environment == :test
     end
 
     # @return [Boolean] wether or not the commerce is in production mode
     def production?
-      !self.test?
+      self.environment == :production
     end
 
     # @return [Integer] RSA key bytes