add some tests for user/universe/pages

app/models/billing/billing_plan.rb

@@ -1,3 +1,4 @@
 class BillingPlan < ApplicationRecord
+  FREE_IDS    = [1]
   PREMIUM_IDS = [2, 3, 4, 5, 6]
 end

app/models/page_types/universe.rb

@@ -53,6 +53,7 @@ class Universe < ApplicationRecord
   has_many :lores
 
   has_many :contributors, dependent: :destroy
+  has_many :contributing_users, through: :contributors, source: :user
 
   has_many :documents
   has_many :timelines

test/fixtures/user.yml

@@ -0,0 +1,11 @@
+starter:
+  id: 1
+  name: Starter User
+  email: test@test.test
+  selected_billing_plan_id: 1
+
+premium:
+  id: 2
+  name: Premium User
+  email: test@test.test
+  selected_billing_plan_id: 3

test/models/content_page_test.rb

@@ -0,0 +1,26 @@
+require 'test_helper'
+
+class ContentPageTest < ActiveSupport::TestCase
+  test "Content pages require a user to create" do
+    Rails.application.config.content_types[:all].each do |page_type|
+      assert_raises(ActiveRecord::RecordInvalid) do
+        page_type.create!
+      end
+    end
+  end
+
+  test "Content page privacy levels" do
+    user = User.first
+    Rails.application.config.content_types[:all].each do |page_type|
+      page = page_type.new(user: user, privacy: 'private')
+
+      assert user.can_read?(page)
+      assert_not User.new.can_read?(page)
+      # TODO collaborators
+
+      assert user.can_update?(page)
+      assert_not User.new.can_update?(page)
+      # TODO collaborators
+    end
+  end
+end

test/models/universe_test.rb

@@ -0,0 +1,46 @@
+require 'test_helper'
+
+class UniverseTest < ActiveSupport::TestCase
+  test "Universe contributors can view/modify pages when private" do
+    owner       = User.first
+    contributor = User.last
+    assert_not_equal(owner.id, contributor.id, "Test problem: Owner and contributor need different ideas!")
+
+    universe = Universe.create!(name: "Test Universe", user: owner)
+    universe.contributing_users << contributor
+    assert universe.contributing_users.include?(contributor), "Contributor is a contributor"
+
+    Rails.application.config.content_types[:all_non_universe].each do |page_type|
+      page = page_type.new(user: owner, universe_id: universe.id, privacy: 'private')
+
+      assert contributor.can_read?(page), "Contributor can view shared #{page_type.name}"
+      assert contributor.can_update?(page), "Contributor can edit shared #{page_type.name}"
+
+      assert owner.can_delete?(page), "Owner can delete their own shared #{page_type.name}"
+      assert_not contributor.can_delete?(page), "Contributors cannot delete shared #{page_type.name}"
+    end
+  end
+
+  test "Universe Premium is shared" do
+    owner       = User.first
+    owner.selected_billing_plan_id = BillingPlan::PREMIUM_IDS.first
+    contributor = User.last
+    contributor.selected_billing_plan_id = BillingPlan::FREE_IDS.first
+    assert_not_equal(owner.id, contributor.id, "Test problem: Owner and contributor need different ideas!")
+
+    Rails.application.config.content_types[:premium].each do |page_type|
+      assert     owner.can_create?(page_type), "Owner can create #{page_type.name}"
+      assert_not contributor.can_create?(page_type), "Contributor cannot create #{page_type.name}"
+    end
+
+    universe = Universe.create!(name: "Test Universe", user: owner)
+    universe.contributing_users << contributor
+
+    Rails.application.config.content_types[:all_non_universe].each do |page_type|
+      page = page_type.new(universe_id: universe.id, user: contributor)
+
+      assert page.creatable_by?(owner), "Owner can create #{page_type}"
+      # assert page.creatable_by?(contributor), "Contributor can create #{page_type}"
+    end
+  end
+end

test/models/user_test.rb

@@ -0,0 +1,45 @@
+require 'test_helper'
+
+class UserTest < ActiveSupport::TestCase
+  test "User can create billing plan -specific pages" do
+    free_user = User.new
+    free_user.selected_billing_plan_id = 1
+
+    premium_user = User.new
+    premium_user.selected_billing_plan_id = BillingPlan::PREMIUM_IDS.first
+
+    Rails.application.config.content_types[:free].each do |page_type|
+      assert free_user.can_create?(page_type), "free user CAN create #{page_type.name}"
+      assert premium_user.can_create?(page_type), "premium user CAN create #{page_type.name}"
+    end
+
+    Rails.application.config.content_types[:premium].each do |page_type|
+      assert_not free_user.can_create?(page_type), "free user CANNOT create #{page_type.name}"
+      assert     premium_user.can_create?(page_type), "premium user CAN create #{page_type.name}"
+    end
+  end
+
+  test "User can view their own content regardless of plan" do
+    user = User.new
+    Rails.application.config.content_types[:all].each do |page_type|
+      page = page_type.new(user: user)
+      assert user.can_read?(page)
+    end
+  end
+
+  test "User can edit their own content regardless of plan" do
+    user = User.new
+    Rails.application.config.content_types[:all].each do |page_type|
+      page = page_type.new(user: user)
+      assert user.can_update?(page)
+    end
+  end
+
+  test "User can delete their own content" do
+    user = User.new
+    Rails.application.config.content_types[:all].each do |page_type|
+      page = page_type.new(user: user)
+      assert user.can_delete?(page)
+    end
+  end
+end