Hoang Phuc Do · @phucdh · Trong Huu Nguyen · Hoang Phuc Do · Trong Huu Nguyen · Hoang Phuc Do
--- a/spec/models/order_spec.rb
+++ b/spec/models/order_spec.rb
    it { is_expected.to belong_to(:user) }
    it { is_expected.to have_many(:product_items) }
  end
+
+  context 'custom methods' do
+    let(:user) { create(:user) }
+    let!(:order) { create(:order, user: user) }
+    let!(:product) { create(:product, quantity: 3, price: 10) }
+    let!(:product_item) { create(:product_item, product: product, order: order) }
+
+    it 'has valid total_price' do
+      expect(order.total_price).to eq 10
--- a/spec/models/product_item_spec.rb
+++ b/spec/models/product_item_spec.rb
    it { is_expected.to belong_to(:order) }
    it { is_expected.to belong_to(:product) }
  end
+
+  context 'custom methods' do
+    let!(:product) { create(:product, price: 10, quantity: 3) }
+    let!(:product_item) { create(:product_item, product: product) }
+
+    it 'has valid total_price' do
+      expect(product_item.total_price).to eq 10
+    end
+
+    it 'has valid update_product_quantity' do
+      product_item.update_product_quantity
+      expect(product.quantity).to eq 2
--- a/spec/models/product_item_spec.rb
+++ b/spec/models/product_item_spec.rb
    it { is_expected.to belong_to(:order) }
    it { is_expected.to belong_to(:product) }
  end
+
+  context 'custom methods' do
+    let!(:product) { create(:product, price: 10, quantity: 3) }
+    let!(:product_item) { create(:product_item, product: product) }
+
+    it 'has valid total_price' do
+      expect(product_item.total_price).to eq 10
--- a/spec/models/product_spec.rb
+++ b/spec/models/product_spec.rb
    it { is_expected.to belong_to(:user) }
    it { is_expected.to belong_to(:category) }
  end
+
+  context 'custom methods' do
+    let!(:user) { create(:user) }
+    let!(:other_user) { create(:user) }
+    let!(:product) { create(:product, quantity: 3, user: user) }
+
+    it 'has valid in_stock?' do
+      expect(product.in_stock?(2)).to be_truthy
--- a/spec/models/product_spec.rb
+++ b/spec/models/product_spec.rb
    it { is_expected.to belong_to(:user) }
    it { is_expected.to belong_to(:category) }
  end
+
+  context 'custom methods' do
+    let!(:user) { create(:user) }
+    let!(:other_user) { create(:user) }
+    let!(:product) { create(:product, quantity: 3, user: user) }
+
+    it 'has valid in_stock?' do
+      expect(product.in_stock?(2)).to be_truthy
+    end
+
+    it 'has invalid in_stock?' do
+      expect(product.in_stock?(4)).not_to be_truthy