Merge pull request #840 from aviav/admin-factory-fix

Generate admins with account membership in specs
codevise · Aug 28, 2017 · 5f21f8e · 5f21f8e
2 parents 1905b0b + 980d471
commit 5f21f8e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/spec/factories/users.rb b/spec/factories/users.rb
@@ -48,6 +48,12 @@ module Pageflow
 
       trait :admin do
         admin true
+        after(:create) do |user, evaluator|
+          create(:membership,
+                 user: user,
+                 role: :member,
+                 entity: evaluator.on || create(:account))
+        end
       end
 
       trait :suspended do

diff --git a/spec/policies/pageflow/folder_policy_spec.rb b/spec/policies/pageflow/folder_policy_spec.rb
@@ -22,7 +22,7 @@ module Pageflow
     it 'does not permit admins when there is no more than one account' do
       Account.delete_all
       folder = create(:folder)
-      admin = create(:user, :admin)
+      admin = create(:user, :admin, on: folder.account)
 
       expect(FolderPolicy.new(admin, folder)).not_to permit_action(:show_account_selection_on)
     end