tests: test is_active user column migration (#42428)

This commit is contained in:
Frédéric Péters 2020-07-27 14:47:19 +02:00
parent b145351e2d
commit 3a2b8d45e4
1 changed files with 29 additions and 0 deletions

View File

@ -2028,3 +2028,32 @@ def test_migration_38_submission_agent_id():
conn.commit()
cur.close()
@postgresql
def test_migration_40_user_is_active():
conn, cur = sql.get_connection_and_cursor()
cur.execute('UPDATE wcs_meta SET value = 39 WHERE key = %s', ('sql_level',))
conn.commit()
cur.close()
sql.SqlUser.wipe()
user = sql.SqlUser()
user.name = 'Jean Sénisme'
user.deleted_timestamp = datetime.datetime.now()
user.store()
user2 = sql.SqlUser()
user2.name = 'Jean II'
user2.store()
assert sql.SqlUser.count() == 2
conn, cur = sql.get_connection_and_cursor()
cur.execute('ALTER TABLE users DROP COLUMN is_active')
assert not column_exists_in_table(cur, 'users', 'is_active')
sql.migrate()
assert column_exists_in_table(cur, 'users', 'is_active')
assert migration_level(cur) >= 40
assert sql.SqlUser.count() == 2
assert sql.SqlUser.get(id=user.id).is_active is False
assert sql.SqlUser.get(id=user2.id).is_active is True