1 files changed, 90 insertions, 0 deletions
diff --git a/spec/models/project_member_spec.rb b/spec/models/project_member_spec.rb
new file mode 100644
index 00000000000..76c5437a555
--- /dev/null
+++ b/spec/models/project_member_spec.rb
@@ -0,0 +1,90 @@
+# == Schema Information
+#
+# Table name: users_projects
+#
+#  id                 :integer          not null, primary key
+#  user_id            :integer          not null
+#  project_id         :integer          not null
+#  created_at         :datetime
+#  updated_at         :datetime
+#  project_access     :integer          default(0), not null
+#  notification_level :integer          default(3), not null
+#
+
+require 'spec_helper'
+
+describe ProjectMember do
+  describe :import_team do
+    before do
+      @abilities = Six.new
+      @abilities << Ability
+
+      @project_1 = create :project
+      @project_2 = create :project
+
+      @user_1 = create :user
+      @user_2 = create :user
+
+      @project_1.team << [ @user_1, :developer ]
+      @project_2.team << [ @user_2, :reporter ]
+
+      @status = @project_2.team.import(@project_1)
+    end
+
+    it { @status.should be_true }
+
+    describe 'project 2 should get user 1 as developer. user_2 should not be changed' do
+      it { @project_2.users.should include(@user_1) }
+      it { @project_2.users.should include(@user_2) }
+
+      it { @abilities.allowed?(@user_1, :write_project, @project_2).should be_true }
+      it { @abilities.allowed?(@user_2, :read_project, @project_2).should be_true }
+    end
+
+    describe 'project 1 should not be changed' do
+      it { @project_1.users.should include(@user_1) }
+      it { @project_1.users.should_not include(@user_2) }
+    end
+  end
+
+  describe :add_users_into_projects do
+    before do
+      @project_1 = create :project
+      @project_2 = create :project
+
+      @user_1 = create :user
+      @user_2 = create :user
+
+      ProjectMember.add_users_into_projects(
+        [@project_1.id, @project_2.id],
+        [@user_1.id, @user_2.id],
+        ProjectMember::MASTER
+      )
+    end
+
+    it { @project_1.users.should include(@user_1) }
+    it { @project_1.users.should include(@user_2) }
+
+
+    it { @project_2.users.should include(@user_1) }
+    it { @project_2.users.should include(@user_2) }
+  end
+
+  describe :truncate_teams do
+    before do
+      @project_1 = create :project
+      @project_2 = create :project
+
+      @user_1 = create :user
+      @user_2 = create :user
+
+      @project_1.team << [ @user_1, :developer]
+      @project_2.team << [ @user_2, :reporter]
+
+      ProjectMember.truncate_teams([@project_1.id, @project_2.id])
+    end
+
+    it { @project_1.users.should be_empty }
+    it { @project_2.users.should be_empty }
+  end
+end