tests: test is_active user column migration (#42428)
This commit is contained in:
parent
b145351e2d
commit
3a2b8d45e4
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue