X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/48d7bce28b18ecef4c52c98821675605464cecb9..80e348724041a694002fc38f57e8868b9b9a2be9:/telegram_fake_helper_test.go

diff --git a/telegram_fake_helper_test.go b/telegram_fake_helper_test.go
index f6ccd84..6082d7d 100644
--- a/telegram_fake_helper_test.go
+++ b/telegram_fake_helper_test.go
@@ -2,8 +2,10 @@ package main
 
 import (
 	"encoding/json"
+	"errors"
 	"fmt"
 	"io"
+	"mime/multipart"
 	"net/http"
 	"net/http/httptest"
 	"strings"
@@ -48,6 +50,42 @@ type TelegramFake struct {
 	images  map[string][]byte
 }
 
+// newTelegramFake starts a fake Bot API server that stops at the end of the
+// test.
+func newTelegramFake(t *testing.T) *TelegramFake {
+	t.Helper()
+
+	f := &TelegramFake{
+		added:   make(chan struct{}, 1),
+		stopped: make(chan struct{}),
+	}
+
+	f.Server = httptest.NewServer(http.HandlerFunc(f.serve))
+	t.Cleanup(f.Close)
+	t.Cleanup(func() {
+		close(f.stopped)
+	})
+
+	return f
+}
+
+// newTelegramLocalFake returns the same fake API without network goroutines.
+// Its RoundTrip handles requests directly for tests with fake time.
+func newTelegramLocalFake(t *testing.T) *TelegramFake {
+	t.Helper()
+
+	f := &TelegramFake{
+		Server:  &httptest.Server{URL: "https://example.com"},
+		added:   make(chan struct{}, 1),
+		stopped: make(chan struct{}),
+	}
+	t.Cleanup(func() {
+		close(f.stopped)
+	})
+
+	return f
+}
+
 // Add makes update available to getUpdates.
 func (f *TelegramFake) Add(update TelegramUpdate) {
 	f.mutex.Lock()
@@ -139,7 +177,38 @@ func (f *TelegramFake) serve(w http.ResponseWriter, r *http.Request) {
 
 	var params map[string]any
 
-	err := json.NewDecoder(r.Body).Decode(&params)
+	var err error
+
+	if strings.HasPrefix(r.Header.Get("Content-Type"), "multipart/form-data;") {
+		err = r.ParseMultipartForm(imageMaxBytes + (1 << 20))
+		if err == nil {
+			params = make(map[string]any)
+			for key, values := range r.MultipartForm.Value {
+				params[key] = values[0]
+			}
+
+			for key, files := range r.MultipartForm.File {
+				var file multipart.File
+
+				file, err = files[0].Open()
+				if err != nil {
+					break
+				}
+
+				params[key], err = io.ReadAll(file)
+				err = errors.Join(err, file.Close())
+				params[key+"_name"] = files[0].Filename
+				if err != nil {
+					break
+				}
+			}
+
+			err = errors.Join(err, r.MultipartForm.RemoveAll())
+		}
+	} else {
+		err = json.NewDecoder(r.Body).Decode(&params)
+	}
+
 	if err != nil {
 		http.Error(w, err.Error(), http.StatusBadRequest)
 
@@ -250,42 +319,6 @@ func (f *TelegramFake) from(offset int64) []TelegramUpdate {
 	return updates
 }
 
-// newTelegramFake starts a fake Bot API server that stops at the end of the
-// test.
-func newTelegramFake(t *testing.T) *TelegramFake {
-	t.Helper()
-
-	f := &TelegramFake{
-		added:   make(chan struct{}, 1),
-		stopped: make(chan struct{}),
-	}
-
-	f.Server = httptest.NewServer(http.HandlerFunc(f.serve))
-	t.Cleanup(f.Close)
-	t.Cleanup(func() {
-		close(f.stopped)
-	})
-
-	return f
-}
-
-// newTelegramLocalFake returns the same fake API without network goroutines.
-// Its RoundTrip handles requests directly for tests with fake time.
-func newTelegramLocalFake(t *testing.T) *TelegramFake {
-	t.Helper()
-
-	f := &TelegramFake{
-		Server:  &httptest.Server{URL: "https://example.com"},
-		added:   make(chan struct{}, 1),
-		stopped: make(chan struct{}),
-	}
-	t.Cleanup(func() {
-		close(f.stopped)
-	})
-
-	return f
-}
-
 // telegramOwnerMessage returns a text message of the owner in the private chat.
 func telegramOwnerMessage(text string) *TelegramMessage {
 	return &TelegramMessage{