From ffbc90e4a9cd180230126fa734508d102e55fed5 Mon Sep 17 00:00:00 2001
From: Claire <claire.github-309c@sitedethib.com>
Date: Wed, 1 Mar 2023 15:29:21 +0100
Subject: [PATCH] Add migration tests

---
 lib/tasks/tests.rake | 36 +++++++++++++++++++++++++++++-------
 1 file changed, 29 insertions(+), 7 deletions(-)

diff --git a/lib/tasks/tests.rake b/lib/tasks/tests.rake
index c8e0312bbd8..c102977f94e 100644
--- a/lib/tasks/tests.rake
+++ b/lib/tasks/tests.rake
@@ -112,6 +112,21 @@ namespace :tests do
         exit(1)
       end
 
+      unless Account.find_by(username: 'suspended', domain: nil).suspended?
+        puts 'Unexpected value for Account#suspended? for user @suspended'
+        exit(1)
+      end
+
+      if Account.find_by(username: 'deleted', domain: nil).suspended?
+        puts 'Unexpected value for Account#suspended? for user @deleted'
+        exit(1)
+      end
+
+      unless Account.find_by(username: 'deleted', domain: nil).deleted?
+        puts 'Unexpected value for Account#deleted? for user @deleted'
+        exit(1)
+      end
+
       unless Identity.where(provider: 'foo', uid: 0).count == 1
         puts 'Identities not deduplicated as expected'
         exit(1)
@@ -212,20 +227,20 @@ namespace :tests do
         INSERT INTO "accounts"
           (id, username, domain, private_key, public_key, created_at, updated_at)
         VALUES
-          (10, 'kmruser', NULL, #{user_private_key}, #{user_public_key}, now(), now()),
-          (11, 'qcuser', NULL, #{user_private_key}, #{user_public_key}, now(), now());
+          (12, 'kmruser', NULL, #{user_private_key}, #{user_public_key}, now(), now()),
+          (13, 'qcuser', NULL, #{user_private_key}, #{user_public_key}, now(), now());
 
         INSERT INTO "users"
           (id, account_id, email, created_at, updated_at, admin, locale, chosen_languages)
         VALUES
-          (4, 10, 'kmruser@localhost', now(), now(), false, 'ku', '{en,kmr,ku,ckb}');
+          (5, 12, 'kmruser@localhost', now(), now(), false, 'ku', '{en,kmr,ku,ckb}');
 
         INSERT INTO "users"
           (id, account_id, email, created_at, updated_at, locale,
            encrypted_otp_secret, encrypted_otp_secret_iv, encrypted_otp_secret_salt,
            otp_required_for_login)
         VALUES
-          (5, 11, 'qcuser@localhost', now(), now(), 'fr-QC',
+          (6, 13, 'qcuser@localhost', now(), now(), 'fr-QC',
            E'Fttsy7QAa0edaDfdfSz094rRLAxc8cJweDQ4BsWH/zozcdVA8o9GLqcKhn2b\nGi/V\n',
            'rys3THICkr60BoWC',
            '_LMkAGvdg7a+sDIKjI3mR2Q==',
@@ -234,7 +249,7 @@ namespace :tests do
         INSERT INTO "settings"
           (id, thing_type, thing_id, var, value, created_at, updated_at)
         VALUES
-          (5, 'User', 4, 'default_language', E'--- kmr\n', now(), now()),
+          (5, 'User', 5, 'default_language', E'--- kmr\n', now(), now()),
           (6, 'User', 1, 'interactions', E'--- !ruby/hash:ActiveSupport::HashWithIndifferentAccess\nmust_be_follower: false\nmust_be_following: true\nmust_be_following_dm: false\n', now(), now());
 
         INSERT INTO "identities"
@@ -310,18 +325,25 @@ namespace :tests do
            1, 'https://activitypub.com/users/evil/inbox', 'https://activitypub.com/users/evil/outbox',
            'https://activitypub.com/users/evil/followers', true);
 
+        INSERT INTO "accounts"
+          (id, username, domain, private_key, public_key, created_at, updated_at, suspended)
+        VALUES
+          (10, 'suspended', NULL, #{admin_private_key}, #{admin_public_key}, now(), now(), true),
+          (11, 'deleted',  NULL, #{user_private_key},  #{user_public_key},  now(), now(), true);
+
         -- users
 
         INSERT INTO "users"
           (id, account_id, email, created_at, updated_at, admin)
         VALUES
           (1, 1, 'admin@localhost', now(), now(), true),
-          (2, 2, 'user@localhost', now(), now(), false);
+          (2, 2, 'user@localhost', now(), now(), false),
+          (3, 10, 'suspended@localhost', now(), now(), false);
 
         INSERT INTO "users"
           (id, account_id, email, created_at, updated_at, admin, locale)
         VALUES
-          (3, 8, 'ptuser@localhost', now(), now(), false, 'pt');
+          (4, 8, 'ptuser@localhost', now(), now(), false, 'pt');
 
         -- conversations
         INSERT INTO "conversations" (id, created_at, updated_at) VALUES (1, now(), now());