Support meeting note upload registration

server/hdl_files_test.go

@@ -1,19 +1,51 @@
 package main
 
 import (
+	"bytes"
 	"encoding/base64"
+	"encoding/json"
+	"io"
+	"mime/multipart"
+	"net/http"
 	"net/http/httptest"
+	"net/url"
 	"testing"
 	"time"
 
 	"github.com/golang/mock/gomock"
 	"github.com/tinode/chat/server/auth"
 	"github.com/tinode/chat/server/auth/mock_auth"
+	"github.com/tinode/chat/server/media"
 	"github.com/tinode/chat/server/store"
 	"github.com/tinode/chat/server/store/mock_store"
 	"github.com/tinode/chat/server/store/types"
 )
 
+type testLargeFileMediaHandler struct {
+	def  *types.FileDef
+	data string
+}
+
+func (h *testLargeFileMediaHandler) Init(string) error { return nil }
+func (h *testLargeFileMediaHandler) Headers(types.TenantID, string, *url.URL, http.Header, bool) (http.Header, int, error) {
+	return nil, 0, nil
+}
+func (h *testLargeFileMediaHandler) Upload(def *types.FileDef, file io.Reader) (string, int64, error) {
+	data, _ := io.ReadAll(file)
+	h.def = def
+	h.data = string(data)
+	def.Location = "/v0/file/s/" + def.Uid().String() + ".txt"
+	def.ETag = "etag"
+	return def.Location, int64(len(data)), nil
+}
+func (h *testLargeFileMediaHandler) Download(types.TenantID, string) (*types.FileDef, media.ReadSeekCloser, error) {
+	return nil, nil, types.ErrUnsupported
+}
+func (h *testLargeFileMediaHandler) Delete([]string) error { return nil }
+func (h *testLargeFileMediaHandler) GetIdFromUrl(string) types.Uid {
+	return types.ZeroUid
+}
+
 func TestAuthFileRequestFromSessionRequiresTenant(t *testing.T) {
 	oldSessionStore := globals.sessionStore
 	globals.sessionStore = NewSessionStore(time.Minute)
@@ -137,3 +169,90 @@ func TestLargeFileGarbageCollectTenantsScopesDeletes(t *testing.T) {
 
 	largeFileGarbageCollectTenants([]types.TenantID{7, 8}, olderThan, 25)
 }
+
+func TestLargeFileReceiveMeetingNoteUploadCreatesNoteFile(t *testing.T) {
+	ctrl := gomock.NewController(t)
+	defer ctrl.Finish()
+
+	oldStore := store.Store
+	oldFiles := store.Files
+	oldMeetingNotes := store.MeetingNotes
+	oldKafkaBus := kafkaBus
+	defer func() {
+		store.Store = oldStore
+		store.Files = oldFiles
+		store.MeetingNotes = oldMeetingNotes
+		kafkaBus = oldKafkaBus
+	}()
+
+	handler := &testLargeFileMediaHandler{}
+	mockStore := mock_store.NewMockPersistentStorageInterface(ctrl)
+	tokenAuth := mock_auth.NewMockAuthHandler(ctrl)
+	files := mock_store.NewMockFilePersistenceInterface(ctrl)
+	notes := &testMeetingNotesStore{}
+	store.Store = mockStore
+	store.Files = files
+	store.MeetingNotes = notes
+	kafkaBus = nil
+
+	fileID := types.Uid(101)
+	mockStore.EXPECT().GetMediaHandler().Return(handler).AnyTimes()
+	mockStore.EXPECT().GetLogicalAuthHandler("token").Return(tokenAuth)
+	mockStore.EXPECT().GetUidString().Return(fileID.String())
+	mockStore.EXPECT().GetUidString().Return("note-1")
+	tokenAuth.EXPECT().Authenticate(auth.AuthContext{RemoteAddr: "192.0.2.1:12345"}, []byte("token-1")).
+		Return(&auth.Rec{TenantID: 7, Uid: 42, AuthLevel: auth.LevelAuth}, nil, nil)
+	files.EXPECT().FinishUpload(types.TenantID(7), gomock.Any(), true, int64(11)).DoAndReturn(
+		func(_ types.TenantID, fd *types.FileDef, _ bool, size int64) (*types.FileDef, error) {
+			fd.Status = types.UploadCompleted
+			fd.Size = size
+			return fd, nil
+		})
+
+	var body bytes.Buffer
+	writer := multipart.NewWriter(&body)
+	if err := writer.WriteField("id", "upload-meeting-note-1"); err != nil {
+		t.Fatal(err)
+	}
+	if err := writer.WriteField("purpose", fileUploadPurposeMeetingNote); err != nil {
+		t.Fatal(err)
+	}
+	if err := writer.WriteField("meeting_id", "mtg_test"); err != nil {
+		t.Fatal(err)
+	}
+	part, err := writer.CreateFormFile("file", "notes.txt")
+	if err != nil {
+		t.Fatal(err)
+	}
+	if _, err := part.Write([]byte("hello notes")); err != nil {
+		t.Fatal(err)
+	}
+	if err := writer.Close(); err != nil {
+		t.Fatal(err)
+	}
+
+	req := httptest.NewRequest(http.MethodPost, "/v0/file/u/", &body)
+	req.RemoteAddr = "192.0.2.1:12345"
+	req.Header.Set("Content-Type", writer.FormDataContentType())
+	req.Header.Set("X-MeetingIM-APIKey", makeTestAPIKey(t))
+	req.Header.Set("X-MeetingIM-Auth", "token "+base64.StdEncoding.EncodeToString([]byte("token-1")))
+	rec := httptest.NewRecorder()
+
+	largeFileReceiveHTTP(rec, req)
+
+	if rec.Code != http.StatusOK {
+		t.Fatalf("status=%d body=%s", rec.Code, rec.Body.String())
+	}
+	var resp ServerComMessage
+	if err := json.Unmarshal(rec.Body.Bytes(), &resp); err != nil {
+		t.Fatalf("decode response: %v body=%s", err, rec.Body.String())
+	}
+	params := resp.Ctrl.Params.(map[string]any)
+	if params["file_id"] != fileID.String() || params["purpose"] != fileUploadPurposeMeetingNote ||
+		params["meeting_id"] != "mtg_test" || params["note_file_id"] != "mnote_note-1" {
+		t.Fatalf("unexpected upload params: %#v", params)
+	}
+	if handler.data != "hello notes" || notes.createdFileID != fileID.String() || notes.createdMeetingID != "mtg_test" || notes.createdIdempotencyKey != "upload-meeting-note-1" {
+		t.Fatalf("unexpected side effects: handler=%#v notes=%#v", handler, notes)
+	}
+}