surveyor 0.6.1

data/script/surveyor/survey_section.rb

@@ -0,0 +1,153 @@
+require File.dirname(__FILE__) + '/question_group'
+require File.dirname(__FILE__) + '/question'
+require File.dirname(__FILE__) + '/answer'
+require File.dirname(__FILE__) + '/dependency'
+require File.dirname(__FILE__) + '/dependency_condition'
+require File.dirname(__FILE__) + '/columnizer'
+require 'activesupport'
+#require 'activesupport/lib/active_support/core_ext/string/inflections'
+
+class SurveySection
+  include Columnizer
+
+  # Context, Content, Display, Reference, Children, Placeholders
+  attr_accessor :id, :survey_id, :parser
+  attr_accessor :title, :description
+  attr_accessor :display_order
+  attr_accessor :reference_identifier, :data_export_identifier
+  attr_accessor :question_groups, :questions, :grid_answers
+  attr_accessor :current_question_group, :current_question, :current_dependency
+
+
+  # id, survey, and title required
+  def initialize(id, survey, title, options = {})
+    self.id = id
+    self.survey_id = survey.id
+    self.parser = survey.parser
+    self.title = title.strip
+    self.questions = []
+    self.question_groups = []
+    self.grid_answers = []
+    self.current_question_group = nil
+    self.current_question = nil
+    self.current_dependency = nil
+    self.default_options(title).merge(options).each{|key,value| self.instance_variable_set("@#{key}", value)}
+  end
+
+  def default_options(title)
+    { :data_export_identifier => Columnizer.to_normalized_column(title)
+    }
+  end
+
+  # This method_missing magic does all the heavy lifting for the DSL
+  def method_missing(missing_method, *args, &block)    
+    method_name, reference_identifier = missing_method.to_s.split("_")
+
+    case method_name
+      
+    when "group", "g", "grid", "repeater"
+      puts "    Group: #{reference_identifier}"
+      raise "Error: A question group cannot be empty" unless block_given?
+      
+      options = {:reference_identifier => reference_identifier, :display_type => (method_name =~ /grid|repeater/)? method_name : nil }
+      self.question_groups << (self.current_question_group = QuestionGroup.new(self, args, options))
+      self.instance_eval(&block)
+      clear_current_question_group
+      
+    when "question", "q", "label", "image"
+      puts "      Question: #{reference_identifier}"
+      raise "Error: I'm dropping the block like it's hot" if block_given?
+      
+      clear_current_question
+      options = {:reference_identifier => reference_identifier, :question_group_id => (current_question_group ? current_question_group.id : nil)}
+      options.merge!({:display_type => "label"}) if method_name == "label"
+      options.merge!({:display_type => "image"}) if method_name == "image"
+      self.questions << (self.current_question = Question.new(self, args, options))
+      add_grid_answers if self.current_question_group and self.current_question_group.display_type == "grid"
+      
+    when "dependency", "d"
+      puts "        Dependency: #{reference_identifier}"
+      raise "Error: I'm dropping the block like it's hot" if block_given?
+      raise "Error: " unless self.current_question
+      
+      options = {:reference_identifier => reference_identifier}
+      self.current_question.dependency = (self.current_dependency = Dependency.new(current_question, args, options))
+      
+    when "condition", "c"
+      puts "          Condition: #{reference_identifier}"
+      raise "Error, I'm dropping the block like it's hot" if block_given?
+      raise "Error: No current dependency" unless self.current_dependency
+      
+      options = {:reference_identifier => reference_identifier}
+      self.current_dependency.add_dependency_condition DependencyCondition.new(self, args, options)
+      
+    when "answer", "a"
+      puts "        Answer: #{reference_identifier}"
+      raise "Error, I'm dropping the block like it's hot" if block_given?
+      
+      if self.current_question_group and self.current_question_group.display_type == "grid"
+        options = {:reference_identifier => reference_identifier, :display_order => self.grid_answers.size + 1}
+        self.grid_answers << Answer.new(nil, args, options)
+      else
+        raise "Error: No current question" unless self.current_question
+        options = {:reference_identifier => reference_identifier, :display_order => self.current_question.answers.size + 1}
+        self.current_question.answers << Answer.new(self.current_question, args, options)
+      end
+      
+    else
+      raise "  ERROR: '#{m_name}' not valid method_missing name"
+    end
+  end
+
+  def clear_current_question_group
+    self.current_question_group = nil
+    self.grid_answers = []
+    self.current_question = nil
+  end
+
+  def clear_current_question
+    self.current_question = nil
+    self.current_dependency = nil
+  end
+
+  def add_grid_answers
+    self.grid_answers.each do |grid_answer|
+      my_answer = grid_answer.dup
+      my_answer.id = self.parser.new_answer_id
+      my_answer.question_id = self.current_question.id
+      my_answer.parser = self.parser
+      self.current_question.answers << my_answer
+    end
+  end
+  
+  # Used to find questions for dependency linking 
+  def find_question_by_reference(ref_id)
+    question = nil
+    count = 0
+    while question.nil? and count < questions.size
+      question = self.questions[count] if self.questions[count].reference_identifier == ref_id
+      count +=1
+    end
+    question
+  end
+  
+
+  def to_yml
+    out =[ %(#{@data_export_identifier}_#{@id}:) ]
+    out << %(  id: #{@id})
+    out << %(  survey_id: #{@survey_id})
+    out << %(  title: "#{@title}")
+    out << %(  description: "#{@description}")
+    out << %(  display_order: #{display_order})
+    out << %(  reference_identifier: #{@reference_identifier})
+    out << %(  data_export_identifier: "#{@data_export_identifier}")
+    (out << nil ).join("\r\n")
+  end
+
+  def to_file
+    File.open(self.parser.survey_sections_yml, File::CREAT|File::APPEND|File::WRONLY) {|f| f << to_yml}
+    self.question_groups.compact.map(&:to_file)
+    self.questions.compact.map(&:to_file)
+  end
+
+end

data/script/surveyor/whr_dsl.tmproj

@@ -0,0 +1,244 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+<dict>
+	<key>currentDocument</key>
+	<string>whrparse.rb</string>
+	<key>documents</key>
+	<array>
+		<dict>
+			<key>expanded</key>
+			<true/>
+			<key>name</key>
+			<string>whr_dsl</string>
+			<key>regexFolderFilter</key>
+			<string>!.*/(\.[^/]*|CVS|_darcs|_MTN|\{arch\}|blib|.*~\.nib|.*\.(framework|app|pbproj|pbxproj|xcode(proj)?|bundle))$</string>
+			<key>sourceDirectory</key>
+			<string></string>
+		</dict>
+	</array>
+	<key>fileHierarchyDrawerWidth</key>
+	<integer>261</integer>
+	<key>metaData</key>
+	<dict>
+		<key>answer.rb</key>
+		<dict>
+			<key>caret</key>
+			<dict>
+				<key>column</key>
+				<integer>5</integer>
+				<key>line</key>
+				<integer>54</integer>
+			</dict>
+			<key>firstVisibleColumn</key>
+			<integer>0</integer>
+			<key>firstVisibleLine</key>
+			<integer>13</integer>
+		</dict>
+		<key>columnizer.rb</key>
+		<dict>
+			<key>caret</key>
+			<dict>
+				<key>column</key>
+				<integer>112</integer>
+				<key>line</key>
+				<integer>3</integer>
+			</dict>
+			<key>firstVisibleColumn</key>
+			<integer>0</integer>
+			<key>firstVisibleLine</key>
+			<integer>0</integer>
+		</dict>
+		<key>partials/_checkboxes.rhtml</key>
+		<dict>
+			<key>caret</key>
+			<dict>
+				<key>column</key>
+				<integer>0</integer>
+				<key>line</key>
+				<integer>0</integer>
+			</dict>
+			<key>firstVisibleColumn</key>
+			<integer>0</integer>
+			<key>firstVisibleLine</key>
+			<integer>0</integer>
+		</dict>
+		<key>partials/_date.rhtml</key>
+		<dict>
+			<key>caret</key>
+			<dict>
+				<key>column</key>
+				<integer>0</integer>
+				<key>line</key>
+				<integer>0</integer>
+			</dict>
+			<key>firstVisibleColumn</key>
+			<integer>0</integer>
+			<key>firstVisibleLine</key>
+			<integer>0</integer>
+		</dict>
+		<key>question.rb</key>
+		<dict>
+			<key>caret</key>
+			<dict>
+				<key>column</key>
+				<integer>68</integer>
+				<key>line</key>
+				<integer>68</integer>
+			</dict>
+			<key>firstVisibleColumn</key>
+			<integer>0</integer>
+			<key>firstVisibleLine</key>
+			<integer>46</integer>
+		</dict>
+		<key>registry_code.rb</key>
+		<dict>
+			<key>caret</key>
+			<dict>
+				<key>column</key>
+				<integer>49</integer>
+				<key>line</key>
+				<integer>19</integer>
+			</dict>
+			<key>firstVisibleColumn</key>
+			<integer>0</integer>
+			<key>firstVisibleLine</key>
+			<integer>0</integer>
+		</dict>
+		<key>section.rb</key>
+		<dict>
+			<key>caret</key>
+			<dict>
+				<key>column</key>
+				<integer>9</integer>
+				<key>line</key>
+				<integer>98</integer>
+			</dict>
+			<key>firstVisibleColumn</key>
+			<integer>0</integer>
+			<key>firstVisibleLine</key>
+			<integer>59</integer>
+		</dict>
+		<key>section_template.rhtml</key>
+		<dict>
+			<key>caret</key>
+			<dict>
+				<key>column</key>
+				<integer>134</integer>
+				<key>line</key>
+				<integer>3</integer>
+			</dict>
+			<key>firstVisibleColumn</key>
+			<integer>0</integer>
+			<key>firstVisibleLine</key>
+			<integer>0</integer>
+		</dict>
+		<key>specs/answer_spec.rb</key>
+		<dict>
+			<key>caret</key>
+			<dict>
+				<key>column</key>
+				<integer>50</integer>
+				<key>line</key>
+				<integer>24</integer>
+			</dict>
+			<key>columnSelection</key>
+			<false/>
+			<key>firstVisibleColumn</key>
+			<integer>0</integer>
+			<key>firstVisibleLine</key>
+			<integer>3</integer>
+			<key>selectFrom</key>
+			<dict>
+				<key>column</key>
+				<integer>60</integer>
+				<key>line</key>
+				<integer>24</integer>
+			</dict>
+			<key>selectTo</key>
+			<dict>
+				<key>column</key>
+				<integer>50</integer>
+				<key>line</key>
+				<integer>24</integer>
+			</dict>
+		</dict>
+		<key>specs/question_spec.rb</key>
+		<dict>
+			<key>caret</key>
+			<dict>
+				<key>column</key>
+				<integer>21</integer>
+				<key>line</key>
+				<integer>5</integer>
+			</dict>
+			<key>firstVisibleColumn</key>
+			<integer>0</integer>
+			<key>firstVisibleLine</key>
+			<integer>0</integer>
+		</dict>
+		<key>specs/section_spec.rb</key>
+		<dict>
+			<key>caret</key>
+			<dict>
+				<key>column</key>
+				<integer>43</integer>
+				<key>line</key>
+				<integer>43</integer>
+			</dict>
+			<key>firstVisibleColumn</key>
+			<integer>0</integer>
+			<key>firstVisibleLine</key>
+			<integer>0</integer>
+		</dict>
+		<key>specs/whrparse_spec.rb</key>
+		<dict>
+			<key>caret</key>
+			<dict>
+				<key>column</key>
+				<integer>5</integer>
+				<key>line</key>
+				<integer>6</integer>
+			</dict>
+			<key>firstVisibleColumn</key>
+			<integer>0</integer>
+			<key>firstVisibleLine</key>
+			<integer>0</integer>
+		</dict>
+		<key>whrparse.rb</key>
+		<dict>
+			<key>caret</key>
+			<dict>
+				<key>column</key>
+				<integer>9</integer>
+				<key>line</key>
+				<integer>20</integer>
+			</dict>
+			<key>firstVisibleColumn</key>
+			<integer>0</integer>
+			<key>firstVisibleLine</key>
+			<integer>0</integer>
+		</dict>
+	</dict>
+	<key>openDocuments</key>
+	<array>
+		<string>registry_code.rb</string>
+		<string>whrparse.rb</string>
+		<string>specs/answer_spec.rb</string>
+		<string>specs/question_spec.rb</string>
+		<string>specs/section_spec.rb</string>
+		<string>section.rb</string>
+		<string>section_template.rhtml</string>
+		<string>partials/_date.rhtml</string>
+		<string>partials/_checkboxes.rhtml</string>
+		<string>columnizer.rb</string>
+		<string>specs/whrparse_spec.rb</string>
+		<string>question.rb</string>
+		<string>answer.rb</string>
+	</array>
+	<key>showFileHierarchyDrawer</key>
+	<true/>
+	<key>windowFrame</key>
+	<string>{{282, 61}, {1119, 816}}</string>
+</dict>
+</plist>

data/spec/controllers/surveyor_controller_spec.rb

@@ -0,0 +1,328 @@
+require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
+
+describe SurveyorController do
+  
+  # map.available_surveys 'surveys',                                        :conditions => {:method => :get}, :action => "new"      # GET survey list
+  # map.take_survey       'surveys/:survey_code',                           :conditions => {:method => :post}, :action => "create"  # Only POST of survey to create
+  # map.view_my_survey    'surveys/:survey_code/:response_set_code',        :conditions => {:method => :get}, :action => "show"     # GET viewable/printable? survey
+  # map.edit_my_survey    'surveys/:survey_code/:response_set_code/take',   :conditions => {:method => :get}, :action => "edit"     # GET editable survey 
+  # map.update_my_survey  'surveys/:survey_code/:response_set_code',        :conditions => {:method => :put}, :action => "update"   # PUT edited survey 
+  # map.finish_my_survey  'surveys/:survey_code/:response_set_code/finish', :conditions => {:method => :put}, :action => "finish"   # PUT to close out the response_set
+  
+  describe "handling GET /surveys (available_surveys)" do
+
+    before(:each) do
+      @survey = mock_model(Survey)
+      Survey.stub!(:find).and_return([@survey])
+    end
+  
+    def do_get
+      get :new
+    end
+  
+    it "should be successful" do
+      do_get
+      response.should be_success
+    end
+
+    it "should render index template" do
+      do_get
+      response.should render_template('new')
+    end
+  
+    it "should find all surveys" do
+      Survey.should_receive(:find).with(:all).and_return([@survey])
+      do_get
+    end
+  
+    it "should assign the found surveys for the view" do
+      do_get
+      assigns[:surveys].should == [@survey]
+    end
+  end
+
+  describe "handling GET /surveys.xml (available_surveys)" do
+    before(:each) do
+      @surveys = mock("Array of Surveys", :to_xml => "XML")
+      Survey.stub!(:find).and_return(@surveys)
+    end
+  
+    def do_get
+      @request.env["HTTP_ACCEPT"] = "application/xml"
+      get :new
+    end
+  
+    it "should be successful" do
+      do_get
+      response.should be_success
+    end
+
+    it "should find all surveys" do
+      Survey.should_receive(:find).with(:all).and_return(@surveys)
+      do_get
+    end
+  
+    it "should render the found surveys as xml" do
+      @surveys.should_receive(:to_xml).and_return("XML")
+      do_get
+      response.body.should == "XML"
+    end
+  end
+
+  describe "handling POST /surveys/XYZ (take_survey)" do
+
+    before(:each) do
+      @survey = mock_model(Survey, :access_code => "XYZ")
+      @response_set = mock_model(ResponseSet, :access_code => "PDQ")
+      ResponseSet.stub!(:new).and_return(@response_set)
+      Survey.stub!(:find_by_access_code).and_return(@survey)
+    end
+    
+    describe "with successful save" do
+  
+      def do_post
+        @response_set.should_receive(:save!).and_return(true)
+        post :create, :survey_code => "XYZ"
+      end
+      
+      it "should look for the survey" do
+        Survey.should_receive(:find_by_access_code).with("XYZ").and_return(@survey)
+        do_post
+      end
+      it "should create a new response_set" do
+        ResponseSet.should_receive(:new).and_return(@response_set)
+        do_post
+      end
+
+      it "should redirect to the new response_set" do
+        do_post
+        response.should redirect_to(edit_my_survey_url(:survey_code => "XYZ", :response_set_code  => "PDQ"))
+      end
+      
+    end
+    
+    describe "with failed save" do
+
+      def do_post
+        @response_set.should_receive(:save!).and_return(false)
+        post :create, :survey_code => "XYZ"
+      end
+  
+      it "should re-redirect to 'new'" do
+        do_post
+        response.should redirect_to(available_surveys_url)
+      end
+      
+    end
+    
+    describe "with survey not found" do
+
+      def do_post
+        Survey.should_receive(:find_by_access_code).and_return(nil)
+        post :create, :survey_code => "XYZ"
+      end
+  
+      it "should re-redirect to 'new'" do
+        do_post
+        response.should redirect_to(available_surveys_url)
+      end
+      
+    end
+  end
+
+  describe "handling GET /surveys/XYZ/PDQ (view_my_survey)" do
+
+    before(:each) do
+      @survey = mock_model(Survey, :access_code => "XYZ", :sections => [mock_model(SurveySection)])  
+      @response_set = mock_model(ResponseSet, :access_code => "PDQ")
+      ResponseSet.stub!(:find_by_access_code).with("PDQ").and_return(@response_set)
+      @response_set.stub!(:survey).and_return(@survey)
+    end
+  
+    def do_get
+      get :show, :survey_code => "XYZ", :response_set_code => "PDQ"
+    end
+
+    it "should be successful" do
+      do_get
+      response.should be_success
+    end
+  
+    it "should render show template" do
+      do_get
+      response.should render_template('show')
+    end
+  
+    it "should find the response_set requested" do
+      ResponseSet.should_receive(:find_by_access_code).with("PDQ").and_return(@response_set)
+      do_get
+    end
+  
+    it "should assign the found response_set and survey for the view" do
+      do_get
+      assigns[:response_set].should equal(@response_set)
+      assigns[:survey].should equal(@survey)
+    end
+    
+    it "should redirect if :response_code not found" do
+      get :show, :survey_code => "XYZ", :response_set_code => "DIFFERENT"
+      response.should redirect_to(available_surveys_url)      
+    end
+    
+    # I'm not sure this is enterly neccessary since we look up the survey from the response_code in the url -BC
+    it "should redirect if :survey_code in url doesn't match response_set.survey.access_code" do
+      pending
+      get :show, :survey_code => "DIFFERENT", :response_set_code => "PDQ"
+      response.should redirect_to(available_surveys_url)
+    end
+  end
+
+  describe "handling GET /surveys/XYZ/PDQ/take (edit_my_survey)" do
+
+    before(:each) do
+      @survey = mock_model(Survey, :access_code => "XYZ")
+      @survey_section = mock_model(SurveySection)
+      @survey.stub!(:sections).and_return([@survey_section])
+      @response_set = mock_model(ResponseSet, :access_code => "PDQ")
+      ResponseSet.stub!(:find_by_access_code).with("PDQ").and_return(@response_set)     
+      @response_set.stub!(:survey).and_return(@survey)
+    end
+
+    it "should be successful, render edit with the requested survey" do
+      ResponseSet.should_receive(:find_by_access_code).with("PDQ").and_return(@response_set)
+        
+      get :edit, :survey_code => "XYZ", :response_set_code => "PDQ"
+      response.should be_success
+      response.should render_template('edit')
+      assigns[:response_set].should equal(@response_set)
+      assigns[:survey].should equal(@survey)
+    end
+    
+    it "should redirect if :response_code not found" do
+      get :edit, :survey_code => "XYZ", :response_set_code => "DIFFERENT"
+      response.should redirect_to(available_surveys_url)      
+    end
+
+  end
+
+  describe "handling PUT /surveys/XYZ/PDQ (update_my_survey)" do
+
+    before(:each) do
+      @survey = mock_model(Survey, :access_code => "XYZ", :sections => [mock_model(SurveySection)])
+      @response_set = mock_model(ResponseSet, :access_code => "PDQ")
+      ResponseSet.stub!(:find_by_access_code).with("PDQ").and_return(@response_set)
+      @response_set.stub!(:survey).and_return(@survey)
+      @response_set.stub!(:add_responses).and_return(true)
+    end
+    
+    describe "with no response_set in update" do
+
+      it "should find the response set requested" do
+        ResponseSet.should_receive(:find_by_access_code).with("PDQ").and_return(@response_set)
+        put :update, :survey_code => "XYZ", :response_set_code => "PDQ"
+        
+      end
+    
+    end
+    
+    describe "with a new response set" do
+      
+      it "should accept properly formatted params and save the data" do
+        response_set_params = {"survey_code"=>"XYZ", "response_set"=>{"new_response_attributes"=>{"1"=>[{"answer_id"=>"2"}]}}}
+        
+        
+      end
+      
+      # describe "issue with posting data to an existing survey and the data not saving properly" do
+      #    @survey = mock_model(Survey, :access_code => "XYZ")
+      #    @response_set = mock_model(ResponseSet, :access_code => "PDQ")
+      #    ResponseSet.stub!(:find_by_access_code).with("PDQ").and_return(@response_set)
+      #    @response_set.stub!(:survey).and_return(@survey)
+      #    @response_set.stub!(:complete!).and_return(Time.now)
+      #    
+      #    
+      #  end
+
+      # first post {"survey_code"=>"test_survey", "commit"=>"Next Section (Utensiles and you!) >>", "response_set"=>{"new_response_attributes"=>{"1"=>[{"answer_id"=>"2"}, {"answer_id"=>"0", "string_value"=>""}], "2"=>[{"answer_id"=>"6"}], "3"=>[{"answer_id"=>"10"}], "4"=>[{"answer_id"=>"14"}], "5"=>[{"answer_id"=>"0"}, {"answer_id"=>"0"}]}, "existing_response_attributes"=>{"6"=>{"1"=>{"answer_id"=>"20", "string_value"=>"B"}}, "7"=>{"2"=>{"text_value"=>"foo", "answer_id"=>"21"}}, "5"=>{"7"=>{"answer_id"=>"17"}, "16"=>{"answer_id"=>"19", "string_value"=>"blah"}}}}, "authenticity_token"=>"d9ba68fe20a46703f3737b5cb0b7e17b7390de32", "_method"=>"put", "action"=>"update", "controller"=>"app", "response_set_code"=>"9VEsec1dK6", "section"=>"2"}
+      # second post {"survey_code"=>"test_survey", "commit"=>"Next Section (Utensiles and you!) >>", "response_set"=>{"new_response_attributes"=>{"1"=>[{"answer_id"=>"2"}, {"answer_id"=>"0", "string_value"=>""}], "2"=>[{"answer_id"=>"6"}], "3"=>[{"answer_id"=>"10"}], "4"=>[{"answer_id"=>"14"}], "5"=>[{"answer_id"=>"0"}, {"answer_id"=>"0"}]}, "existing_response_attributes"=>{"6"=>{"1"=>{"answer_id"=>"20", "string_value"=>"B"}}, "7"=>{"2"=>{"text_value"=>"boooo", "answer_id"=>"21"}}, "5"=>{"7"=>{"answer_id"=>"17"}, "16"=>{"answer_id"=>"19", "string_value"=>"blahblahstink"}}}}, "authenticity_token"=>"d9ba68fe20a46703f3737b5cb0b7e17b7390de32", "_method"=>"put", "action"=>"update", "controller"=>"app", "response_set_code"=>"9VEsec1dK6", "section"=>"2"}
+
+      
+    end
+    
+    
+    describe "with failed update" do
+
+      it "should re-render 'edit'" do
+        put :update, :survey_code => "XYZ", :response_set_code => "PDQ"
+        response.should be_success
+        response.should render_template('edit')
+        flash[:notice].should == "Unable to update survey"
+      end
+
+    end
+  end
+
+  describe "handling PUT /surveys/XYZ/PDQ/finish (finish_my_survey)" do
+
+    before(:each) do
+      @survey = mock_model(Survey, :access_code => "XYZ", :sections => [mock_model(SurveySection)])
+      @response_set = mock_model(ResponseSet, :access_code => "PDQ")
+      ResponseSet.stub!(:find_by_access_code).with("PDQ").and_return(@response_set)
+      @response_set.stub!(:survey).and_return(@survey)
+      @response_set.stub!(:complete!).and_return(Time.now)
+    end
+    
+    describe "with successful update" do
+
+      def do_put
+        put :finish, :survey_code => "XYZ", :response_set_code => "PDQ"
+      end
+
+      it "should find the response_set requested" do
+        ResponseSet.should_receive(:find_by_access_code).with("PDQ").and_return(@response_set)
+        do_put
+      end
+
+      it "should update the found response set" do
+        @response_set.should_receive(:complete!).and_return(Time.now)
+        do_put
+      end
+
+      it "should assign the found response set and survey for the view" do
+        do_put
+        assigns(:response_set).should equal(@response_set)
+        assigns(:survey).should equal(@response_set.survey)
+      end
+
+      it "should render the 'edit' template" do
+        do_put
+        response.should render_template('edit')
+        flash[:notice].should == "Completed survey"
+      end
+      
+      it "should redirect to available surveys if :response_code not found" do
+        put :update, :survey_code => "XYZ", :response_set_code => "DIFFERENT"
+        response.should redirect_to(available_surveys_url)
+        flash[:notice].should == "Unable to find your responses to the survey"
+      end
+
+    end
+    
+    describe "with failed update" do
+
+      def do_put
+        put :finish, :survey_code => "XYZ", :response_set_code => "PDQ"
+      end
+
+      it "should re-render 'edit'" do
+        @response_set.should_receive(:complete!).and_return(false)
+        do_put
+        response.should render_template('edit')
+        flash[:notice].should == "Unable to complete survey"
+      end
+
+    end
+  end
+  
+
+end