Add postgres grant/revoke tests

Add  grant, revoke, update password and privilege validation
tests.

Change-Id: Ifc1d263998ecd8f0783ce2c7eb33fb2feda1f281
Signed-off-by: wu.chunyang <wchy1001@gmail.com>
This commit is contained in:
wu.chunyang
2026-02-10 16:56:14 +08:00
parent 13c53e4157
commit 01672d24d2
@@ -75,7 +75,9 @@ class TestInstanceBasicPostgreSQL(base_basic.TestInstanceBasicBase):
@decorators.idempotent_id('6c2c710c-0138-4215-8e08-6dfe605ba6a6')
def test_user_database(self):
db1 = 'foo'
db2 = 'bar'
user1 = 'foo_user'
new_pass = 'new_password'
user2 = 'bar_user'
users = self.get_users(self.instance_id)
@@ -85,12 +87,16 @@ class TestInstanceBasicPostgreSQL(base_basic.TestInstanceBasicBase):
databases = self.get_databases(self.instance_id)
cur_db_names = [db['name'] for db in databases]
self.assertNotIn(db1, cur_db_names)
self.assertNotIn(db2, cur_db_names)
LOG.info(f"Creating databases in instance {self.instance_id}")
create_db = {
"databases": [
{
"name": db1
},
{
"name": db2
}
]
}
@@ -98,19 +104,21 @@ class TestInstanceBasicPostgreSQL(base_basic.TestInstanceBasicBase):
create_db, expected_status_code=202,
need_response=False)
def _wait_db():
def _wait_db_create():
try:
databases = self.get_databases(self.instance_id)
cur_db_names = [db['name'] for db in databases]
self.assertIn(db1, cur_db_names)
self.assertIn(db2, cur_db_names)
raise loopingcall.LoopingCallDone()
except AssertionError:
return
timer = loopingcall.FixedIntervalWithTimeoutLoopingCall(_wait_db)
timer = loopingcall.FixedIntervalWithTimeoutLoopingCall(
_wait_db_create)
try:
timer.start(interval=5, timeout=30, initial_delay=5).wait()
except loopingcall.LoopingCallTimeOut as e:
message = f"failed to create db: {db1} in 30 seconds"
message = f"failed to create dbs: {db1}, {db2} in 30 seconds"
raise exceptions.TimeoutException(message) from e
LOG.info(f"Creating users in instance {self.instance_id}")
@@ -140,16 +148,73 @@ class TestInstanceBasicPostgreSQL(base_basic.TestInstanceBasicBase):
f"db: {db1}")
self._access_db(self.instance_ip, user1, constants.DB_PASS, db1)
LOG.info(f"Updating password for user {user1}")
update_user_body = {"user": {"password": new_pass}}
self.client.put_resource(
f'instances/{self.instance_id}/users/{user1}', update_user_body)
time.sleep(3)
LOG.info("Accessing database with updated password, "
f"user: {user1}, db: {db1}")
self._access_db(self.instance_ip, user1, new_pass, db1)
# user1 should not have access to db2 in Trove API mapping
# until granted
user1_dbs = self.client.list_resources(
f'instances/{self.instance_id}/users/{user1}/databases')
user1_dbs = [db['name'] for db in user1_dbs['databases']]
self.assertNotIn(db2, user1_dbs)
LOG.info(f"Granting user {user1} access to database {db2}")
grant_access = {
"databases": [
{
"name": db2
}
]
}
self.client.put_resource(
f'/instances/{self.instance_id}/users/{user1}/databases',
grant_access)
time.sleep(3)
user1_dbs = self.client.list_resources(
f'instances/{self.instance_id}/users/{user1}/databases')
user1_dbs = [db['name'] for db in user1_dbs['databases']]
self.assertIn(db2, user1_dbs)
LOG.info(f"Accessing database on {self.instance_ip}, user: {user1}, "
f"db: {db2}")
self._access_db(self.instance_ip, user1, new_pass, db2)
LOG.info(f"Revoking user {user1} access to database {db1}")
self.client.delete_resource(
f'instances/{self.instance_id}/users/{user1}/databases', db1)
# FIXME: user1 still have access to db1 because of the
# havior of public schema
# self.assertRaises(exceptions.TempestException, self._access_db,
# self.instance_ip, user1, constants.DB_PASS, db1)
# self.assertFalse(self._check_db_privilege(self.instance_ip,
# user1,
# constants.DB_PASS, db1))
self.assertRaises(exceptions.TempestException, self._access_db,
self.instance_ip, user1, new_pass, db1)
LOG.info(f"Revoking user {user1} access to database {db2}")
self.client.delete_resource(
f'instances/{self.instance_id}/users/{user1}/databases', db2)
self.assertRaises(exceptions.TempestException, self._access_db,
self.instance_ip, user1, new_pass, db2)
LOG.info(f"Granting user {user1} access to database {db2} again")
self.client.put_resource(
f'/instances/{self.instance_id}/users/{user1}/databases',
grant_access)
time.sleep(3)
LOG.info(f"Accessing database on {self.instance_ip}, user: {user1}, "
f"db: {db2}")
self._access_db(self.instance_ip, user1, new_pass, db2)
LOG.info(f"Deleting database {db2}")
self.client.delete_resource(
f'instances/{self.instance_id}/databases', db2)
# Wait for database deletion to complete
self.wait_for_database_deletion(self.instance_id, db2)
databases = self.get_databases(self.instance_id)
cur_db_names = [db['name'] for db in databases]
self.assertNotIn(db2, cur_db_names)
# test update_attributes interface
LOG.info(f"Updating user {user1} to {user2}")