Section 6.5.5: Associate users to tickets when tickets are created

Author: Ryan Bigg

ticketee/app/controllers/application_controller.rb

@@ -14,4 +14,12 @@ def signed_in?
     env["warden"].authenticated?
   end
   helper_method :signed_in?
+
+  def require_sign_in!
+    unless current_user.present?
+      message = "You need to sign in or sign up before continuing."
+      flash[:warning] = message
+      redirect_to sign_in_url
+    end
+  end
 end

ticketee/app/controllers/tickets_controller.rb

@@ -1,4 +1,5 @@
 class TicketsController < ApplicationController
+  before_action :require_sign_in!, except: [:show, :index]
   before_action :set_project
   before_action :set_ticket, only: [:show, :edit, :update, :destroy]
 
@@ -8,6 +9,7 @@ def new
 
   def create
     @ticket = @project.tickets.build(ticket_params)
+    @ticket.user = current_user
 
     if @ticket.save
       flash[:success] = "Ticket has been created."

ticketee/app/models/ticket.rb

@@ -1,5 +1,6 @@
 class Ticket < ActiveRecord::Base
   belongs_to :project
+  belongs_to :user
 
   validates :title, presence: true
   validates :description, presence: true,

ticketee/app/views/tickets/show.html.erb

@@ -1,5 +1,6 @@
 <div id='ticket'>
   <h2><%= @ticket.title %></h2>
+  <span id='author'>Created by <%= @ticket.user.email %></span>
   <%= link_to "Edit Ticket",
       [:edit, @project, @ticket],
       class: "edit" %>

ticketee/db/migrate/20141113064950_add_user_id_to_tickets.rb

@@ -0,0 +1,5 @@
+class AddUserIdToTickets < ActiveRecord::Migration
+  def change
+    add_reference :tickets, :user, index: true
+  end
+end

ticketee/db/schema.rb

@@ -26,9 +26,11 @@
     t.integer  "project_id"
     t.datetime "created_at",  null: false
     t.datetime "updated_at",  null: false
+    t.integer  "user_id"
   end
 
   add_index "tickets", ["project_id"], name: "index_tickets_on_project_id"
+  add_index "tickets", ["user_id"], name: "index_tickets_on_user_id"
 
   create_table "users", force: true do |t|
     t.string   "name"

ticketee/spec/features/creating_tickets_spec.rb

@@ -1,11 +1,22 @@
 require "rails_helper"
 
 feature "Creating Tickets" do
+  let(:user) { FactoryGirl.create(:user) }
+
   before do
-    FactoryGirl.create(:project, name: "Internet Explorer")
+    project = FactoryGirl.create(:project, name: "Internet Explorer")
 
     visit '/'
-    click_link "Internet Explorer"
+    click_link project.name
+    click_link "New Ticket"
+    message = "You need to sign in or sign up before continuing."
+    expect(page).to have_content(message)
+
+    fill_in "Email", with: user.email
+    fill_in "Password", with: 'password'
+    click_button "Sign in"
+
+    click_link project.name
     click_link "New Ticket"
   end
 
@@ -15,6 +26,9 @@
     click_button "Create Ticket"
 
     expect(page).to have_content("Ticket has been created.")
+    within "#ticket #author" do
+      expect(page).to have_content("Created by #{user.email}")
+    end
   end
 
   scenario "Creating a ticket without valid attributes fails" do

ticketee/spec/features/deleting_tickets_spec.rb

@@ -2,9 +2,13 @@
 
 feature "Deleting tickets" do
   let!(:project) { FactoryGirl.create(:project) }
-  let!(:ticket) { FactoryGirl.create(:ticket, project: project) }
+  let!(:user) { FactoryGirl.create(:user) }
+  let!(:ticket) do
+    FactoryGirl.create(:ticket, project: project, user: user)
+  end
 
   before do
+    login_as(user)
     visit "/"
     click_link project.name
     click_link ticket.title

ticketee/spec/features/editing_tickets_spec.rb

@@ -2,9 +2,13 @@
 
 feature "Editing tickets" do
   let!(:project) { FactoryGirl.create(:project) }
-  let!(:ticket) { FactoryGirl.create(:ticket, project: project) }
+  let!(:user) { FactoryGirl.create(:user) }
+  let!(:ticket) do
+    FactoryGirl.create(:ticket, project: project, user: user)
+  end
 
   before do
+    login_as(user)
     visit "/"
     click_link project.name
     click_link ticket.title

ticketee/spec/features/viewing_tickets_spec.rb

@@ -2,18 +2,26 @@
 
 feature "Viewing tickets" do
   before do
-    sublime = FactoryGirl.create(:project, name: "Sublime Text 3")
+    sublime = FactoryGirl.create(:project,
+      name: "Sublime Text 3")
 
-    FactoryGirl.create(:ticket,
+    user = FactoryGirl.create(:user)
+    ticket = FactoryGirl.create(:ticket,
       project: sublime,
       title: "Make it shiny!",
-      description: "Gradients! Starbursts! Oh my!")
+      description: "Gradients! Starbursts! Oh my!",
+      user: user
+    )
+
+    ie = FactoryGirl.create(:project,
+      name: "Internet Explorer")
 
-    ie = FactoryGirl.create(:project, name: "Internet Explorer")
     FactoryGirl.create(:ticket,
       project: ie,
       title: "Standards compliance",
-      description: "Isn't a joke.")
+      description: "Isn't a joke.",
+      user: user
+    )
 
     visit "/"
   end