first, added sql for tables and jwt emulation
This commit is contained in:
parent
0b69db0b4d
commit
3e30800ee0
88
tests/jwt_helper.go
Normal file
88
tests/jwt_helper.go
Normal file
@ -0,0 +1,88 @@
|
|||||||
|
package tests
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"github.com/golang-jwt/jwt/v5"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
func CreateJWT(userID string) string {
|
||||||
|
var privateKey = strings.Replace(`-----BEGIN RSA PRIVATE KEY-----
|
||||||
|
MIICWwIBAAKBgHgnvr7O2tiApjJfid1orFnIGm6980fZp+Lpbjo+NC/0whMFga2B
|
||||||
|
iw5b1G2Q/B2u0tpO1Fs/E8z7Lv1nYfr5jx2S8x6BdA4TS2kB9Kf0wn0+7wSlyikH
|
||||||
|
oKhbtzwXHZl17GsyEi6wHnsqNBSauyIWhpha8i+Y+3GyaOY536H47qyXAgMBAAEC
|
||||||
|
gYAOphnVPXbk6lpYzdkLC1Xn5EOEuNfOLLURLxBnPWozZo26r/Mtahu/9mYhrYlv
|
||||||
|
PP8r6mxta3VIil8iOdZyOLa/4d1LPd+UehgEXIJEiYXLtn7RS5eUnoPuQxssfs1k
|
||||||
|
OWjdN8p6SzppleegFTvGRX4KM3cDLfSphOk8JuBCrpSSYQJBAOdqizTSrdKMTuVe
|
||||||
|
c7Jk1JOJkyFuFs+N5zeryyeFGH7IpRdWy0rkWMxIUAi8Ap1vYVBPHv4tDOo3sy5X
|
||||||
|
VLc/knkCQQCE62pg+0TmsrhO/2Pgog6MLBkzlzXYMRp/01HbmznwYF+ejfPnzLkz
|
||||||
|
hnUlxRUNK3lhXM/7H6oAjvqF2R72u/OPAkEAterkmdbQfEZ+MwNoEiH/lie9OLdx
|
||||||
|
SSI1VGdBYcTYN7qFRW6eizYstBJYkDU0HQ0Uw+we4hMKJwk4W0KdvxxDiQJAeqlB
|
||||||
|
V1QqBneBbK10PzVuFV8QtrJhJyxRVwrtbKq38iMNuqUnI4+ijXEUpJFWVvv6nKXo
|
||||||
|
7McQvEk12dU/JNTX8wJAOlAtSNjp9tVwpMpC0w2St1eKc1L2SknjeohA5ldoBz8sGeZsPhTU3eHSD1neAZXLKN5K68z3zFBr20ubY9nyLw==
|
||||||
|
-----END RSA PRIVATE KEY-----`, "\t", "", -1)
|
||||||
|
|
||||||
|
privateKeyPem, err := jwt.ParseRSAPrivateKeyFromPEM([]byte(privateKey))
|
||||||
|
if err != nil {
|
||||||
|
panic(fmt.Errorf("failed to parse private key on <Create> of <JWT>: %w", err))
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
now := time.Now().UTC()
|
||||||
|
|
||||||
|
claims := jwt.MapClaims{
|
||||||
|
"id": userID,
|
||||||
|
"exp": now.Add(5 * time.Minute).Unix(),
|
||||||
|
"aud": "pena",
|
||||||
|
"iss": "pena-auth-service",
|
||||||
|
}
|
||||||
|
|
||||||
|
token, err := jwt.NewWithClaims(jwt.SigningMethodRS256, claims).SignedString(privateKeyPem)
|
||||||
|
if err != nil {
|
||||||
|
panic(fmt.Errorf("failed to sing on <Create> of <JWT>: %w", err))
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
return token
|
||||||
|
}
|
||||||
|
|
||||||
|
func CreateExpiredToken(userID string) string {
|
||||||
|
var privateKey = strings.Replace(`-----BEGIN RSA PRIVATE KEY-----
|
||||||
|
MIICWwIBAAKBgHgnvr7O2tiApjJfid1orFnIGm6980fZp+Lpbjo+NC/0whMFga2B
|
||||||
|
iw5b1G2Q/B2u0tpO1Fs/E8z7Lv1nYfr5jx2S8x6BdA4TS2kB9Kf0wn0+7wSlyikH
|
||||||
|
oKhbtzwXHZl17GsyEi6wHnsqNBSauyIWhpha8i+Y+3GyaOY536H47qyXAgMBAAEC
|
||||||
|
gYAOphnVPXbk6lpYzdkLC1Xn5EOEuNfOLLURLxBnPWozZo26r/Mtahu/9mYhrYlv
|
||||||
|
PP8r6mxta3VIil8iOdZyOLa/4d1LPd+UehgEXIJEiYXLtn7RS5eUnoPuQxssfs1k
|
||||||
|
OWjdN8p6SzppleegFTvGRX4KM3cDLfSphOk8JuBCrpSSYQJBAOdqizTSrdKMTuVe
|
||||||
|
c7Jk1JOJkyFuFs+N5zeryyeFGH7IpRdWy0rkWMxIUAi8Ap1vYVBPHv4tDOo3sy5X
|
||||||
|
VLc/knkCQQCE62pg+0TmsrhO/2Pgog6MLBkzlzXYMRp/01HbmznwYF+ejfPnzLkz
|
||||||
|
hnUlxRUNK3lhXM/7H6oAjvqF2R72u/OPAkEAterkmdbQfEZ+MwNoEiH/lie9OLdx
|
||||||
|
SSI1VGdBYcTYN7qFRW6eizYstBJYkDU0HQ0Uw+we4hMKJwk4W0KdvxxDiQJAeqlB
|
||||||
|
V1QqBneBbK10PzVuFV8QtrJhJyxRVwrtbKq38iMNuqUnI4+ijXEUpJFWVvv6nKXo
|
||||||
|
7McQvEk12dU/JNTX8wJAOlAtSNjp9tVwpMpC0w2St1eKc1L2SknjeohA5ldoBz8sGeZsPhTU3eHSD1neAZXLKN5K68z3zFBr20ubY9nyLw==
|
||||||
|
-----END RSA PRIVATE KEY-----`, "\t", "", -1)
|
||||||
|
|
||||||
|
privateKeyPem, err := jwt.ParseRSAPrivateKeyFromPEM([]byte(privateKey))
|
||||||
|
if err != nil {
|
||||||
|
panic(fmt.Errorf("failed to parse private key on <Create> of <JWT>: %w", err))
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
now := time.Now().UTC()
|
||||||
|
|
||||||
|
claims := jwt.MapClaims{
|
||||||
|
"id": userID,
|
||||||
|
"exp": now.Add(-1 * time.Hour).Unix(),
|
||||||
|
"aud": "pena",
|
||||||
|
"iss": "pena-auth-service",
|
||||||
|
}
|
||||||
|
|
||||||
|
token, err := jwt.NewWithClaims(jwt.SigningMethodRS256, claims).SignedString(privateKeyPem)
|
||||||
|
if err != nil {
|
||||||
|
panic(fmt.Errorf("failed to sing on <Create> of <JWT>: %w", err))
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
return token
|
||||||
|
}
|
||||||
@ -14,33 +14,33 @@ import (
|
|||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
var PublicKey = `-----BEGIN PUBLIC KEY-----MIGeMA0GCSqGSIb3DQEBAQUAA4GMADCBiAKBgHgnvr7O2tiApjJfid1orFnIGm6980fZp+Lpbjo+NC/0whMFga2Biw5b1G2Q/B2u0tpO1Fs/E8z7Lv1nYfr5jx2S8x6BdA4TS2kB9Kf0wn0+7wSlyikHoKhbtzwXHZl17GsyEi6wHnsqNBSauyIWhpha8i+Y+3GyaOY536H47qyXAgMBAAE=-----END PUBLIC KEY-----`
|
||||||
|
|
||||||
var baseURL = os.Getenv("API_BASE_URL")
|
var baseURL = os.Getenv("API_BASE_URL")
|
||||||
var validToken = os.Getenv("VALID_JWT_TOKEN")
|
var validToken = CreateJWT(validUserID) // validUserID
|
||||||
var expiredToken = os.Getenv("EXPIRED_JWT_TOKEN")
|
var expiredToken = CreateExpiredToken(validUserID)
|
||||||
|
|
||||||
//todo
|
// todo
|
||||||
var validTokenForDelete = os.Getenv("VALID_JWT_TOKEN_FOR_DELETE")
|
var validTokenForDelete = CreateJWT(userIDForDelete) // userIDForDelete
|
||||||
|
var validAdminToken = CreateJWT(validUserID) // os.Getenv("VALID_ADMIN_JWT_TOKEN")
|
||||||
var validAdminToken = os.Getenv("VALID_ADMIN_JWT_TOKEN")
|
var existingUserIDToken = CreateJWT(existingUserID) // existingUserID
|
||||||
var existingUserIDToken = os.Getenv("EXISTING_USER_ID_JWT_TOKEN")
|
|
||||||
|
|
||||||
// Токены для операций удаления
|
// Токены для операций удаления
|
||||||
// todo
|
// todo
|
||||||
var deleteAccountToken = os.Getenv("DELETE_ACCOUNT_TOKEN")
|
var deleteAccountToken = CreateJWT(userIDForDelete) // userIDForDelete
|
||||||
|
var deleteQuestionToken = CreateJWT(userIDForDelete) // userIDForDelete
|
||||||
|
var deleteQuizToken = CreateJWT(userIDForDelete) // userIDForDelete
|
||||||
|
var deleteResultToken = CreateJWT(userIDForDelete) // userIDForDelete
|
||||||
|
var deleteLeadTargetToken = CreateJWT(userIDForDelete) // userIDForDelete
|
||||||
|
var deletedAccountToken = CreateJWT(userIDForDelete) // userIDForDelete
|
||||||
|
var AccountWithOutPrivilegeToken = CreateJWT(userWithoutPrivileges) // userWithoutPrivileges
|
||||||
|
|
||||||
var deleteQuestionToken = os.Getenv("DELETE_QUESTION_TOKEN")
|
// todo
|
||||||
var deleteQuizToken = os.Getenv("DELETE_QUIZ_TOKEN")
|
var userIDForDelete = "user_for_delete_789"
|
||||||
var deleteResultToken = os.Getenv("DELETE_RESULT_TOKEN")
|
var existingUserID = "existing_user_456"
|
||||||
var deleteLeadTargetToken = os.Getenv("DELETE_LEAD_TARGET_TOKEN")
|
var testUserID = "test_user_123"
|
||||||
var deletedAccountToken = os.Getenv("DELETED_ACCOUNT_TOKEN")
|
var userWithoutPrivileges = "no_privileges_user"
|
||||||
var AccountWithOutPrivilegeToken = os.Getenv("ACCOUNT_WITH_PRIVILEGE_TOKEN")
|
var validUserID = "multi_privileges_user"
|
||||||
|
|
||||||
//todo
|
|
||||||
var userIDForDelete = os.Getenv("USER_ID_FOR_DELETE")
|
|
||||||
|
|
||||||
var existingUserID = os.Getenv("EXISTING_USER_ID")
|
|
||||||
var testUserID = os.Getenv("TEST_USER_ID")
|
|
||||||
var userWithoutPrivileges = os.Getenv("USER_NO_PRIVILEGES")
|
|
||||||
|
|
||||||
var sqlInjectionInput = "'; DROP TABLE accounts; --"
|
var sqlInjectionInput = "'; DROP TABLE accounts; --"
|
||||||
var xssInput = "<script>alert('xss')</script>"
|
var xssInput = "<script>alert('xss')</script>"
|
||||||
|
|||||||
94
tests/test_data.sql
Normal file
94
tests/test_data.sql
Normal file
@ -0,0 +1,94 @@
|
|||||||
|
-- DELETE FROM quiz_privilege_usage;
|
||||||
|
-- DELETE FROM privileges;
|
||||||
|
-- DELETE FROM answer;
|
||||||
|
-- DELETE FROM question;
|
||||||
|
-- DELETE FROM quiz;
|
||||||
|
-- DELETE FROM account;
|
||||||
|
|
||||||
|
|
||||||
|
INSERT INTO account (id, user_id, created_at, deleted) VALUES
|
||||||
|
('550e8400-e29b-41d4-a716-446655440001', 'test_user_123', CURRENT_TIMESTAMP, false),
|
||||||
|
('550e8400-e29b-41d4-a716-446655440002', 'existing_user_456', CURRENT_TIMESTAMP, false),
|
||||||
|
('550e8400-e29b-41d4-a716-446655440003', 'user_for_delete_789', CURRENT_TIMESTAMP, false),
|
||||||
|
('550e8400-e29b-41d4-a716-446655440004', 'deleted_user_999', CURRENT_TIMESTAMP, true),
|
||||||
|
('550e8400-e29b-41d4-a716-446655440005', 'no_privileges_user', CURRENT_TIMESTAMP, false),
|
||||||
|
('550e8400-e29b-41d4-a716-446655440006', 'multi_privileges_user', CURRENT_TIMESTAMP, false);
|
||||||
|
|
||||||
|
INSERT INTO privileges (privilegeID, account_id, privilege_name, amount) VALUES
|
||||||
|
('quizCnt', '550e8400-e29b-41d4-a716-446655440001', 'Количество Заявок', 10),
|
||||||
|
('quizUnlimTime', '550e8400-e29b-41d4-a716-446655440002', 'Безлимит Опросов', 30),
|
||||||
|
('squizHideBadge', '550e8400-e29b-41d4-a716-446655440002', 'Скрытие шильдика в опроснике', 15),
|
||||||
|
('quizManual', '550e8400-e29b-41d4-a716-446655440003', 'Заказать quiz', 2),
|
||||||
|
('quizGigaChat', '550e8400-e29b-41d4-a716-446655440003', 'Услуга от GigaChat', 5),
|
||||||
|
('quizCnt', '550e8400-e29b-41d4-a716-446655440006', 'Количество Заявок', 50),
|
||||||
|
('quizUnlimTime', '550e8400-e29b-41d4-a716-446655440006', 'Безлимит Опросов', 90),
|
||||||
|
('squizHideBadge', '550e8400-e29b-41d4-a716-446655440006', 'Скрытие шильдика в опроснике', 60),
|
||||||
|
('quizManual', '550e8400-e29b-41d4-a716-446655440006', 'Заказать quiz', 10),
|
||||||
|
('quizGigaChat', '550e8400-e29b-41d4-a716-446655440006', 'Услуга от GigaChat', 20);
|
||||||
|
|
||||||
|
INSERT INTO quiz (qid, accountid, name, description, status, created_at, updated_at) VALUES
|
||||||
|
(gen_random_uuid(), 'test_user_123', 'Тестовый квиз 1', 'Описание тестового квиза 1', 'start', CURRENT_TIMESTAMP, CURRENT_TIMESTAMP),
|
||||||
|
(gen_random_uuid(), 'test_user_123', 'Тестовый квиз 2', 'Описание тестового квиза 2', 'draft', CURRENT_TIMESTAMP, CURRENT_TIMESTAMP),
|
||||||
|
(gen_random_uuid(), 'existing_user_456', 'Существующий квиз 1', 'Описание существующего квиза 1', 'start', CURRENT_TIMESTAMP, CURRENT_TIMESTAMP),
|
||||||
|
(gen_random_uuid(), 'existing_user_456', 'Существующий квиз 2', 'Описание существующего квиза 2', 'template', CURRENT_TIMESTAMP, CURRENT_TIMESTAMP),
|
||||||
|
(gen_random_uuid(), 'user_for_delete_789', 'Квиз для удаления 1', 'Описание квиза для удаления 1', 'start', CURRENT_TIMESTAMP, CURRENT_TIMESTAMP),
|
||||||
|
(gen_random_uuid(), 'user_for_delete_789', 'Квиз для удаления 2', 'Описание квиза для удаления 2', 'draft', CURRENT_TIMESTAMP, CURRENT_TIMESTAMP),
|
||||||
|
(gen_random_uuid(), 'multi_privileges_user', 'Мультипривилегированный квиз 1', 'Описание мультипривилегированного квиза 1', 'start', CURRENT_TIMESTAMP, CURRENT_TIMESTAMP),
|
||||||
|
(gen_random_uuid(), 'multi_privileges_user', 'Мультипривилегированный квиз 2', 'Описание мультипривилегированного квиза 2', 'template', CURRENT_TIMESTAMP, CURRENT_TIMESTAMP);
|
||||||
|
|
||||||
|
DO $$
|
||||||
|
DECLARE
|
||||||
|
quiz1_id BIGINT;
|
||||||
|
quiz2_id BIGINT;
|
||||||
|
quiz3_id BIGINT;
|
||||||
|
quiz4_id BIGINT;
|
||||||
|
quiz5_id BIGINT;
|
||||||
|
quiz6_id BIGINT;
|
||||||
|
quiz7_id BIGINT;
|
||||||
|
quiz8_id BIGINT;
|
||||||
|
BEGIN
|
||||||
|
SELECT id INTO quiz1_id FROM quiz WHERE name = 'Тестовый квиз 1' LIMIT 1;
|
||||||
|
SELECT id INTO quiz2_id FROM quiz WHERE name = 'Тестовый квиз 2' LIMIT 1;
|
||||||
|
SELECT id INTO quiz3_id FROM quiz WHERE name = 'Существующий квиз 1' LIMIT 1;
|
||||||
|
SELECT id INTO quiz4_id FROM quiz WHERE name = 'Существующий квиз 2' LIMIT 1;
|
||||||
|
SELECT id INTO quiz5_id FROM quiz WHERE name = 'Квиз для удаления 1' LIMIT 1;
|
||||||
|
SELECT id INTO quiz6_id FROM quiz WHERE name = 'Квиз для удаления 2' LIMIT 1;
|
||||||
|
SELECT id INTO quiz7_id FROM quiz WHERE name = 'Мультипривилегированный квиз 1' LIMIT 1;
|
||||||
|
SELECT id INTO quiz8_id FROM quiz WHERE name = 'Мультипривилегированный квиз 2' LIMIT 1;
|
||||||
|
|
||||||
|
INSERT INTO question (quiz_id, title, description, questiontype, required, deleted, page, content, version) VALUES
|
||||||
|
(quiz1_id, 'Первый вопрос тестового квиза', 'Описание первого вопроса', 'variant', true, false, 1, '{"options": ["Вариант 1", "Вариант 2", "Вариант 3"]}', 1),
|
||||||
|
(quiz1_id, 'Второй вопрос тестового квиза', 'Описание второго вопроса', 'text', false, false, 1, '{"placeholder": "Введите ответ"}', 1),
|
||||||
|
(quiz2_id, 'Вопрос черновика', 'Описание вопроса черновика', 'select', true, false, 1, '{"options": ["Выбор 1", "Выбор 2"]}', 1),
|
||||||
|
(quiz3_id, 'Вопрос существующего квиза', 'Описание вопроса существующего квиза', 'rating', false, false, 1, '{"max_rating": 5}', 1),
|
||||||
|
(quiz4_id, 'Вопрос шаблона', 'Описание вопроса шаблона', 'emoji', true, false, 1, '{"emojis": ["😀", "😢", "😡"]}', 1),
|
||||||
|
(quiz5_id, 'Вопрос для удаления 1', 'Описание вопроса для удаления 1', 'variant', true, false, 1, '{"options": ["Удалить 1", "Удалить 2"]}', 1),
|
||||||
|
(quiz5_id, 'Вопрос для удаления 2', 'Описание вопроса для удаления 2', 'text', false, false, 1, '{"placeholder": "Введите текст для удаления"}', 1),
|
||||||
|
(quiz6_id, 'Вопрос черновика для удаления', 'Описание вопроса черновика для удаления', 'select', true, false, 1, '{"options": ["Удалить выбор 1", "Удалить выбор 2"]}', 1),
|
||||||
|
(quiz7_id, 'Вопрос мультипривилегированного квиза 1', 'Описание вопроса мультипривилегированного квиза 1', 'images', true, false, 1, '{"images": ["image1.jpg", "image2.jpg"]}', 1),
|
||||||
|
(quiz8_id, 'Вопрос мультипривилегированного квиза 2', 'Описание вопроса мультипривилегированного квиза 2', 'varimg', false, false, 1, '{"options": ["Вариант с картинкой 1", "Вариант с картинкой 2"], "images": ["img1.jpg", "img2.jpg"]}', 1);
|
||||||
|
|
||||||
|
INSERT INTO answer (content, quiz_id, question_id, fingerprint, session, version) VALUES
|
||||||
|
('Ответ на первый вопрос', quiz1_id, (SELECT id FROM question WHERE title = 'Первый вопрос тестового квиза' LIMIT 1), 'fp_test_1', 'session_1', 1),
|
||||||
|
('Ответ на второй вопрос', quiz1_id, (SELECT id FROM question WHERE title = 'Второй вопрос тестового квиза' LIMIT 1), 'fp_test_2', 'session_1', 1),
|
||||||
|
('Ответ на вопрос существующего квиза', quiz3_id, (SELECT id FROM question WHERE title = 'Вопрос существующего квиза' LIMIT 1), 'fp_existing_1', 'session_2', 1),
|
||||||
|
('Ответ на вопрос шаблона', quiz4_id, (SELECT id FROM question WHERE title = 'Вопрос шаблона' LIMIT 1), 'fp_template_1', 'session_3', 1),
|
||||||
|
('Ответ на вопрос для удаления 1', quiz5_id, (SELECT id FROM question WHERE title = 'Вопрос для удаления 1' LIMIT 1), 'fp_delete_1', 'session_4', 1),
|
||||||
|
('Ответ на вопрос для удаления 2', quiz5_id, (SELECT id FROM question WHERE title = 'Вопрос для удаления 2' LIMIT 1), 'fp_delete_2', 'session_4', 1),
|
||||||
|
('Ответ на вопрос мультипривилегированного квиза 1', quiz7_id, (SELECT id FROM question WHERE title = 'Вопрос мультипривилегированного квиза 1' LIMIT 1), 'fp_multipriv_1', 'session_5', 1),
|
||||||
|
('Ответ на вопрос мультипривилегированного квиза 2', quiz8_id, (SELECT id FROM question WHERE title = 'Вопрос мультипривилегированного квиза 2' LIMIT 1), 'fp_multipriv_2', 'session_6', 1);
|
||||||
|
|
||||||
|
INSERT INTO quiz_privilege_usage (quiz_id, privilege_id, used_count) VALUES
|
||||||
|
(quiz1_id, 'quizCnt', 2),
|
||||||
|
(quiz3_id, 'quizUnlimTime', 1),
|
||||||
|
(quiz4_id, 'squizHideBadge', 1),
|
||||||
|
(quiz5_id, 'quizManual', 2),
|
||||||
|
(quiz7_id, 'quizGigaChat', 1),
|
||||||
|
(quiz8_id, 'quizCnt', 1);
|
||||||
|
|
||||||
|
END $$;
|
||||||
|
|
||||||
|
UPDATE quiz SET
|
||||||
|
questions_count = (SELECT COUNT(*) FROM question WHERE quiz_id = quiz.id AND deleted = false),
|
||||||
|
answers_count = (SELECT COUNT(*) FROM answer WHERE quiz_id = quiz.id)
|
||||||
|
WHERE id IN (SELECT DISTINCT quiz_id FROM question);
|
||||||
Loading…
Reference in New Issue
Block a user