template.rb in mirage-3.0.0.alpha.3

- old
+ new

@@ -1,9 +1,10 @@
 require 'ostruct'
 require 'json'
 require 'httparty'
 require 'hashie/mash'
+
 module Mirage
 
   class Template
     include HTTParty
 
@@ -21,10 +22,11 @@
         template.id = response_hashie.id
         template.default = response_config['default']
         template.delay = response_config.delay
         template.content_type = response_config.content_type
         template.status = response_config.status
+        template.headers = response_config.headers
 
         template.required_parameters = request_config.parameters
         template.required_body_content = request_config.body_content
         template.http_method = request_config.http_method
         template.url = url
@@ -35,11 +37,11 @@
       end
     end
 
     format :json
 
-    attr_accessor :content_type, :http_method, :default, :status, :delay, :required_parameters, :required_body_content, :required_headers, :endpoint, :id, :url, :requests_url
+    attr_accessor :content_type, :http_method, :default, :status, :delay, :required_parameters, :required_body_content, :required_headers, :endpoint, :id, :url, :requests_url, :headers
     attr_reader :value
 
 
     def initialize endpoint, response, default_config=TemplateConfiguration.new
 
@@ -50,10 +52,11 @@
       @status = default_config.status
       @delay = default_config.delay
       @required_parameters = {}
       @required_headers = {}
       @required_body_content = []
+      @headers = {}
       @default = default_config.default
     end
 
     def create
       @id = self.class.put("#{@endpoint}", :body => self.to_json, :headers => {'content-type' => 'application/json'})['id']
@@ -70,18 +73,18 @@
       {
           :response => {
               :body => Base64.encode64(@value),
               :status => status,
               :default => default,
-              :content_type => content_type
+              :content_type => content_type,
+              :headers => headers
 
           },
           :request => {
               :parameters => encode_regexs(required_parameters),
               :headers => encode_regexs(required_headers),
               :body_content => encode_regexs(required_body_content),
-              :http_method => http_method
-
+              :http_method => http_method,
           },
           :delay => delay
       }.to_json
     end
 
\ No newline at end of file