asscociate quotes and users to companies

dalima-dev · Aug 26, 2024 · bbaf368 · bbaf368
1 parent 5fefbb3
commit bbaf368
Show file tree

Hide file tree

Showing 7 changed files with 28 additions and 1 deletion.
diff --git a/app/models/company.rb b/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
diff --git a/app/models/user.rb b/app/models/user.rb
@@ -1,3 +1,5 @@
 class User < ApplicationRecord
   devise :database_authenticatable, :validatable
+
+  belongs_to :company
 end
diff --git a/db/migrate/20240826230630_add_company_reference_to_quotes.rb b/db/migrate/20240826230630_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
diff --git a/db/migrate/20240826230631_add_company_reference_to_users.rb b/db/migrate/20240826230631_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
diff --git a/db/schema.rb b/db/schema.rb
diff --git a/test/fixtures/quotes.yml b/test/fixtures/quotes.yml
@@ -1,8 +1,11 @@
 first:
+  company: kpmg
   name: First quote
 
 second:
+  company: kpmg
   name: Second quote
 
 third:
+  company: kpmg
   name: Third quote
diff --git a/test/fixtures/users.yml b/test/fixtures/users.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') %>