- шаги PocketBase переехали из файла в пакет internal/adapter/repo/pocketbase/migrations, файл на шаг с именем зарегистрированного шага; туда же имена коллекций, срок сессии — в provider.go - ключ migrations в docs/.docs.json наведён на этот каталог: прежнее значение указывало на несуществующий migrations/, и шаг гейта проходил зелёным при всякой правке схемы - app.go подключает пакет шагов явным пустым импортом: пропавшая ссылка на константы унесла бы регистрацию, и хранилище поднялось бы без коллекций
730 lines
30 KiB
Go
730 lines
30 KiB
Go
package http
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"log/slog"
|
|
"mime/multipart"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"regexp"
|
|
"strings"
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/pocketbase/pocketbase/apis"
|
|
"github.com/pocketbase/pocketbase/core"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"git.vakhrushev.me/av/transcriber/internal/adapter/recognizer"
|
|
pbrepo "git.vakhrushev.me/av/transcriber/internal/adapter/repo/pocketbase"
|
|
"git.vakhrushev.me/av/transcriber/internal/adapter/repo/pocketbase/migrations"
|
|
"git.vakhrushev.me/av/transcriber/internal/contract"
|
|
"git.vakhrushev.me/av/transcriber/internal/entity"
|
|
"git.vakhrushev.me/av/transcriber/internal/service"
|
|
)
|
|
|
|
// Подставные адаптеры вместо ffprobe и ffmpeg. Проверки судят приём — что
|
|
// запись сохранена, задача заведена и ответ такой, какой обещан, — а не
|
|
// способность внешней программы разобрать звук. Внешних программ здесь нет
|
|
// ни одной, и видно это по списку импортов.
|
|
|
|
// stubMetaViewer отдаёт заданную длительность либо заданную ошибку.
|
|
type stubMetaViewer struct {
|
|
seconds int
|
|
err error
|
|
}
|
|
|
|
func (m *stubMetaViewer) GetInfo(string) (*contract.AudioInfo, error) {
|
|
if m.err != nil {
|
|
return nil, m.err
|
|
}
|
|
return &contract.AudioInfo{Seconds: m.seconds}, nil
|
|
}
|
|
|
|
// stubConverter молчалив: приём конвертацию не делает, и ни одна проверка
|
|
// этого файла её не зовёт.
|
|
type stubConverter struct{}
|
|
|
|
func (c *stubConverter) Convert(string, string) error { return nil }
|
|
|
|
// TestTgSender: приём по HTTP в Telegram не отвечает, но сервису отправитель нужен.
|
|
type TestTgSender struct{}
|
|
|
|
func (s *TestTgSender) Send(msg string, chatId int64, replyMsgId *int) error {
|
|
return nil
|
|
}
|
|
|
|
// readableMetaViewer — источник метаданных, который читает любую запись.
|
|
func readableMetaViewer() *stubMetaViewer {
|
|
return &stubMetaViewer{seconds: 42}
|
|
}
|
|
|
|
// testEnv — собранное окружение одной проверки. Каталог данных свой у каждой:
|
|
// рабочий каталог процесса проверки не трогают.
|
|
type testEnv struct {
|
|
mux http.Handler
|
|
handler *TranscribeHandler
|
|
app core.App
|
|
journal *journalBuffer
|
|
// session — значение сессии вошедшего. Приём и опрос закрыты за
|
|
// аутентификацией, и проверка, судящая их по существу, обязана предъявить
|
|
// сессию ровно так же, как это делает браузер.
|
|
session string
|
|
// account — учётная запись, которой выдана сессия. Нужна проверкам выхода.
|
|
account *core.Record
|
|
}
|
|
|
|
// serve шлёт запрос от имени вошедшего: сессия предъявляется кукой — тем же
|
|
// способом, каким её предъявляет браузер. Заголовок проверки не ставят: куку в
|
|
// него перекладывает слой предъявления, и подмена его здесь означала бы проверку
|
|
// не той цепочки.
|
|
//
|
|
// Проверки, судящие отказ без сессии, зовут `mux` напрямую.
|
|
func (e *testEnv) serve(w http.ResponseWriter, req *http.Request) {
|
|
req.AddCookie(&http.Cookie{Name: SessionCookieName, Value: e.session})
|
|
e.mux.ServeHTTP(w, req)
|
|
}
|
|
|
|
// newTestAccount заводит учётную запись и выдаёт ей сессию.
|
|
//
|
|
// Запись создаётся прямым сохранением, а не запросом к API: заводить её
|
|
// запросом больше нельзя — создание закрыто шагом схемы, и в этом весь смысл
|
|
// изменения. Прямое сохранение идёт мимо правил доступа так же, как идёт вход,
|
|
// когда учётную запись заводит само хранилище.
|
|
func newTestAccount(t *testing.T, app core.App) (*core.Record, string) {
|
|
t.Helper()
|
|
|
|
users, err := app.FindCollectionByNameOrId("users")
|
|
require.NoError(t, err)
|
|
|
|
record := core.NewRecord(users)
|
|
record.Set("email", "person@example.com")
|
|
record.Set("verified", true)
|
|
// Случайный пароль ставит и само хранилище, когда заводит запись по входу у
|
|
// провайдера: запись auth-коллекции без пароля не сохраняется, а войти по
|
|
// нему всё равно нельзя — парольный вход выключен шагом схемы.
|
|
record.SetRandomPassword()
|
|
require.NoError(t, app.Save(record))
|
|
|
|
token, err := record.NewAuthToken()
|
|
require.NoError(t, err)
|
|
|
|
return record, token
|
|
}
|
|
|
|
// journalBuffer — перехваченный журнал одной проверки. Свой на случай: общий на
|
|
// пакет сделал бы исход функцией от соседних случаев — «поля на месте» прошло бы
|
|
// на чужой строке, а «маркера нет» покраснело бы от чужой. Замок нужен потому,
|
|
// что пишущих в него потоков два: логгер сервиса и логгер обработчика.
|
|
type journalBuffer struct {
|
|
mu sync.Mutex
|
|
text strings.Builder
|
|
}
|
|
|
|
func (b *journalBuffer) Write(p []byte) (int, error) {
|
|
b.mu.Lock()
|
|
defer b.mu.Unlock()
|
|
return b.text.Write(p)
|
|
}
|
|
|
|
// String отдаёт весь перехваченный текст. Проверки ищут в нём значение, а не имя
|
|
// поля: имя, вернувшееся под другим ключом, поиск по ключу не разбудил бы.
|
|
func (b *journalBuffer) String() string {
|
|
b.mu.Lock()
|
|
defer b.mu.Unlock()
|
|
return b.text.String()
|
|
}
|
|
|
|
// newTestStorage поднимает хранилище на пустом каталоге и накатывает схему —
|
|
// ровно тем же путём, каким это делает сервис при старте.
|
|
func newTestStorage(t *testing.T) core.App {
|
|
t.Helper()
|
|
|
|
app, err := pbrepo.New(t.TempDir())
|
|
require.NoError(t, err)
|
|
t.Cleanup(func() {
|
|
if err := app.ResetBootstrapState(); err != nil {
|
|
t.Logf("не удалось закрыть хранилище: %v", err)
|
|
}
|
|
})
|
|
|
|
return app
|
|
}
|
|
|
|
func setupTestEnv(t *testing.T, metaviewer contract.AudioMetaViewer) *testEnv {
|
|
app := newTestStorage(t)
|
|
|
|
pbrepo.BindPanelRules(app)
|
|
|
|
fileRepo := pbrepo.NewFileRepository(app)
|
|
jobRepo := pbrepo.NewTranscriptJobRepository(app)
|
|
|
|
// Журнал уходит в буфер, а не в никуда: по нему судит проверка запрета на
|
|
// имя отправителя. Вывод прогона от этого не меняется — ERROR-строки ветки
|
|
// отказа по-прежнему не попадают на экран и не размывают признак, по
|
|
// которому отличают новый красный шаг гейта от объявленного долга.
|
|
journal := &journalBuffer{}
|
|
logger := slog.New(slog.NewTextHandler(journal, nil))
|
|
|
|
trsService := service.NewTranscribeService(
|
|
jobRepo,
|
|
fileRepo,
|
|
metaviewer,
|
|
&stubConverter{},
|
|
&recognizer.MemoryAudioRecognizer{},
|
|
&TestTgSender{},
|
|
logger,
|
|
)
|
|
|
|
handler := NewTranscribeHandler(jobRepo, trsService, logger)
|
|
|
|
// Роутер собирается тем же способом, что и боевой: маршруты вешает сам
|
|
// обработчик, и проверка судит ту же цепочку, что и прод.
|
|
r, err := apis.NewRouter(app)
|
|
require.NoError(t, err)
|
|
handler.Register(r)
|
|
|
|
mux, err := r.BuildMux()
|
|
require.NoError(t, err)
|
|
|
|
account, session := newTestAccount(t, app)
|
|
|
|
return &testEnv{
|
|
mux: mux,
|
|
handler: handler,
|
|
app: app,
|
|
journal: journal,
|
|
session: session,
|
|
account: account,
|
|
}
|
|
}
|
|
|
|
// createMultipartRequest собирает запрос из имени и содержимого. Файла на диске
|
|
// для этого не нужно: имя проверяет выбор расширения, содержимое — сохранение.
|
|
func createMultipartRequest(t *testing.T, fileName string, content []byte) *http.Request {
|
|
return createMultipartRequestWithField(t, "audio", fileName, content)
|
|
}
|
|
|
|
// createMultipartRequestWithField кладёт запись в поле с заданным именем —
|
|
// нужно, чтобы построить форму без поля `audio`.
|
|
func createMultipartRequestWithField(t *testing.T, field, fileName string, content []byte) *http.Request {
|
|
var buf bytes.Buffer
|
|
writer := multipart.NewWriter(&buf)
|
|
|
|
part, err := writer.CreateFormFile(field, fileName)
|
|
require.NoError(t, err)
|
|
|
|
_, err = part.Write(content)
|
|
require.NoError(t, err)
|
|
|
|
err = writer.Close()
|
|
require.NoError(t, err)
|
|
|
|
req := httptest.NewRequest("POST", "/api/audio", &buf)
|
|
req.Header.Set("Content-Type", writer.FormDataContentType())
|
|
|
|
return req
|
|
}
|
|
|
|
// storedFileNames отдаёт имена, под которыми файлы легли в хранилище.
|
|
func storedFileNames(t *testing.T, env *testEnv) []string {
|
|
records, err := env.app.FindAllRecords(migrations.FilesCollection)
|
|
require.NoError(t, err)
|
|
|
|
var names []string
|
|
for _, record := range records {
|
|
names = append(names, record.GetStringSlice("file")...)
|
|
}
|
|
return names
|
|
}
|
|
|
|
// countFiles считает записи о файлах.
|
|
func countFiles(t *testing.T, env *testEnv) int {
|
|
records, err := env.app.FindAllRecords(migrations.FilesCollection)
|
|
require.NoError(t, err)
|
|
return len(records)
|
|
}
|
|
|
|
// countJobs считает заведённые задачи расшифровки.
|
|
func countJobs(t *testing.T, env *testEnv) int {
|
|
records, err := env.app.FindAllRecords(migrations.JobsCollection)
|
|
require.NoError(t, err)
|
|
return len(records)
|
|
}
|
|
|
|
// jobWithFile заводит задачу вместе с её записью: ссылка на файл обязательна
|
|
// схемой, потому что без неё задача не пройдёт ни одного шага.
|
|
func jobWithFile(t *testing.T, env *testEnv) *entity.TranscribeJob {
|
|
t.Helper()
|
|
|
|
repo := pbrepo.NewFileRepository(env.app)
|
|
work, err := repo.Stage(".mp3", strings.NewReader("запись"))
|
|
require.NoError(t, err)
|
|
defer func() { require.NoError(t, work.Close()) }()
|
|
|
|
file, err := repo.CreateLocal("sample.mp3", work)
|
|
require.NoError(t, err)
|
|
|
|
job := &entity.TranscribeJob{State: entity.StateCreated, Source: entity.SourceApi, FileID: &file.Id}
|
|
require.NoError(t, env.handler.jobRepo.Create(job))
|
|
return job
|
|
}
|
|
|
|
// storedContent читает содержимое файла из хранилища.
|
|
func storedContent(t *testing.T, env *testEnv, fileID string) []byte {
|
|
repo := pbrepo.NewFileRepository(env.app)
|
|
reader, err := repo.Open(fileID)
|
|
require.NoError(t, err)
|
|
defer reader.Close()
|
|
|
|
var buf bytes.Buffer
|
|
_, err = buf.ReadFrom(reader)
|
|
require.NoError(t, err)
|
|
return buf.Bytes()
|
|
}
|
|
|
|
func TestCreateTranscribeJob_Success(t *testing.T) {
|
|
env := setupTestEnv(t, readableMetaViewer())
|
|
|
|
content := []byte("содержимое записи, которое обязано доехать до диска целиком")
|
|
req := createMultipartRequest(t, "sample.m4a", content)
|
|
|
|
w := httptest.NewRecorder()
|
|
env.serve(w, req)
|
|
|
|
require.Equal(t, http.StatusCreated, w.Code)
|
|
|
|
// Имена полей ответа нормативны: контракт HTTP API объявлен необратимым.
|
|
// Судим по сырому JSON — разбор в CreateTranscribeJobResponse переименовал
|
|
// бы тег вместе с ожиданием, и проверка не смогла бы упасть.
|
|
var raw map[string]json.RawMessage
|
|
err := json.Unmarshal(w.Body.Bytes(), &raw)
|
|
require.NoError(t, err)
|
|
assert.Contains(t, raw, "job_id")
|
|
assert.Contains(t, raw, "status")
|
|
|
|
var response CreateTranscribeJobResponse
|
|
err = json.Unmarshal(w.Body.Bytes(), &response)
|
|
require.NoError(t, err)
|
|
|
|
assert.NotEmpty(t, response.JobID)
|
|
assert.Equal(t, entity.StateCreated, response.State)
|
|
|
|
// Задача действительно заведена, а не только названа в ответе: иначе
|
|
// отправитель получит идентификатор записи, которой не будет никогда.
|
|
require.Equal(t, 1, countJobs(t, env))
|
|
|
|
job, err := env.handler.jobRepo.GetByID(response.JobID)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, entity.StateCreated, job.State)
|
|
require.NotNil(t, job.FileID)
|
|
assert.NotEmpty(t, *job.FileID)
|
|
|
|
// Содержимое лежит в хранилище одним файлом и целиком.
|
|
require.Equal(t, 1, countFiles(t, env))
|
|
assert.Equal(t, content, storedContent(t, env, *job.FileID))
|
|
}
|
|
|
|
func TestCreateTranscribeJob_NoFile(t *testing.T) {
|
|
// Две ветки одного сценария: тела нет вовсе и форма есть, а поля в ней нет.
|
|
// Вторая — та, что описана требованием; первая ходит тем же путём.
|
|
testCases := []struct {
|
|
name string
|
|
req func(t *testing.T) *http.Request
|
|
}{
|
|
{
|
|
name: "no body at all",
|
|
req: func(t *testing.T) *http.Request {
|
|
// Запрос строится так, как его видит сервер: у пришедшего по
|
|
// проводу тело не бывает пустым указателем.
|
|
return httptest.NewRequest("POST", "/api/audio", http.NoBody)
|
|
},
|
|
},
|
|
{
|
|
name: "form without audio field",
|
|
req: func(t *testing.T) *http.Request {
|
|
return createMultipartRequestWithField(t, "attachment", "sample.m4a", []byte("запись"))
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
env := setupTestEnv(t, readableMetaViewer())
|
|
|
|
w := httptest.NewRecorder()
|
|
env.serve(w, tc.req(t))
|
|
|
|
require.Equal(t, http.StatusBadRequest, w.Code)
|
|
|
|
var response map[string]string
|
|
err := json.Unmarshal(w.Body.Bytes(), &response)
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, "No audio file provided", response["error"])
|
|
assert.Equal(t, 0, countFiles(t, env))
|
|
assert.Equal(t, 0, countJobs(t, env))
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestCreateTranscribeJob_EmptyFile(t *testing.T) {
|
|
env := setupTestEnv(t, readableMetaViewer())
|
|
|
|
// Собственного порога по размеру у приёма нет: годность записи судит
|
|
// источник метаданных, а не приём.
|
|
req := createMultipartRequest(t, "empty.m4a", nil)
|
|
|
|
w := httptest.NewRecorder()
|
|
env.serve(w, req)
|
|
|
|
require.Equal(t, http.StatusCreated, w.Code)
|
|
|
|
var response CreateTranscribeJobResponse
|
|
err := json.Unmarshal(w.Body.Bytes(), &response)
|
|
require.NoError(t, err)
|
|
|
|
assert.NotEmpty(t, response.JobID)
|
|
assert.Equal(t, entity.StateCreated, response.State)
|
|
}
|
|
|
|
func TestCreateTranscribeJob_DifferentFileExtensions(t *testing.T) {
|
|
testCases := []struct {
|
|
name string
|
|
fileName string
|
|
expectExt string
|
|
}{
|
|
{
|
|
name: "m4a file",
|
|
fileName: "test.m4a",
|
|
expectExt: ".m4a",
|
|
},
|
|
{
|
|
name: "mp3 file",
|
|
fileName: "test.mp3",
|
|
expectExt: ".mp3",
|
|
},
|
|
{
|
|
name: "wav file",
|
|
fileName: "test.wav",
|
|
expectExt: ".wav",
|
|
},
|
|
{
|
|
name: "file without extension",
|
|
fileName: "test",
|
|
expectExt: ".audio",
|
|
},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
env := setupTestEnv(t, readableMetaViewer())
|
|
|
|
req := createMultipartRequest(t, tc.fileName, []byte("запись"))
|
|
|
|
w := httptest.NewRecorder()
|
|
env.serve(w, req)
|
|
|
|
require.Equal(t, http.StatusCreated, w.Code)
|
|
|
|
names := storedFileNames(t, env)
|
|
require.Len(t, names, 1)
|
|
|
|
// Имя отправителя в хранилище не попадает: имя файла — свой
|
|
// идентификатор, от отправителя взято только расширение. Суффикс
|
|
// дописывает само хранилище, поэтому сверяем хвост, а не Ext.
|
|
assert.True(t, strings.HasSuffix(names[0], tc.expectExt),
|
|
"имя в хранилище %q оканчивается на %q", names[0], tc.expectExt)
|
|
assert.NotContains(t, names[0], strings.TrimSuffix(tc.fileName, tc.expectExt))
|
|
})
|
|
}
|
|
}
|
|
|
|
// Имя, данное отправителем, в хранилище не попадает целиком — умолчание
|
|
// библиотеки, строящее имя файла из него, не применяется. Проверка отдельная от
|
|
// перебора расширений: там сверяется хвост, здесь — что основы имени нет.
|
|
func TestCreateTranscribeJob_SenderFileNameNotStored(t *testing.T) {
|
|
env := setupTestEnv(t, readableMetaViewer())
|
|
|
|
req := createMultipartRequest(t, "секретное-слово.mp3", []byte("запись"))
|
|
|
|
w := httptest.NewRecorder()
|
|
env.serve(w, req)
|
|
|
|
require.Equal(t, http.StatusCreated, w.Code)
|
|
|
|
names := storedFileNames(t, env)
|
|
require.Len(t, names, 1)
|
|
|
|
assert.NotContains(t, names[0], "секретное-слово",
|
|
"имя, данное отправителем, в хранилище не попадает")
|
|
assert.True(t, strings.HasSuffix(names[0], ".mp3"),
|
|
"расширение при этом сохраняется: %q", names[0])
|
|
}
|
|
|
|
func TestCreateTranscribeJob_MetaViewerFailure(t *testing.T) {
|
|
env := setupTestEnv(t, &stubMetaViewer{err: errors.New("не удалось прочитать запись")})
|
|
|
|
req := createMultipartRequest(t, "broken.m4a", []byte("не запись вовсе"))
|
|
|
|
w := httptest.NewRecorder()
|
|
env.serve(w, req)
|
|
|
|
require.Equal(t, http.StatusInternalServerError, w.Code)
|
|
|
|
var response map[string]string
|
|
err := json.Unmarshal(w.Body.Bytes(), &response)
|
|
require.NoError(t, err)
|
|
|
|
// Причина отказа принадлежит журналу, а не отправителю.
|
|
assert.Equal(t, "Failed to create transcibe job", response["error"])
|
|
assert.NotContains(t, w.Body.String(), "не удалось прочитать запись")
|
|
|
|
assert.Equal(t, 0, countJobs(t, env))
|
|
}
|
|
|
|
// senderNameMarker — метка внутри имени, которое даёт отправитель. ASCII и
|
|
// заведомо уникальна: в остальном выводе прогона такой строки нет, поэтому
|
|
// находка означает утечку, а не совпадение. Ищется она **значением**, а не
|
|
// именем журнального поля: имя, вернувшееся под другим ключом, поиск по ключу
|
|
// пропустил бы.
|
|
const senderNameMarker = "SENDERNAMELEAKMARKER7Q2"
|
|
|
|
// Тексты, по которым проверки находят журнальные строки. Первый — записанный
|
|
// долг `docs/conventions/logging.md`: `msg` обязан стать короткой категорией.
|
|
// Когда долг закроют, правка будет здесь и одна.
|
|
const (
|
|
msgIntake = "Creating transcribe job"
|
|
// Отказ пишет доменная граница — приём, — а не транспорт: конвенция просит
|
|
// логировать ошибку один раз, и повторная запись транспорта снята.
|
|
msgIntakeErr = "Failed to get file info"
|
|
)
|
|
|
|
func TestCreateTranscribeJob_SenderFileNameNotLogged(t *testing.T) {
|
|
env := setupTestEnv(t, readableMetaViewer())
|
|
|
|
// Метка стоит в основе имени, а расширение обычное: расширение запретом не
|
|
// накрыто и в журнале остаётся законно.
|
|
req := createMultipartRequest(t, senderNameMarker+".mp3", []byte("запись"))
|
|
|
|
w := httptest.NewRecorder()
|
|
env.serve(w, req)
|
|
|
|
require.Equal(t, http.StatusCreated, w.Code)
|
|
|
|
journal := env.journal.String()
|
|
|
|
// Сперва — что журнал приёма вообще перехвачен. Без этого утверждения
|
|
// пустой буфер сделал бы проверку запрета зелёной, ничего не прочитав.
|
|
require.Contains(t, journal, msgIntake,
|
|
"строка приёма попадает в перехваченный журнал")
|
|
|
|
assert.NotContains(t, journal, senderNameMarker,
|
|
"имя, данное отправителем, не пишется в журнал: инвариант приватности")
|
|
}
|
|
|
|
func TestCreateTranscribeJob_SenderFileNameNotLoggedOnFailure(t *testing.T) {
|
|
// Отказ — тот путь, где имя приехало бы в журнал текстом ошибки: приём
|
|
// назван конвенцией логирующей границей, и цепочка `%w` осядет полем error.
|
|
env := setupTestEnv(t, &stubMetaViewer{err: errors.New("не удалось прочитать запись")})
|
|
|
|
req := createMultipartRequest(t, senderNameMarker+".mp3", []byte("не запись вовсе"))
|
|
|
|
w := httptest.NewRecorder()
|
|
env.serve(w, req)
|
|
|
|
require.Equal(t, http.StatusInternalServerError, w.Code)
|
|
|
|
journal := env.journal.String()
|
|
|
|
// Сперва — что журнал ветки отказа вообще перехвачен: обработчик пишет свою
|
|
// строку, и без неё оракул молча сузился бы вдвое.
|
|
require.Contains(t, journal, msgIntakeErr,
|
|
"строка приёма об отказе попадает в перехваченный журнал")
|
|
|
|
assert.NotContains(t, journal, senderNameMarker,
|
|
"имя отправителя не пишется в журнал и на пути отказа")
|
|
}
|
|
|
|
// Имя, под которым файл лёг в хранилище, — это последняя часть ссылки
|
|
// `/api/files/...`, по которой запись скачивают. Попав в журнал, строка стала бы
|
|
// бессрочным ключом к чужому аудио, поэтому в журнал идёт имя, заданное
|
|
// сервисом, а суффикс хранилища — нет.
|
|
func TestCreateTranscribeJob_StorageFileNameNotLogged(t *testing.T) {
|
|
env := setupTestEnv(t, readableMetaViewer())
|
|
|
|
req := createMultipartRequest(t, "sample.mp3", []byte("запись"))
|
|
|
|
w := httptest.NewRecorder()
|
|
env.serve(w, req)
|
|
|
|
require.Equal(t, http.StatusCreated, w.Code)
|
|
|
|
names := storedFileNames(t, env)
|
|
require.Len(t, names, 1)
|
|
|
|
journal := env.journal.String()
|
|
require.Contains(t, journal, msgIntake, "журнал приёма перехвачен")
|
|
|
|
assert.NotContains(t, journal, names[0],
|
|
"имени файла в хранилище в журнале нет: по нему собирается ссылка на скачивание")
|
|
}
|
|
|
|
func TestCreateTranscribeJob_JournalTracesRecord(t *testing.T) {
|
|
env := setupTestEnv(t, readableMetaViewer())
|
|
|
|
content := []byte("содержимое записи")
|
|
req := createMultipartRequest(t, "sample.mp3", content)
|
|
|
|
w := httptest.NewRecorder()
|
|
env.serve(w, req)
|
|
|
|
require.Equal(t, http.StatusCreated, w.Code)
|
|
|
|
var response CreateTranscribeJobResponse
|
|
require.NoError(t, json.Unmarshal(w.Body.Bytes(), &response))
|
|
|
|
job, err := env.handler.jobRepo.GetByID(response.JobID)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, job.FileID)
|
|
|
|
// Отбор по идентификатору **этого** прогона: иначе утверждение прошло бы по
|
|
// строке, оставленной соседней проверкой, и прослеживаемость числилась бы
|
|
// сохранённой при пустом журнале.
|
|
journal := env.journal.String()
|
|
assert.Contains(t, journal, *job.FileID, "по журналу видно, какой файл заведён")
|
|
assert.Contains(t, journal, ".mp3", "расширение принятой записи в журнале остаётся")
|
|
|
|
// Разделитель ключа и значения задаёт обработчик: сегодня текстовый, по
|
|
// конвенции — JSON. Утверждение держится на значении и переживёт замену.
|
|
assert.Regexp(t, fmt.Sprintf(`size["=:\s]+%d`, len(content)), journal,
|
|
"размер принятой записи в байтах в журнале остаётся")
|
|
|
|
// Запись о приёме не сменила адресата: на DEBUG её в боевой настройке не
|
|
// будет вовсе, и разбор постфактум опереться будет не на что. Уровень
|
|
// ищется в строке самого приёма — соседние строки тоже идут на INFO, и
|
|
// поиск по всему журналу не упал бы от понижения этой.
|
|
assert.Regexp(t, `(?m)^.*level["=:\s]+INFO.*`+regexp.QuoteMeta(msgIntake)+`.*$`, journal,
|
|
"строка приёма остаётся на уровне INFO")
|
|
|
|
// И она ровно одна: вторая строка приёма означала бы второй путь заведения
|
|
// задачи мимо общей точки, то есть место, куда запрет не доехал.
|
|
assert.Equal(t, 1, strings.Count(journal, msgIntake),
|
|
"на принятую запись приходится одна журнальная строка приёма")
|
|
}
|
|
|
|
// metricLabelValues собирает значения меток названного семейства метрик из
|
|
// общего реестра процесса. Проверка судит реестр, а не функцию приведения:
|
|
// приведение, снятое в точке употребления, функцию не ломает, а хвост имени
|
|
// уходит на страницу метрик, которая отдаётся без проверки отправителя.
|
|
func metricLabelValues(t *testing.T, family string) []string {
|
|
families, err := prometheus.DefaultGatherer.Gather()
|
|
require.NoError(t, err)
|
|
|
|
var values []string
|
|
for _, mf := range families {
|
|
if mf.GetName() != family {
|
|
continue
|
|
}
|
|
for _, m := range mf.GetMetric() {
|
|
for _, label := range m.GetLabel() {
|
|
values = append(values, label.GetValue())
|
|
}
|
|
}
|
|
}
|
|
return values
|
|
}
|
|
|
|
func TestCreateTranscribeJob_MetricLabelCarriesNoSenderName(t *testing.T) {
|
|
env := setupTestEnv(t, readableMetaViewer())
|
|
|
|
// Хвост после последней точки — это тоже кусок имени, данного отправителем.
|
|
req := createMultipartRequest(t, "sample."+senderNameMarker, []byte("запись"))
|
|
|
|
w := httptest.NewRecorder()
|
|
env.serve(w, req)
|
|
|
|
require.Equal(t, http.StatusCreated, w.Code)
|
|
|
|
values := metricLabelValues(t, "transcriber_input_file_size_bytes")
|
|
require.NotEmpty(t, values, "метрика размера принятой записи заполняется приёмом")
|
|
assert.NotContains(t, values, "."+senderNameMarker,
|
|
"метка метрики не несёт хвоста имени, данного отправителем")
|
|
assert.NotContains(t, values, senderNameMarker,
|
|
"метка метрики не несёт хвоста имени и без ведущей точки")
|
|
assert.Contains(t, values, "other",
|
|
"незнакомое расширение приведено к общему значению")
|
|
|
|
// А в хранилище расширение остаётся пришедшим: приведение сюда не
|
|
// распространяется.
|
|
names := storedFileNames(t, env)
|
|
require.Len(t, names, 1)
|
|
assert.True(t, strings.HasSuffix(names[0], "."+senderNameMarker),
|
|
"имя в хранилище сохраняет пришедшее расширение: %q", names[0])
|
|
}
|
|
|
|
func TestGetTranscribeJobStatus_Success(t *testing.T) {
|
|
env := setupTestEnv(t, readableMetaViewer())
|
|
|
|
job := jobWithFile(t, env)
|
|
|
|
req := httptest.NewRequest("GET", "/api/status/"+job.Id, http.NoBody)
|
|
|
|
w := httptest.NewRecorder()
|
|
env.serve(w, req)
|
|
|
|
require.Equal(t, http.StatusOK, w.Code)
|
|
|
|
var response GetTranscribeJobResponse
|
|
require.NoError(t, json.Unmarshal(w.Body.Bytes(), &response))
|
|
|
|
assert.Equal(t, job.Id, response.JobID)
|
|
assert.Equal(t, entity.StateCreated, response.State)
|
|
assert.NotZero(t, response.CreatedAt)
|
|
}
|
|
|
|
func TestGetTranscribeJobStatus_NoTranscriptionText(t *testing.T) {
|
|
env := setupTestEnv(t, readableMetaViewer())
|
|
|
|
job := jobWithFile(t, env)
|
|
|
|
req := httptest.NewRequest("GET", "/api/status/"+job.Id, http.NoBody)
|
|
|
|
w := httptest.NewRecorder()
|
|
env.serve(w, req)
|
|
|
|
require.Equal(t, http.StatusOK, w.Code)
|
|
|
|
// Судим по сырому JSON: пустая строка на месте отсутствующего текста
|
|
// читается клиентом как «расшифровка пуста», и разобранная структура
|
|
// эти два случая не различает.
|
|
var raw map[string]json.RawMessage
|
|
require.NoError(t, json.Unmarshal(w.Body.Bytes(), &raw))
|
|
|
|
assert.Contains(t, raw, "job_id")
|
|
assert.Contains(t, raw, "status")
|
|
assert.Contains(t, raw, "created_at")
|
|
assert.NotContains(t, raw, "transcription_text")
|
|
}
|
|
|
|
func TestGetTranscribeJobStatus_NotFound(t *testing.T) {
|
|
env := setupTestEnv(t, readableMetaViewer())
|
|
|
|
req := httptest.NewRequest("GET", "/api/status/non-existent-id", http.NoBody)
|
|
|
|
w := httptest.NewRecorder()
|
|
env.serve(w, req)
|
|
|
|
require.Equal(t, http.StatusNotFound, w.Code)
|
|
|
|
var response map[string]string
|
|
require.NoError(t, json.Unmarshal(w.Body.Bytes(), &response))
|
|
|
|
assert.Equal(t, "Job not found", response["error"])
|
|
}
|