From 32c5fd987a06e11b14a4247d13187657c14adedd Mon Sep 17 00:00:00 2001
From: Tobi Smethurst <31960611+tsmethurst@users.noreply.github.com>
Date: Mon, 19 Apr 2021 19:42:19 +0200
Subject: Api/v1/statuses (#11)

This PR adds:
Statuses

    New status creation.
    View existing status
    Delete a status
    Fave a status
    Unfave a status
    See who's faved a status

Media

    Upload media attachment and store/retrieve it
    Upload custom emoji and store/retrieve it

Fileserver

    Serve files from storage

Testing

    Test models, testrig -- run a GTS test instance and play around with it.
---
 testrig/util.go | 64 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 64 insertions(+)
 create mode 100644 testrig/util.go

(limited to 'testrig/util.go')

diff --git a/testrig/util.go b/testrig/util.go
new file mode 100644
index 000000000..96a979342
--- /dev/null
+++ b/testrig/util.go
@@ -0,0 +1,64 @@
+/*
+   GoToSocial
+   Copyright (C) 2021 GoToSocial Authors admin@gotosocial.org
+
+   This program is free software: you can redistribute it and/or modify
+   it under the terms of the GNU Affero General Public License as published by
+   the Free Software Foundation, either version 3 of the License, or
+   (at your option) any later version.
+
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU Affero General Public License for more details.
+
+   You should have received a copy of the GNU Affero General Public License
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+package testrig
+
+import (
+	"bytes"
+	"io"
+	"mime/multipart"
+	"os"
+)
+
+// CreateMultipartFormData is a handy function for taking a fieldname and a filename, and creating a multipart form bytes buffer
+// with the file contents set in the given fieldname. The extraFields param can be used to add extra FormFields to the request, as necessary.
+// The returned bytes.Buffer b can be used like so:
+// 	httptest.NewRequest(http.MethodPost, "https://example.org/whateverpath", bytes.NewReader(b.Bytes()))
+// The returned *multipart.Writer w can be used to set the content type of the request, like so:
+// 	req.Header.Set("Content-Type", w.FormDataContentType())
+func CreateMultipartFormData(fieldName string, fileName string, extraFields map[string]string) (bytes.Buffer, *multipart.Writer, error) {
+	var b bytes.Buffer
+	var err error
+	w := multipart.NewWriter(&b)
+	var fw io.Writer
+	file, err := os.Open(fileName)
+	if err != nil {
+		return b, nil, err
+	}
+	if fw, err = w.CreateFormFile(fieldName, file.Name()); err != nil {
+		return b, nil, err
+	}
+	if _, err = io.Copy(fw, file); err != nil {
+		return b, nil, err
+	}
+
+	for k, v := range extraFields {
+		f, err := w.CreateFormField(k)
+		if err != nil {
+			return b, nil, err
+		}
+		if _, err := io.Copy(f, bytes.NewBufferString(v)); err != nil {
+			return b, nil, err
+		}
+	}
+
+	if err := w.Close(); err != nil {
+		return b, nil, err
+	}
+	return b, w, nil
+}
-- 
cgit v1.2.3