Methods to help on listing user projects

Rafael Manzo
1 parent f80c1659
Showing 4 changed files with 41 additions and 0 deletions Show diff stats
app/models/project_ownership.rb
app/models/user.rb
spec/models/project_ownership_spec.rb
spec/models/user_spec.rb
 class ProjectOwnership < ActiveRecord::Base
   belongs_to :user
   validates :project_id, presence: true
+
+  def project
+    Project.find(project_id)
+  end
 end
@@ -10,4 +10,9 @@ class User &lt; ActiveRecord::Base
   has_many :project_ownerships
   # Alert: when adding new parameters to this model, they should also be added to registrations_controller
+
+  def projects
+    #raise project_ownerships.inspect
+    project_ownerships.map { |project_ownership| project_ownership.project }
+  end
 end
@@ -4,4 +4,19 @@ describe ProjectOwnership do
   describe 'associations' do
     it { should belong_to(:user) }
   end
+
+  describe 'methods' do
+    describe 'project' do
+      subject {FactoryGirl.build(:project_ownership)}
+      let(:project) {FactoryGirl.build(:project)}
+
+      before :each do
+        Project.expects(:find).with(subject.project_id).returns(project)
+      end
+
+      it 'should return the project' do
+        subject.project.should eq(project)
+      end
+    end
+  end
 end
@@ -12,4 +12,21 @@ describe User do
   describe 'associations' do
     it { should have_many(:project_ownerships) }
   end
+
+  describe 'methods' do
+    describe 'projects' do
+      subject { FactoryGirl.build(:user) }
+      let(:project) {FactoryGirl.build(:project)}
+      let(:project_ownership) {FactoryGirl.build(:project_ownership)}
+
+      before :each do
+        project_ownership.expects(:project).returns(project)
+        subject.expects(:project_ownerships).returns([project_ownership])
+      end
+
+      it 'should return a list of projects owned by the user' do
+        subject.projects.should eq([project])
+      end
+    end    
+  end
 end
	@@ -10,4 +10,9 @@ class User < ActiveRecord::Base		@@ -10,4 +10,9 @@ class User < ActiveRecord::Base
10		10
11	has_many :project_ownerships	11	has_many :project_ownerships
12	# Alert: when adding new parameters to this model, they should also be added to registrations_controller	12	# Alert: when adding new parameters to this model, they should also be added to registrations_controller
		13	+
		14	+ def projects
		15	+ #raise project_ownerships.inspect
		16	+ project_ownerships.map { \|project_ownership\| project_ownership.project }
		17	+ end
13	end	18	end
	@@ -4,4 +4,19 @@ describe ProjectOwnership do		@@ -4,4 +4,19 @@ describe ProjectOwnership do
4	describe 'associations' do	4	describe 'associations' do
5	it { should belong_to(:user) }	5	it { should belong_to(:user) }
6	end	6	end
		7	+
		8	+ describe 'methods' do
		9	+ describe 'project' do
		10	+ subject {FactoryGirl.build(:project_ownership)}
		11	+ let(:project) {FactoryGirl.build(:project)}
		12	+
		13	+ before :each do
		14	+ Project.expects(:find).with(subject.project_id).returns(project)
		15	+ end
		16	+
		17	+ it 'should return the project' do
		18	+ subject.project.should eq(project)
		19	+ end
		20	+ end
		21	+ end
7	end	22	end
	@@ -12,4 +12,21 @@ describe User do		@@ -12,4 +12,21 @@ describe User do
12	describe 'associations' do	12	describe 'associations' do
13	it { should have_many(:project_ownerships) }	13	it { should have_many(:project_ownerships) }
14	end	14	end
		15	+
		16	+ describe 'methods' do
		17	+ describe 'projects' do
		18	+ subject { FactoryGirl.build(:user) }
		19	+ let(:project) {FactoryGirl.build(:project)}
		20	+ let(:project_ownership) {FactoryGirl.build(:project_ownership)}
		21	+
		22	+ before :each do
		23	+ project_ownership.expects(:project).returns(project)
		24	+ subject.expects(:project_ownerships).returns([project_ownership])
		25	+ end
		26	+
		27	+ it 'should return a list of projects owned by the user' do
		28	+ subject.projects.should eq([project])
		29	+ end
		30	+ end
		31	+ end
15	end	32	end