Netpulse_SasS/server/cmd/netpulse-migrate/roles_test.go

package main

import (
	"context"
	"strings"
	"testing"

	"github.com/jackc/pgx/v5"
)

// Перевіряються рівно ті гілки ensureRoles, які завершуються ДО першого
// звернення до бази, — тому conn тут nil, і це навмисно: якби якась із
// цих гілок раптом почала ходити в базу, тест упав би на nil, а не
// пройшов би тихо.
//
// Головне, що тут захищається, — властивість «порожні змінні = стара
// поведінка». Її легко зламати наступною правкою, а помітити зламану
// можна лише на чужому робочому стенді.
//
// Чого цей тест НЕ покриває: усе, що починається з підключення. Видача
// паролів, перевірка входу, атрибут BYPASSRLS на тому кінці — жодного
// разу не виконуються без живого Postgres.
func TestEnsureRolesEarlyBranches(t *testing.T) {
	const owner = "postgres://netpulse:ownerpw@db:5432/netpulse?sslmode=disable"

	fresh := installInfo{fresh: true, appliedWas: 0}
	existing := installInfo{fresh: false, appliedWas: 68}

	// DSN, який docker compose збирає при порожньому NETPULSE_APP_PASSWORD:
	// та сама роль, що й у власника.
	collapsed := "postgres://netpulse:ownerpw@db:5432/netpulse?sslmode=disable"
	appRole := "postgres://netpulse_app:3f9a2b7c1d@db:5432/netpulse?sslmode=disable"

	cases := []struct {
		name    string
		appDSN  string
		wrkDSN  string
		inst    installInfo
		wantErr string // порожньо — має пройти без помилки й без роботи
	}{
		{"наявний стенд, .env без нових рядків", collapsed, "", existing, ""},
		{"чиста база без паролів — попередження, не помилка", collapsed, "", fresh, ""},
		{"нічого не передано (тести, go run)", "", "", fresh, ""},
		{"роль застосунку без ролі воркера", appRole, "", fresh, "NETPULSE_WORKER_PASSWORD"},
		{"те саме на наявній інсталяції", appRole, "", existing, "NETPULSE_WORKER_PASSWORD"},
	}

	for _, c := range cases {
		t.Run(c.name, func(t *testing.T) {
			err := ensureRoles(context.Background(), nil, owner, c.appDSN, c.wrkDSN, c.inst)
			switch {
			case c.wantErr == "" && err != nil:
				t.Fatalf("мало бути без роботи й без помилки, вийшло: %v", err)
			case c.wantErr != "" && err == nil:
				t.Fatalf("очікували відмову зі згадкою %q, вийшло nil", c.wantErr)
			case c.wantErr != "" && !strings.Contains(err.Error(), c.wantErr):
				t.Fatalf("очікували відмову зі згадкою %q, вийшло: %v", c.wantErr, err)
			}
		})
	}
}

// Чому паролі ролей мають бути в hex — на прикладі того самого розбирача,
// яким їх читатиме застосунок.
//
// Тест закріплює не бажану поведінку pgx, а факт: те, що виглядає як
// «пароль не підійшов», насправді буває зіпсованим DSN.
func TestPasswordCharactersInDSN(t *testing.T) {
	dsn := func(pw string) string {
		return "postgres://netpulse_app:" + pw + "@db:5432/netpulse?sslmode=disable"
	}

	// hex доїжджає до бази як є.
	cfg, err := pgx.ParseConfig(dsn("3f9a2b7c1d"))
	if err != nil {
		t.Fatalf("hex-пароль мав розібратись, а вийшло: %v", err)
	}
	if cfg.Password != "3f9a2b7c1d" || cfg.User != "netpulse_app" {
		t.Fatalf("hex-пароль спотворено: user=%q password=%q", cfg.User, cfg.Password)
	}

	// Скісна риска (звичайний символ base64) обриває DSN — і це помітно.
	if _, err := pgx.ParseConfig(dsn("aB/cD+eF=")); err == nil {
		t.Fatal("пароль зі скісною рискою мав дати помилку розбору")
	}

	// Відсоток гірший за скісну риску: помилки немає, а пароль інший.
	cfg, err = pgx.ParseConfig(dsn("aB%cDef"))
	if err != nil {
		t.Fatalf("несподівана помилка розбору: %v", err)
	}
	if cfg.Password == "aB%cDef" {
		t.Fatal("очікували, що відсоткову послідовність буде розкодовано; " +
			"якщо ні — підказку в dsnHint треба переписати")
	}
}