company refs for users, quotes

Author: thejonroberts

app/models/company.rb

@@ -1,3 +1,6 @@
 class Company < ApplicationRecord
+  has_many :users, dependent: :destroy
+  has_many :quotes, dependent: :destroy
+
   validates :name, presence: true
 end

app/models/quote.rb

@@ -1,4 +1,6 @@
 class Quote < ApplicationRecord
+  belongs_to :company
+
   validates :name, presence: true
 
   scope :desc_id_ordered, -> { order(id: :desc) }

app/models/user.rb

@@ -1,5 +1,5 @@
 class User < ApplicationRecord
-  # Include default devise modules. Others available are:
-  # :confirmable, :lockable, :timeoutable, :trackable and :omniauthable
   devise :database_authenticatable, :validatable
+
+  belongs_to :company
 end

db/migrate/20240705021718_add_company_reference_to_quotes.rb

@@ -0,0 +1,5 @@
+class AddCompanyReferenceToQuotes < ActiveRecord::Migration[7.1]
+  def change
+    add_reference :quotes, :company, null: false, foreign_key: true
+  end
+end

db/migrate/20240705021730_add_company_reference_to_users.rb

@@ -0,0 +1,5 @@
+class AddCompanyReferenceToUsers < ActiveRecord::Migration[7.1]
+  def change
+    add_reference :users, :company, null: false, foreign_key: true
+  end
+end

db/schema.rb

@@ -1,8 +1,11 @@
 one:
+  company: kpmg
   name: First quote
 
 two:
+  company: kpmg
   name: Second quote
 
 three:
+  company: kpmg
   name: Third quote

test/fixtures/quotes.yml

@@ -1,11 +1,14 @@
 accountant:
+  company: kpmg
   email: [email protected]
   encrypted_password: <%= Devise::Encryptor.digest(User, 'password') %>
 
 manager:
+  company: kpmg
   email: [email protected]
   encrypted_password: <%= Devise::Encryptor.digest(User, 'password') %>
 
 eavesdropper:
+  company: pwc
   email: [email protected]
   encrypted_password: <%= Devise::Encryptor.digest(User, 'password') %>