Example for NWRUG visitor

risk.rb

require 'rspec'

class Risk

  def code(user)
    user_site = user.site[0..1]
    if department
      department_name = department.name[0..1]
    end
    [user_site,department_name,"001"].compact.join('-').upcase
  end

  # def department
  #   Department.new()
  #   sleep 10
  # end
end

# class Department
#   def initialize
#     sleep 10
#   end

#   def name
    
#   end
# end

describe Risk do
  let(:current_user) {mock "User", :site => "Atherton"}
  let(:department) {mock "Department", :name => "All Departments"}


  describe ".code" do
    context "with department" do
      it "returns AT-AL-0001" do
        subject.stub(:department).and_return(department)
        subject.code(current_user).should eql "AT-AL-001"
      end
    end
    context "without department" do
      it "returns AT-001" do
        subject.stub(:department)
        subject.code(current_user).should eql "AT-001"
      end
    end
  end
end