From bb6ec52732f86ce6c6666cde59a4f3647ab0d943 Mon Sep 17 00:00:00 2001 From: F04C Date: Thu, 12 Mar 2026 17:41:39 +0800 Subject: [PATCH] added db name --- repository/permission_repository.go | 14 +++++++------- repository/permission_repository_test.go | 2 +- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/repository/permission_repository.go b/repository/permission_repository.go index 4794125..66aba23 100644 --- a/repository/permission_repository.go +++ b/repository/permission_repository.go @@ -14,8 +14,8 @@ func GetPermissionByResourceActionAndRole(resource, action string, roleID int) ( query := ` SELECT p.permissions_id, p.permission_name, p.description, p.resource, p.action - FROM permissions p - INNER JOIN role_permissions rp + FROM uess_user_management.permissions p + INNER JOIN uess_user_management.role_permissions rp ON p.permissions_id = rp.permission_id WHERE p.resource = ? AND p.action = ? AND rp.role_id = ? AND rp.is_deleted = 0 LIMIT 1 @@ -48,7 +48,7 @@ func GetPermissionByResourceActionAndRole(resource, action string, roleID int) ( func GetPolicyAttributesByPermission(permissionID int) ([]models.PolicyAttribute, error) { query := ` SELECT id, attribute_name, attribute_type, comparison, attribute_value, permission_id - FROM policy_attributes + FROM uess_user_management.policy_attributes WHERE permission_id = ? ` @@ -84,7 +84,7 @@ func GetUserAttributes(userID string) (map[string]string, error) { query := ` SELECT attribute_name, attribute_value - FROM user_attributes + FROM uess_user_management.user_attributes WHERE users_id = ? ` @@ -116,7 +116,7 @@ func GetUserByID(userID string) (*models.User, error) { query := ` SELECT users_id, email_address - FROM users + FROM uess_user_management.users WHERE users_id = ? ` @@ -139,7 +139,7 @@ func GetUserByID(userID string) (*models.User, error) { func GetAllPermissions() ([]models.Permission, error) { query := ` SELECT id, permission_name, description, resource, action - FROM permissions + FROM uess_user_management.permissions ORDER BY id ` @@ -172,7 +172,7 @@ func GetAllPermissions() ([]models.Permission, error) { func GetAllPolicyAttributes() (map[int][]models.PolicyAttribute, error) { query := ` SELECT policy_attributes_id, attribute_name, attribute_type, comparison, attribute_value, permission_id - FROM policy_attributes + FROM uess_user_management.policy_attributes ORDER BY permission_id, policy_attributes_id ` diff --git a/repository/permission_repository_test.go b/repository/permission_repository_test.go index e0ad6d4..a644c9a 100644 --- a/repository/permission_repository_test.go +++ b/repository/permission_repository_test.go @@ -35,7 +35,7 @@ func TestGetPolicyAttributesByPermissionSuccess(t *testing.T) { AddRow(1, "department", "user", "=", "engineering", 1). AddRow(2, "level", "user", ">=", "5", 1) - mock.ExpectQuery("SELECT id, attribute_name, attribute_type, comparison, attribute_value, permission_id FROM policy_attributes WHERE permission_id = \\?"). + mock.ExpectQuery("SELECT id, attribute_name, attribute_type, comparison, attribute_value, permission_id FROM uess_user_management.policy_attributes WHERE permission_id = \\?"). WithArgs(1). WillReturnRows(rows)