spec/requests/user_invitation_spec.rb in iugusdk-1.0.0.alpha.1 vs spec/requests/user_invitation_spec.rb in iugusdk-1.0.0.alpha.2

@@ -24,19 +24,50 @@
   context "new view" do
     before(:each) do
       visit'/account/auth/facebook'
       @user = User.last
       @account = @user.accounts.first
-      visit new_invite_path(@account.id)
+      @account.users << Fabricate(:user, :email => "[e-mail]")
+      visit new_invite_path(:account_id => @account.id)
     end
 
     APP_ROLES['roles'].each do |role|
       it { page.should have_content role }
     end
 
     it { page.should have_content 'Email' }
 
     it { page.should have_button I18n.t("iugu.invite") }
+
+    context "when current_user is owner" do
+      before(:each) do
+        @account_user = AccountUser.find_by_user_id_and_account_id(@user.id, @account.id)
+        @account_user.set_roles ["owner"]
+        visit new_invite_path(@account.id)
+      end
+
+      APP_ROLES['roles'].each do |role|
+        it { page.should have_content role }
+      end
+
+    end
   
+    context "when current_user is admin" do
+      before(:each) do
+        @account_user = AccountUser.find_by_user_id_and_account_id(@user.id, @account.id)
+        @account_user.set_roles ["admin"]
+        visit new_invite_path(@account.id)
+      end
+
+      APP_ROLES['roles'].each do |role|
+        unless role == APP_ROLES['owner_role'] || role == APP_ROLES['admin_role']
+          it { page.should have_content role }
+        else
+          it { page.should_not have_content role }
+        end
+      end
+
+    end
+
   end
   
 end