From 34d83aaa7cc50a9437bd00a29ad3370e2d7a2345 Mon Sep 17 00:00:00 2001
From: "Stanko K.R." <stanko@stanko.io>
Date: Thu, 13 Nov 2025 16:58:17 +0100
Subject: [PATCH] Fix tests after the removal of memberships

---
 app/channels/application_cable/connection.rb  |  5 +-
 .../confirmations_controller.rb               | 12 ++--
 .../users/email_addresses_controller.rb       | 12 ++--
 app/models/board.rb                           |  2 +-
 app/models/current.rb                         | 12 +++-
 app/models/user.rb                            |  2 +-
 app/views/users/edit.html.erb                 |  2 +-
 config/routes.rb                              | 10 ++-
 .../signups/completions_controller_test.rb    |  7 +-
 .../signups/memberships_controller_test.rb    | 67 -------------------
 .../signup/account_name_generator_test.rb     | 25 ++++---
 gems/fizzy-saas/test/models/signup_test.rb    | 38 +----------
 test/controllers/boards_controller_test.rb    |  2 +-
 .../controllers/join_codes_controller_test.rb |  4 +-
 .../confirmations_controller_test.rb          | 42 ------------
 .../email_addresses_controller_test.rb        | 43 ------------
 .../sessions/menus_controller_test.rb         | 41 +++---------
 test/helpers/application_helper_test.rb       |  6 +-
 test/models/account_test.rb                   |  5 +-
 test/models/board/accessible_test.rb          |  4 +-
 test/models/filter_test.rb                    |  2 +-
 test/models/identity_test.rb                  | 10 ++-
 .../email_address_changeable_test.rb          | 59 ----------------
 test/models/membership_test.rb                | 18 -----
 test/models/user/accessor_test.rb             |  4 +-
 test/models/user/configurable_test.rb         |  2 +-
 test/models/user_test.rb                      |  8 ++-
 27 files changed, 84 insertions(+), 360 deletions(-)
 delete mode 100644 gems/fizzy-saas/test/controllers/signups/memberships_controller_test.rb
 delete mode 100644 test/controllers/memberships/email_addresses/confirmations_controller_test.rb
 delete mode 100644 test/controllers/memberships/email_addresses_controller_test.rb
 delete mode 100644 test/models/membership/email_address_changeable_test.rb
 delete mode 100644 test/models/membership_test.rb

diff --git a/app/channels/application_cable/connection.rb b/app/channels/application_cable/connection.rb
index cb6370d8e..85809ad2f 100644
--- a/app/channels/application_cable/connection.rb
+++ b/app/channels/application_cable/connection.rb
@@ -9,9 +9,8 @@ module ApplicationCable
     private
       def set_current_user
         if session = find_session_by_cookie
-          account = Account.find(request.env["fizzy.external_account_id"])
-          user = session.identity.user.find_by!(account: account)
-          self.current_user = user if user.active?
+          account = Account.find_by(external_account_id: request.env["fizzy.external_account_id"])
+          self.current_user = session.identity.users.find_by!(account: account) if account
         end
       end
 
diff --git a/app/controllers/users/email_addresses/confirmations_controller.rb b/app/controllers/users/email_addresses/confirmations_controller.rb
index 5e9deb703..5e5d8f0ed 100644
--- a/app/controllers/users/email_addresses/confirmations_controller.rb
+++ b/app/controllers/users/email_addresses/confirmations_controller.rb
@@ -2,24 +2,24 @@ class Users::EmailAddresses::ConfirmationsController < ApplicationController
   disallow_account_scope
   allow_unauthenticated_access
 
-  before_action :set_membership
+  before_action :set_user
   rate_limit to: 5, within: 1.hour, only: :create
 
   def show
   end
 
   def create
-    membership = Membership.change_email_address_using_token(token)
+    user = User.change_email_address_using_token(token)
 
     terminate_session if Current.session
-    start_new_session_for membership.reload.identity
+    start_new_session_for user.reload.identity
 
-    redirect_to edit_user_url(script_name: "/#{@membership.tenant}", id: @membership.user)
+    redirect_to edit_user_url(script_name: user.account.slug, id: user)
   end
 
   private
-    def set_membership
-      @membership = Membership.find(params[:membership_id])
+    def set_user
+      @user = User.find(params[:user_id])
     end
 
     def token
diff --git a/app/controllers/users/email_addresses_controller.rb b/app/controllers/users/email_addresses_controller.rb
index f90c838b5..691c50e42 100644
--- a/app/controllers/users/email_addresses_controller.rb
+++ b/app/controllers/users/email_addresses_controller.rb
@@ -3,7 +3,7 @@ class Users::EmailAddressesController < ApplicationController
 
   layout "public"
 
-  before_action :set_membership
+  before_action :set_user
   rate_limit to: 5, within: 1.hour, only: :create
 
   def new
@@ -12,17 +12,17 @@ class Users::EmailAddressesController < ApplicationController
   def create
     identity = Identity.find_by_email_address(new_email_address)
 
-    if identity&.memberships&.exists?(tenant: @membership.tenant)
+    if identity&.users&.exists?(account: @user.account)
       flash[:alert] = "You already have a user in this account with that email address"
-      redirect_to new_email_address_path
+      redirect_to new_user_email_address_path(@user)
     else
-      @membership.send_email_address_change_confirmation(new_email_address)
+      @user.send_email_address_change_confirmation(new_email_address)
     end
   end
 
   private
-    def set_membership
-      @membership = Current.identity.memberships.find(params[:membership_id])
+    def set_user
+      @user = Current.identity.users.find(params[:user_id])
     end
 
     def new_email_address
diff --git a/app/models/board.rb b/app/models/board.rb
index b39e47441..9f1bcd5d7 100644
--- a/app/models/board.rb
+++ b/app/models/board.rb
@@ -1,8 +1,8 @@
 class Board < ApplicationRecord
   include Accessible, AutoPostponing, Broadcastable, Cards, Entropic, Filterable, Publishable, Triageable
 
-  belongs_to :account, default: -> { creator.account }
   belongs_to :creator, class_name: "User", default: -> { Current.user }
+  belongs_to :account, default: -> { creator.account }
 
   has_rich_text :public_description
 
diff --git a/app/models/current.rb b/app/models/current.rb
index 26f4ef189..a8ef58461 100644
--- a/app/models/current.rb
+++ b/app/models/current.rb
@@ -7,8 +7,16 @@ class Current < ActiveSupport::CurrentAttributes
   def session=(value)
     super(value)
 
-    if value.present? && Current.account.present?
-      self.user = identity.users.find_by(account: Current.account)
+    if value.present? && account.present?
+      self.user = identity.users.find_by(account: account)
+    end
+  end
+
+  def account=(value)
+    super(value)
+
+    if value.present? && identity.present?
+      self.user = identity.users.find_by(account: value)
     end
   end
 end
diff --git a/app/models/user.rb b/app/models/user.rb
index cef823a1f..6b985c2df 100644
--- a/app/models/user.rb
+++ b/app/models/user.rb
@@ -20,7 +20,7 @@ class User < ApplicationRecord
   def deactivate
     transaction do
       accesses.destroy_all
-      update! active: false
+      update! active: false, identity: nil
     end
   end
 end
diff --git a/app/views/users/edit.html.erb b/app/views/users/edit.html.erb
index dda329f84..4dd365d23 100644
--- a/app/views/users/edit.html.erb
+++ b/app/views/users/edit.html.erb
@@ -40,7 +40,7 @@
       <div class="flex align-center gap">
         <div class="flex align-center gap input input--actor">
           <%= form.email_field :email_address, class: "input full-width", autocomplete: "username", placeholder: "Email address", required: true, readonly: true, value: @user.identity.email_address %>
-          <%= link_to "Change email", new_email_address_url(script_name: nil, user_id: Current.user.id), class: "btn btn--plain txt-link txt-small txt-nowrap" %>
+          <%= link_to "Change email", new_user_email_address_url(script_name: nil, user_id: Current.user.id), class: "btn btn--plain txt-link txt-small txt-nowrap" %>
         </div>
       </div>
       <button type="submit" id="log_in" class="btn btn--reversed center" data-form-target="submit">
diff --git a/config/routes.rb b/config/routes.rb
index 7609cf04a..d4577d4f9 100644
--- a/config/routes.rb
+++ b/config/routes.rb
@@ -14,6 +14,10 @@ Rails.application.routes.draw do
       resource :events
 
       resources :push_subscriptions
+
+      resources :email_addresses, param: :token do
+        resource :confirmation, module: :email_addresses
+      end
     end
   end
 
@@ -147,12 +151,6 @@ Rails.application.routes.draw do
 
   resource :landing
 
-  scope module: :memberships, path: "memberships/:membership_id" do
-    resources :email_addresses, param: :token do
-      resource :confirmation, module: :email_addresses
-    end
-  end
-
   namespace :my do
     resources :pins
     resource :timezone
diff --git a/gems/fizzy-saas/test/controllers/signups/completions_controller_test.rb b/gems/fizzy-saas/test/controllers/signups/completions_controller_test.rb
index 231aab94c..89c67ca64 100644
--- a/gems/fizzy-saas/test/controllers/signups/completions_controller_test.rb
+++ b/gems/fizzy-saas/test/controllers/signups/completions_controller_test.rb
@@ -7,16 +7,11 @@ class Signup::CompletionsControllerTest < ActionDispatch::IntegrationTest
     @signup.create_identity || raise("Failed to create identity")
 
     sign_in_as @signup.identity
-
-    @signup.create_membership || raise("Failed to create membership")
   end
 
   test "new" do
     untenanted do
-      get saas.new_signup_completion_path(signup: {
-        membership_id: @signup.membership_id,
-        full_name: @signup.full_name,
-        account_name: @signup.account_name })
+      get saas.new_signup_completion_path
     end
 
     assert_response :success
diff --git a/gems/fizzy-saas/test/controllers/signups/memberships_controller_test.rb b/gems/fizzy-saas/test/controllers/signups/memberships_controller_test.rb
deleted file mode 100644
index 57010d2bd..000000000
--- a/gems/fizzy-saas/test/controllers/signups/memberships_controller_test.rb
+++ /dev/null
@@ -1,67 +0,0 @@
-require "test_helper"
-
-class Signup::MembershipsControllerTest < ActionDispatch::IntegrationTest
-  setup do
-    @identity = Identity.create!(email_address: "newuser@example.com")
-    magic_link = @identity.send_magic_link
-
-    untenanted do
-      post session_magic_link_url, params: { code: magic_link.code }
-      assert_response :redirect, "Magic link should succeed"
-
-      cookie = cookies.get_cookie "session_token"
-      assert_not_nil cookie, "Expected session_token cookie to be set after magic link consumption"
-    end
-  end
-
-  test "new" do
-    untenanted do
-      get saas.new_signup_membership_path
-
-      assert_response :success
-    end
-  end
-
-  test "new with new_user param" do
-    untenanted do
-      get saas.new_signup_membership_path(signup: { new_user: true })
-
-      assert_response :success
-    end
-  end
-
-  test "create" do
-    untenanted do
-      assert_difference -> { Membership.count }, 1 do
-        post saas.signup_membership_path, params: {
-          signup: {
-            full_name: "New User"
-          }
-        }
-      end
-
-      membership = Membership.last
-      assert_redirected_to saas.new_signup_completion_path(
-        signup: {
-          membership_id: membership.signed_id(purpose: :account_creation),
-          full_name: "New User",
-          account_name: "New's Fizzy"
-        }
-      )
-    end
-  end
-
-  test "create with invalid params" do
-    untenanted do
-      assert_no_difference -> { Membership.count } do
-        post saas.signup_membership_path, params: {
-          signup: {
-            full_name: ""
-          }
-        }
-      end
-
-      assert_response :unprocessable_entity, "Invalid params should return unprocessable entity"
-    end
-  end
-end
diff --git a/gems/fizzy-saas/test/models/signup/account_name_generator_test.rb b/gems/fizzy-saas/test/models/signup/account_name_generator_test.rb
index fdaac96ce..877b72608 100644
--- a/gems/fizzy-saas/test/models/signup/account_name_generator_test.rb
+++ b/gems/fizzy-saas/test/models/signup/account_name_generator_test.rb
@@ -11,34 +11,39 @@ class Signup::AccountNameGeneratorTest < ActiveSupport::TestCase
     account_name = @generator.generate
     assert_equal "Newart's Fizzy", account_name, "The 1st account doesn't have 1st in the name"
 
-    first_membership = @identity.memberships.create(tenant: "1st")
-    first_membership.stubs(:account_name).returns(account_name)
+    first_account = Account.create!(external_account_id: "1st", name: account_name)
+    @identity.users.create!(account: first_account, name: @name)
+    @identity.reload
 
     account_name = @generator.generate
     assert_equal "Newart's 2nd Fizzy", account_name
 
-    second_membership = @identity.memberships.create(tenant: "2nd")
-    second_membership.stubs(:account_name).returns(account_name)
+    second_account = Account.create!(external_account_id: "2nd", name: account_name)
+    @identity.users.create!(account: second_account, name: @name)
+    @identity.reload
 
     account_name = @generator.generate
     assert_equal "Newart's 3rd Fizzy", account_name
 
-    third_membership = @identity.memberships.create(tenant: "3nd")
-    third_membership.stubs(:account_name).returns(account_name)
+    third_account = Account.create!(external_account_id: "3rd", name: account_name)
+    @identity.users.create!(account: third_account, name: @name)
+    @identity.reload
 
     account_name = @generator.generate
     assert_equal "Newart's 4th Fizzy", account_name
 
-    fourth_membership = @identity.memberships.create(tenant: "4th")
-    fourth_membership.stubs(:account_name).returns(account_name)
+    fourth_account = Account.create!(external_account_id: "4th", name: account_name)
+    @identity.users.create!(account: fourth_account, name: @name)
+    @identity.reload
 
     account_name = @generator.generate
     assert_equal "Newart's 5th Fizzy", account_name
   end
 
   test "generate continues from the previous highest index" do
-    membership = @identity.memberships.create(tenant: "12th")
-    membership.stubs(:account_name).returns("Newart's 12th Fizzy")
+    account = Account.create!(external_account_id: "12th", name: "Newart's 12th Fizzy")
+    @identity.users.create!(account: account, name: @name)
+    @identity.reload
 
     account_name = @generator.generate
     assert_equal "Newart's 13th Fizzy", account_name
diff --git a/gems/fizzy-saas/test/models/signup_test.rb b/gems/fizzy-saas/test/models/signup_test.rb
index d64b647c7..a35e50cb9 100644
--- a/gems/fizzy-saas/test/models/signup_test.rb
+++ b/gems/fizzy-saas/test/models/signup_test.rb
@@ -28,59 +28,25 @@ class SignupTest < ActiveSupport::TestCase
     end
   end
 
-  test "#create_membership" do
-    signup = Signup.new(
-      full_name: "Kevin",
-      identity: identities(:kevin)
-    )
-
-    assert_difference -> { Membership.count }, 1 do
-      assert signup.create_membership
-    end
-
-    assert signup.tenant
-    assert signup.membership
-    assert signup.membership_id
-
-    signup_invalid = Signup.new(
-      full_name: "",
-      identity: identities(:kevin)
-    )
-    assert_not signup_invalid.create_membership, "Create membership should fail with invalid params"
-    assert_not_empty signup_invalid.errors[:full_name], "Should have validation error for full_name"
-  end
-
   test "#complete" do
     Account.any_instance.expects(:setup_customer_template).once
 
-    signup_for_membership = Signup.new(
-      full_name: "Kevin",
-      identity: identities(:kevin)
-    )
-    signup_for_membership.create_membership
-
     signup = Signup.new(
       full_name: "Kevin",
-      account_name: "37signals",
-      membership_id: signup_for_membership.membership_id,
       identity: identities(:kevin)
     )
 
     assert signup.complete
 
-    assert signup.tenant
     assert signup.account
     assert signup.user
     assert_equal "Kevin", signup.user.name
-    assert_equal "37signals", signup.account.name
 
     signup_invalid = Signup.new(
       full_name: "",
-      account_name: "37signals",
-      membership_id: signup_for_membership.membership_id,
       identity: identities(:kevin)
     )
-    assert_not signup_invalid.complete, "Complete should fail with invalid params"
-    assert_not_empty signup_invalid.errors[:full_name], "Should have validation error for full_name"
+    assert_not signup_invalid.complete
+    assert_not_empty signup_invalid.errors[:full_name]
   end
 end
diff --git a/test/controllers/boards_controller_test.rb b/test/controllers/boards_controller_test.rb
index 7f892e935..2a94e524e 100644
--- a/test/controllers/boards_controller_test.rb
+++ b/test/controllers/boards_controller_test.rb
@@ -77,7 +77,7 @@ class BoardsControllerTest < ActionDispatch::IntegrationTest
   end
 
   test "update all access" do
-    board = Current.set(session: sessions(:kevin)) do
+    board = Current.set(account: accounts("37s"), session: sessions(:kevin), user: users(:kevin)) do
       Board.create! name: "New board", all_access: false
     end
     assert_equal [ users(:kevin) ], board.users
diff --git a/test/controllers/join_codes_controller_test.rb b/test/controllers/join_codes_controller_test.rb
index 097d30ca1..9752ed7bd 100644
--- a/test/controllers/join_codes_controller_test.rb
+++ b/test/controllers/join_codes_controller_test.rb
@@ -36,7 +36,7 @@ class JoinCodesControllerTest < ActionDispatch::IntegrationTest
   test "create" do
     untenanted do
       assert_difference -> { Identity.count }, 1 do
-        assert_difference -> { Membership.count }, 1 do
+        assert_difference -> { User.count }, 1 do
           post join_path(tenant: @tenant, code: @join_code.code), params: { email_address: "new_user@example.com" }
         end
       end
@@ -52,7 +52,7 @@ class JoinCodesControllerTest < ActionDispatch::IntegrationTest
 
     untenanted do
       assert_no_difference -> { Identity.count } do
-        assert_no_difference -> { Membership.count } do
+        assert_no_difference -> { User.count } do
           post join_path(tenant: @tenant, code: @join_code.code), params: { email_address: identity.email_address }
         end
       end
diff --git a/test/controllers/memberships/email_addresses/confirmations_controller_test.rb b/test/controllers/memberships/email_addresses/confirmations_controller_test.rb
deleted file mode 100644
index 86c5ff171..000000000
--- a/test/controllers/memberships/email_addresses/confirmations_controller_test.rb
+++ /dev/null
@@ -1,42 +0,0 @@
-require "test_helper"
-
-class Memberships::EmailAddresses::ConfirmationsControllerTest < ActionDispatch::IntegrationTest
-  test "show" do
-    untenanted do
-      membership = memberships(:kevin_in_37signals)
-
-      get email_address_confirmation_path(
-        membership_id: membership.id,
-        email_address_token: "dummy_token"
-      )
-
-      assert_response :success
-    end
-  end
-
-  test "create" do
-    untenanted do
-      membership = memberships(:kevin_in_37signals)
-      old_identity = membership.identity
-      new_email = "updated@example.com"
-
-      # Generate a real token
-      token = membership.send(:generate_email_address_change_token, to: new_email)
-
-      assert_difference -> { Identity.count }, 1 do
-        post email_address_confirmation_path(
-          membership_id: membership.id,
-          email_address_token: token
-        ),
-        params: { email_address_token: token }
-      end
-
-      membership.reload
-      assert_equal new_email, membership.identity.email_address
-      assert_not_equal old_identity.id, membership.identity_id
-
-      assert cookies[:session_token].present?, "Should have started new session"
-      assert_redirected_to edit_user_url(script_name: "/#{membership.tenant}", id: membership.user)
-    end
-  end
-end
diff --git a/test/controllers/memberships/email_addresses_controller_test.rb b/test/controllers/memberships/email_addresses_controller_test.rb
deleted file mode 100644
index ef88bfaa1..000000000
--- a/test/controllers/memberships/email_addresses_controller_test.rb
+++ /dev/null
@@ -1,43 +0,0 @@
-require "test_helper"
-
-class Memberships::EmailAddressesControllerTest < ActionDispatch::IntegrationTest
-  test "new" do
-    untenanted do
-      sign_in_as :kevin
-
-      membership = memberships(:kevin_in_37signals)
-
-      get new_email_address_path(membership_id: membership.id)
-      assert_response :success
-    end
-  end
-
-  test "create" do
-    untenanted do
-      sign_in_as :kevin
-
-      membership = memberships(:kevin_in_37signals)
-
-      assert_enqueued_emails 1 do
-        post email_addresses_path(membership_id: membership.id),
-             params: { email_address: "newemail@example.com" }
-      end
-
-      assert_response :success
-    end
-  end
-
-  test "create with an email for someone already in the account" do
-    untenanted do
-      sign_in_as :kevin
-
-      membership = memberships(:kevin_in_37signals)
-
-      post email_addresses_path(membership_id: membership.id),
-           params: { email_address: identities(:david).email_address }
-
-      assert_redirected_to new_email_address_path
-      assert_equal "You already have a user in this account with that email address", flash[:alert]
-    end
-  end
-end
diff --git a/test/controllers/sessions/menus_controller_test.rb b/test/controllers/sessions/menus_controller_test.rb
index c6b64d316..689c6e7f5 100644
--- a/test/controllers/sessions/menus_controller_test.rb
+++ b/test/controllers/sessions/menus_controller_test.rb
@@ -7,7 +7,7 @@ class Sessions::MenusControllerTest < ActionDispatch::IntegrationTest
 
   test "show with no memberships" do
     sign_in_as @identity
-    @identity.memberships.delete_all
+    @identity.users.delete_all
 
     untenanted do
       get session_menu_url
@@ -18,22 +18,25 @@ class Sessions::MenusControllerTest < ActionDispatch::IntegrationTest
 
   test "show with exactly one membership" do
     sign_in_as @identity
-    @identity.memberships.delete_all
-    @identity.memberships.create(tenant: "37signals")
+    @identity.users.delete_all
+    account = Account.create!(external_account_id: 9999991, name: "Test Account")
+    @identity.users.create!(account: account, name: "Kevin")
 
     untenanted do
       get session_menu_url
     end
 
     assert_response :redirect
-    assert_redirected_to root_url(script_name: "/37signals")
+    assert_redirected_to root_url(script_name: "/9999991")
   end
 
   test "show with multiple memeberships" do
     sign_in_as @identity
-    @identity.memberships.delete_all
-    @identity.memberships.create(tenant: "37signals")
-    @identity.memberships.create(tenant: "acme")
+    @identity.users.delete_all
+    account1 = Account.create!(external_account_id: 9999992, name: "37signals")
+    account2 = Account.create!(external_account_id: 9999993, name: "Acme")
+    @identity.users.create!(account: account1, name: "Kevin")
+    @identity.users.create!(account: account2, name: "Kevin")
 
     untenanted do
       get session_menu_url
@@ -42,28 +45,4 @@ class Sessions::MenusControllerTest < ActionDispatch::IntegrationTest
     assert_response :success
   end
 
-  test "show renders as a menu section" do
-    sign_in_as @identity
-    @identity.memberships.delete_all
-    @identity.memberships.create(tenant: "37signals")
-    @identity.memberships.create(tenant: "acme")
-
-    untenanted do
-      get session_menu_url menu_section: true
-    end
-
-    assert_response :success
-  end
-
-  test "show doesn't redirect when rendered as a menu section" do
-    sign_in_as @identity
-    @identity.memberships.delete_all
-    @identity.memberships.create(tenant: "37signals")
-
-    untenanted do
-      get session_menu_url menu_section: true
-    end
-
-    assert_response :success
-  end
 end
diff --git a/test/helpers/application_helper_test.rb b/test/helpers/application_helper_test.rb
index 1a050f17d..8fbfa2a0c 100644
--- a/test/helpers/application_helper_test.rb
+++ b/test/helpers/application_helper_test.rb
@@ -26,7 +26,8 @@ class ApplicationHelperTest < ActionView::TestCase
 
   test "page_title_tag on tenanted page when user has multiple memberships" do
     Current.session = sessions(:david)
-    identities(:david).memberships.create!(tenant: "dangling-tenant")
+    other_account = Account.create!(external_account_id: "dangling-tenant", name: "Other Account")
+    identities(:david).users.create!(account: other_account, name: "David")
 
     assert_select parse(page_title_tag), "title", text: "37signals | Fizzy"
   end
@@ -40,7 +41,8 @@ class ApplicationHelperTest < ActionView::TestCase
 
   test "page_title_tag on tenanted page with a page title when user has multiple memberships" do
     Current.session = sessions(:david)
-    identities(:david).memberships.create!(tenant: "dangling-tenant")
+    other_account = Account.create!(external_account_id: "dangling-tenant", name: "Other Account")
+    identities(:david).users.create!(account: other_account, name: "David")
     @page_title = "Holodeck"
 
     assert_select parse(page_title_tag), "title", text: "Holodeck | 37signals | Fizzy"
diff --git a/test/models/account_test.rb b/test/models/account_test.rb
index e2b553331..f9fefda09 100644
--- a/test/models/account_test.rb
+++ b/test/models/account_test.rb
@@ -14,11 +14,10 @@ class AccountTest < ActiveSupport::TestCase
 
   test ".create_with_admin_user creates a new local account" do
     identity = identities(:david)
-    membership = identity.memberships.create!(tenant: ActiveRecord::FixtureSet.identify("account-create-with-admin-user-test"))
     account = nil
 
     assert_changes -> { Account.count }, +1 do
-      assert_changes -> { User.count }, +1 do
+      assert_changes -> { User.count }, +2 do
         account = Account.create_with_admin_user(
           account: {
             external_account_id: ActiveRecord::FixtureSet.identify("account-create-with-admin-user-test"),
@@ -26,7 +25,7 @@ class AccountTest < ActiveSupport::TestCase
           },
           owner: {
             name: "David",
-            membership: membership
+            identity: identity
           }
         )
       end
diff --git a/test/models/board/accessible_test.rb b/test/models/board/accessible_test.rb
index 7677d51a5..79def5994 100644
--- a/test/models/board/accessible_test.rb
+++ b/test/models/board/accessible_test.rb
@@ -15,14 +15,14 @@ class Board::AccessibleTest < ActiveSupport::TestCase
   end
 
   test "grants access to everyone after creation" do
-    board = Current.set(session: sessions(:david)) do
+    board = Current.set(session: sessions(:david), user: users(:david)) do
       Board.create! name: "New board", all_access: true
     end
     assert_equal accounts("37s").users.active.sort, board.users.sort
   end
 
   test "grants access to everyone after update" do
-    board = Current.set(session: sessions(:david)) do
+    board = Current.set(session: sessions(:david), user: users(:david)) do
       Board.create! name: "New board"
     end
     assert_equal [ users(:david) ], board.users
diff --git a/test/models/filter_test.rb b/test/models/filter_test.rb
index 091580ec3..8531a7770 100644
--- a/test/models/filter_test.rb
+++ b/test/models/filter_test.rb
@@ -7,7 +7,7 @@ class FilterTest < ActiveSupport::TestCase
 
   test "cards" do
     Current.set session: sessions(:david) do
-      @new_board = Board.create! name: "Inaccessible Board"
+      @new_board = Board.create! name: "Inaccessible Board", creator: users(:david)
       @new_card = @new_board.cards.create!(status: "published")
 
       cards(:layout).comments.create!(body: "I hate haggis")
diff --git a/test/models/identity_test.rb b/test/models/identity_test.rb
index 972a68e05..b1b0a2e6d 100644
--- a/test/models/identity_test.rb
+++ b/test/models/identity_test.rb
@@ -23,15 +23,13 @@ class IdentityTest < ActiveSupport::TestCase
     identity = identities(:david)
     account = accounts(:initech)
 
-    assert_difference [ "Membership.count", "User.count" ], 1 do
+    assert_difference "User.count", 1 do
       identity.join(account)
     end
 
-    membership = identity.memberships.order(created_at: :desc).first
-    assert_equal account.external_account_id.to_s, membership.tenant
-
-    user = account.users.order(created_at: :desc).first
-    assert_equal membership, user.membership
+    user = account.users.find_by(identity: identity)
+    assert_not_nil user
+    assert_equal identity, user.identity
     assert_equal identity.email_address, user.name
   end
 end
diff --git a/test/models/membership/email_address_changeable_test.rb b/test/models/membership/email_address_changeable_test.rb
deleted file mode 100644
index ac3b1e77a..000000000
--- a/test/models/membership/email_address_changeable_test.rb
+++ /dev/null
@@ -1,59 +0,0 @@
-require "test_helper"
-
-class Membership::EmailAddressChangeableTest < ActiveSupport::TestCase
-  include ActionMailer::TestHelper
-
-  setup do
-    @identity = identities(:kevin)
-    @tenant = accounts("37s").external_account_id
-    @membership = @identity.memberships.find_by!(tenant: @tenant)
-    @new_email = "newart@example.com"
-  end
-
-  test "send_email_address_change_confirmation" do
-    assert_emails 1 do
-      @membership.send_email_address_change_confirmation(@new_email)
-    end
-  end
-
-  test "change_email_address" do
-    old_identity = @identity
-
-    assert_difference -> { Identity.count }, +1 do
-      @membership.change_email_address(@new_email)
-    end
-
-    assert_equal @new_email, @membership.reload.identity.email_address
-    assert_not old_identity.reload.memberships.exists?(id: @membership.id)
-    assert_equal @new_email, @membership.reload.identity.email_address
-
-    # Make sure that a prior membership doesn't exist
-    identities(:david).memberships.where(tenant: @tenant).delete_all
-
-    assert_no_difference -> { Identity.count } do
-      @membership.change_email_address(identities(:david).email_address)
-    end
-    assert_equal identities(:david).email_address, @membership.reload.identity.email_address
-  end
-
-  test "change_email_address_using_token" do
-    token = @membership.send(:generate_email_address_change_token, to: @new_email)
-
-    Membership.change_email_address_using_token(token)
-
-    assert_equal @new_email, @membership.reload.identity.email_address
-  end
-
-  test "change_email_address_using_token with invalid token" do
-    assert_raises(ArgumentError, match: /invalid/) do
-      Membership.change_email_address_using_token("invalid_token")
-    end
-
-    token = @membership.send(:generate_email_address_change_token, to: @new_email)
-    @identity.update!(email_address: "different@example.com")
-
-    assert_raises(ArgumentError, match: /different email address/) do
-      Membership.change_email_address_using_token(token)
-    end
-  end
-end
diff --git a/test/models/membership_test.rb b/test/models/membership_test.rb
deleted file mode 100644
index 1ec6dcaee..000000000
--- a/test/models/membership_test.rb
+++ /dev/null
@@ -1,18 +0,0 @@
-require "test_helper"
-
-class MembershipTest < ActiveSupport::TestCase
-  test "change_email_address" do
-    tenant = accounts("37s").external_account_id
-    old_identity = identities(:kevin)
-    new_email = "kevin.new@37signals.com"
-
-    assert_difference -> { Identity.count }, 1 do
-      Membership.change_email_address(from: old_identity.email_address, to: new_email, tenant: tenant)
-    end
-
-    new_identity = Identity.find_by(email_address: new_email)
-    assert new_identity
-    assert new_identity.memberships.exists?(tenant: tenant)
-    assert_not old_identity.reload.memberships.exists?(tenant: tenant)
-  end
-end
diff --git a/test/models/user/accessor_test.rb b/test/models/user/accessor_test.rb
index 6c1d6c927..95cb84edb 100644
--- a/test/models/user/accessor_test.rb
+++ b/test/models/user/accessor_test.rb
@@ -2,14 +2,14 @@ require "test_helper"
 
 class User::AccessorTest < ActiveSupport::TestCase
   test "new users get added to all_access boards on creation" do
-    user = User.create!(name: "Jorge")
+    user = User.create!(account: accounts("37s"), name: "Jorge")
 
     assert_includes user.boards, boards(:writebook)
     assert_equal user.account.boards.all_access.count, user.boards.count
   end
 
   test "system user does not get added to boards on creation" do
-    system_user = User.create!(role: "system", name: "Test System User")
+    system_user = User.create!(account: accounts("37s"), role: "system", name: "Test System User")
     assert_empty system_user.boards
   end
 end
diff --git a/test/models/user/configurable_test.rb b/test/models/user/configurable_test.rb
index b49b78c1b..d5f467f7c 100644
--- a/test/models/user/configurable_test.rb
+++ b/test/models/user/configurable_test.rb
@@ -2,7 +2,7 @@ require "test_helper"
 
 class User::ConfigurableTest < ActiveSupport::TestCase
   test "should create settings for new users" do
-    user = User.create! name: "Some new user"
+    user = User.create! account: accounts("37s"), name: "Some new user"
     assert user.settings.present?
   end
 end
diff --git a/test/models/user_test.rb b/test/models/user_test.rb
index 5a6971558..8098e2190 100644
--- a/test/models/user_test.rb
+++ b/test/models/user_test.rb
@@ -2,18 +2,22 @@ require "test_helper"
 
 class UserTest < ActiveSupport::TestCase
   test "create" do
-    user = User.create! \
+    user = User.create!(
+      account: accounts("37s"),
       role: "member",
       name: "Victor Cooper"
+    )
 
     assert_equal [ boards(:writebook) ], user.boards
     assert user.settings.present?
   end
 
   test "creation gives access to all_access boards" do
-    user = User.create! \
+    user = User.create!(
+      account: accounts("37s"),
       role: "member",
       name: "Victor Cooper"
+    )
 
     assert_equal [ boards(:writebook) ], user.boards
   end