Including contamination reports to enable survivors report contamination of other survivors

Author: andreealencar

app/blueprinter/contamination_report_blueprint.rb

@@ -0,0 +1,5 @@
+class ContaminationReportBlueprint < Blueprinter::Base  
+  #== ASSOCIATIONS =========================================
+  association :accuser, blueprint: SurvivorBlueprint
+  association :suspect, blueprint: SurvivorBlueprint
+end

app/controllers/api/v1/contamination_reports_controller.rb

@@ -0,0 +1,20 @@
+module Api::V1
+  class ContaminationReportsController < ApplicationController
+    # POST /contamination_report/
+    def create
+      @contamination_report = ContaminationReport.new(contamination_report_params)
+
+      if @contamination_report.save
+        render json: ContaminationReportBlueprint.render(@contamination_report), status: :created
+      else
+        render json: @contamination_report.errors, status: :unprocessable_entity
+      end
+    end
+
+    private
+
+    def contamination_report_params
+      params.require(:contamination_report).permit(:accuser_id, :suspect_id)
+    end  
+  end
+end

app/models/contamination_report.rb

@@ -0,0 +1,28 @@
+class ContaminationReport < ApplicationRecord
+  #== ASSOCIATIONS =========================================
+  belongs_to :accuser, class_name: 'Survivor'
+  belongs_to :suspect, class_name: 'Survivor'
+
+  #== VALIDATIONS ==========================================
+  validates :accuser, :suspect, presence: true
+  validates :suspect, uniqueness: { scope: :accuser_id }
+
+  validate :accusations_count
+  
+  #== CALLBACKS ============================================
+  after_save :update_survivor
+
+  def accusations_count
+    total_accusations = suspect&.accusers&.count || 0
+    if total_accusations > 2
+      errors.add(:suspect, "have 3 accuations limit")
+    end
+  end
+
+  def update_survivor
+    total_accusations = suspect&.accusers&.count || 0
+    if total_accusations > 2
+      suspect.update(infected: true)
+    end
+  end
+end

app/models/survivor.rb

@@ -3,6 +3,9 @@ class Survivor < ApplicationRecord
   has_one :last_location, class_name: "Location", foreign_key: :survivor_id
   has_many :inventory_items
   has_many :items, through: :inventory_items
+
+  has_many :accusers, class_name: 'ContaminationReport', foreign_key: :suspect_id
+  has_many :suspects, class_name: 'ContaminationReport', foreign_key: :accuser_id
 
   #== ACCEPTED ATTRIBUTES ==================================
   enum gender: { female: 'female', male: 'male' }

config/routes.rb

@@ -6,6 +6,8 @@
         put :last_location, on: :member
         post :trade, on: :collection
       end
+
+      resources :contamination_reports, only: :create
     end
   end
 

db/migrate/20200120034704_create_contamination_reports.rb

@@ -0,0 +1,10 @@
+class CreateContaminationReports < ActiveRecord::Migration[6.0]
+  def change
+    create_table :contamination_reports do |t|
+      t.references :accuser, null: false, foreign_key: { to_table: :survivors }
+      t.references :suspect, null: false, foreign_key: { to_table: :survivors }
+
+      t.timestamps
+    end
+  end
+end

db/migrate/20200120035617_add_infected_flag_to_survivors.rb

@@ -0,0 +1,6 @@
+class AddInfectedFlagToSurvivors < ActiveRecord::Migration[6.0]
+  def change
+    add_column :survivors, :infected, :boolean, default: false 
+    add_index :contamination_reports, [:accuser_id, :suspect_id], unique: true
+  end
+end

db/schema.rb

@@ -10,11 +10,21 @@
 #
 # It's strongly recommended that you check this file into your version control system.
 
-ActiveRecord::Schema.define(version: 2020_01_17_140644) do
+ActiveRecord::Schema.define(version: 2020_01_20_035617) do
 
   # These are extensions that must be enabled in order to support this database
   enable_extension "plpgsql"
 
+  create_table "contamination_reports", force: :cascade do |t|
+    t.bigint "accuser_id", null: false
+    t.bigint "suspect_id", null: false
+    t.datetime "created_at", precision: 6, null: false
+    t.datetime "updated_at", precision: 6, null: false
+    t.index ["accuser_id", "suspect_id"], name: "index_contamination_reports_on_accuser_id_and_suspect_id", unique: true
+    t.index ["accuser_id"], name: "index_contamination_reports_on_accuser_id"
+    t.index ["suspect_id"], name: "index_contamination_reports_on_suspect_id"
+  end
+
   create_table "inventory_items", force: :cascade do |t|
     t.integer "quantity"
     t.bigint "survivor_id", null: false
@@ -45,8 +55,11 @@
     t.integer "age"
     t.datetime "created_at", precision: 6, null: false
     t.datetime "updated_at", precision: 6, null: false
+    t.boolean "infected", default: false
   end
 
+  add_foreign_key "contamination_reports", "survivors", column: "accuser_id"
+  add_foreign_key "contamination_reports", "survivors", column: "suspect_id"
   add_foreign_key "inventory_items", "items"
   add_foreign_key "inventory_items", "survivors"
   add_foreign_key "locations", "survivors"

spec/factories/contamination_reports.rb

@@ -0,0 +1,6 @@
+FactoryBot.define do
+  factory :contamination_report do
+    accuser { create(:survivor) }
+    suspect { create(:survivor) }
+  end
+end

spec/models/contamination_report_spec.rb

@@ -0,0 +1,30 @@
+require 'rails_helper'
+
+RSpec.describe ContaminationReport, type: :model do
+  describe "associations" do
+    it { should belong_to(:accuser).class_name("Survivor") }
+    it { should belong_to(:suspect).class_name("Survivor") }
+  end
+  
+  describe "validations" do
+    subject { create(:contamination_report) }
+
+    it { should validate_presence_of(:accuser) }
+    it { should validate_presence_of(:suspect) }
+
+    it { should validate_uniqueness_of(:suspect).scoped_to(:accuser_id) }
+
+    describe "custom validations" do
+      before(:each) do
+        @survivor = create(:survivor)
+        create_list(:contamination_report, 3, suspect: @survivor)
+      end
+
+      it "accusers limit" do
+        contamination_report = ContaminationReport.new(accuser: create(:survivor), suspect: @survivor)
+        expect(contamination_report.valid?).to eq(false)
+        expect(contamination_report.errors.messages).to eq({ suspect: ["have 3 accuations limit"] })
+      end
+    end
+  end
+end