added in testing for relations between models

app/models/work.rb

@@ -1,5 +1,5 @@
 class Work < ApplicationRecord
-  validates :title, presence: true
+  validates :title, presence: true, uniqueness: true
 
   has_many :votes
   has_many :users, through: :votes

test/models/user_test.rb

@@ -40,4 +40,21 @@
     end
   end
 
+  describe 'relations' do
+    it 'has many votes' do
+      user = users(:harry)
+      vote = Vote.create!(work_id: 3, user_id: 1)
+
+      expect(user.votes.first).must_be_instance_of Vote
+      expect(user.votes.count > 1).must_equal true
+    end
+
+    it 'has many users through votes' do
+      user = users(:harry)
+      vote = Vote.create!(work_id: 3, user_id: 1)
+
+      expect(user.works.first).must_be_instance_of Work
+      expect(user.works.count > 1).must_equal true
+    end
+  end
 end

test/models/vote_test.rb

@@ -36,10 +36,68 @@
       # Assert
       expect(duplicate_vote.valid?).must_equal false
       expect(duplicate_vote.errors.messages).must_include :work_id
-      expect(duplicate_vote.errors.messages[:work_id]).must_equal ["User cannot upvote work more than once"]
     end
+  end
 
+  describe 'relations' do
+    it 'can set the user through user' do
+      # Create two models
+      user = User.create!(name: "test user")
+      vote = Vote.new(work_id: 1, user_id: 4)
 
-  end
+      # Make the models relate to one another
+      vote.user = user
+
+      # author_id should have changed accordingly
+      expect(vote.user).must_equal user
+    end
+
+    it 'can set the user through "user_id"' do
+      # Create two models
+      user = User.create!(name: "test user")
+      vote = Vote.new(work_id: 1, user_id: 4)
+
+      # Make the models relate to one another
+      vote.user_id = user.id
+
+      # author_id should have changed accordingly
+      expect(vote.user_id).must_equal user.id
+    end
+
+    it 'can set the work through work' do
+      # Create two models
+      work = works(:world)
+      vote = Vote.new(work_id: 1, user_id: 4)
+
+      # Make the models relate to one another
+      vote.work = work
 
+      # author_id should have changed accordingly
+      expect(vote.work).must_equal work
+    end
+
+    it 'can set the work through "work_id"' do
+      # Create two models
+      work = works(:world)
+      vote = Vote.new(work_id: 1, user_id: 4)
+
+      # Make the models relate to one another
+      vote.work_id = work.id
+
+      # author_id should have changed accordingly
+      expect(vote.work_id).must_equal work.id
+    end
+
+    it "belongs to one user" do
+      vote = votes(:first)
+      expect(vote.user).must_be_instance_of User
+      expect(vote.user).must_equal users(:harry)
+    end
+
+    it "belongs to one work" do
+      vote = votes(:first)
+      expect(vote.work).must_be_instance_of Work
+      expect(vote.work).must_equal works(:gods)
+    end
+  end
 end

test/models/work_test.rb

@@ -45,8 +45,24 @@
       expect(new_work_2.errors.messages).must_include :title
       expect(new_work_2.errors.messages[:title]).must_equal ["has already been taken"]
     end
-
   end
 
+  describe 'relations' do
+    it 'has many votes' do
+      work = works(:gods)
+      vote = Vote.create!(work_id: 1, user_id: 4)
+
+      expect(work.votes.first).must_be_instance_of Vote
+      expect(work.votes.count > 1).must_equal true
+    end
+
+    it 'has many users through votes' do
+      work = works(:gods)
+      vote = Vote.create!(work_id: 1, user_id: 4)
+
+      expect(work.users.first).must_be_instance_of User
+      expect(work.users.count > 1).must_equal true
+    end
+  end
 
 end