X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/c781f79f3c8cb9a7577ee1574d196988da2719b3..c76d3832afce8da932860294d27bd167bf8a928c:/telegram_fake_helper_test.go?ds=sidebyside

diff --git a/telegram_fake_helper_test.go b/telegram_fake_helper_test.go
index e860d65..6ca755e 100644
--- a/telegram_fake_helper_test.go
+++ b/telegram_fake_helper_test.go
@@ -87,69 +87,6 @@ func newTelegramLocalFake(t *testing.T) *TelegramFake {
 	return f
 }
 
-// Add makes update available to getUpdates.
-func (f *TelegramFake) Add(update TelegramUpdate) {
-	f.mutex.Lock()
-
-	f.updates = append(f.updates, update)
-	f.mutex.Unlock()
-
-	select {
-	case f.added <- struct{}{}:
-	default:
-	}
-}
-
-// Calls returns the calls so far.
-func (f *TelegramFake) Calls() []TelegramFakeCall {
-	f.mutex.Lock()
-	defer f.mutex.Unlock()
-
-	return append([]TelegramFakeCall(nil), f.calls...)
-}
-
-// Sent returns the texts of the sendMessage calls and the Markdown of the
-// sendRichMessage calls that succeeded, in order.
-func (f *TelegramFake) Sent() []string {
-	var texts []string
-
-	for _, call := range f.Calls() {
-		switch {
-		case !call.ok:
-		case call.method == "sendMessage":
-			texts = append(texts, call.params["text"].(string))
-		case call.method == "sendRichMessage":
-			texts = append(texts, call.params["rich_message"].(map[string]any)["markdown"].(string))
-		}
-	}
-
-	return texts
-}
-
-func (f *TelegramFake) Image(id string, data []byte) {
-	f.mutex.Lock()
-	defer f.mutex.Unlock()
-
-	if f.images == nil {
-		f.images = make(map[string][]byte)
-	}
-
-	f.images[id] = data
-}
-
-// RoundTrip handles a request in memory, so synctest can control its timers.
-func (f *TelegramFake) RoundTrip(request *http.Request) (*http.Response, error) {
-	response := httptest.NewRecorder()
-	f.serve(response, request)
-
-	ctx := request.Context()
-	if ctx.Err() != nil {
-		return nil, ctx.Err()
-	}
-
-	return response.Result(), nil
-}
-
 func (f *TelegramFake) serve(w http.ResponseWriter, r *http.Request) {
 	id, file := strings.CutPrefix(r.URL.Path, "/file/bot"+telegramTestToken+"/")
 	if file {
@@ -320,6 +257,69 @@ func (f *TelegramFake) from(offset int64) []TelegramUpdate {
 	return updates
 }
 
+// Add makes update available to getUpdates.
+func (f *TelegramFake) Add(update TelegramUpdate) {
+	f.mutex.Lock()
+
+	f.updates = append(f.updates, update)
+	f.mutex.Unlock()
+
+	select {
+	case f.added <- struct{}{}:
+	default:
+	}
+}
+
+// Sent returns the texts of the sendMessage calls and the Markdown of the
+// sendRichMessage calls that succeeded, in order.
+func (f *TelegramFake) Sent() []string {
+	var texts []string
+
+	for _, call := range f.Calls() {
+		switch {
+		case !call.ok:
+		case call.method == "sendMessage":
+			texts = append(texts, call.params["text"].(string))
+		case call.method == "sendRichMessage":
+			texts = append(texts, call.params["rich_message"].(map[string]any)["markdown"].(string))
+		}
+	}
+
+	return texts
+}
+
+// Calls returns the calls so far.
+func (f *TelegramFake) Calls() []TelegramFakeCall {
+	f.mutex.Lock()
+	defer f.mutex.Unlock()
+
+	return append([]TelegramFakeCall(nil), f.calls...)
+}
+
+func (f *TelegramFake) Image(id string, data []byte) {
+	f.mutex.Lock()
+	defer f.mutex.Unlock()
+
+	if f.images == nil {
+		f.images = make(map[string][]byte)
+	}
+
+	f.images[id] = data
+}
+
+// RoundTrip handles a request in memory, so synctest can control its timers.
+func (f *TelegramFake) RoundTrip(request *http.Request) (*http.Response, error) {
+	response := httptest.NewRecorder()
+	f.serve(response, request)
+
+	ctx := request.Context()
+	if ctx.Err() != nil {
+		return nil, ctx.Err()
+	}
+
+	return response.Result(), nil
+}
+
 // telegramOwnerMessage returns a text message of the owner in the private chat.
 func telegramOwnerMessage(text string) *TelegramMessage {
 	return &TelegramMessage{