| 
									
										
										
										
											2023-03-12 16:00:57 +01:00
										 |  |  | // GoToSocial | 
					
						
							|  |  |  | // Copyright (C) GoToSocial Authors admin@gotosocial.org | 
					
						
							|  |  |  | // SPDX-License-Identifier: AGPL-3.0-or-later | 
					
						
							|  |  |  | // | 
					
						
							|  |  |  | // 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/>. | 
					
						
							| 
									
										
										
										
											2022-01-09 18:41:22 +01:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-01-08 17:17:01 +01:00
										 |  |  | package media_test | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-01-09 18:41:22 +01:00
										 |  |  | import ( | 
					
						
							| 
									
										
										
										
											2022-01-16 18:52:55 +01:00
										 |  |  | 	"bytes" | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | 	"context" | 
					
						
							|  |  |  | 	"fmt" | 
					
						
							| 
									
										
										
										
											2022-01-16 18:52:55 +01:00
										 |  |  | 	"io" | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | 	"os" | 
					
						
							| 
									
										
										
										
											2022-01-29 12:16:12 +01:00
										 |  |  | 	"path" | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | 	"testing" | 
					
						
							| 
									
										
										
										
											2023-01-11 11:13:13 +00:00
										 |  |  | 	"time" | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-11-05 12:10:19 +01:00
										 |  |  | 	"codeberg.org/gruf/go-store/v2/storage" | 
					
						
							| 
									
										
										
										
											2022-01-09 18:41:22 +01:00
										 |  |  | 	"github.com/stretchr/testify/suite" | 
					
						
							| 
									
										
										
										
											2022-03-07 11:08:26 +01:00
										 |  |  | 	gtsmodel "github.com/superseriousbusiness/gotosocial/internal/gtsmodel" | 
					
						
							| 
									
										
										
										
											2022-01-09 18:41:22 +01:00
										 |  |  | 	"github.com/superseriousbusiness/gotosocial/internal/media" | 
					
						
							| 
									
										
										
										
											2023-02-13 18:40:48 +00:00
										 |  |  | 	"github.com/superseriousbusiness/gotosocial/internal/state" | 
					
						
							| 
									
										
										
										
											2022-07-03 12:08:30 +02:00
										 |  |  | 	gtsstorage "github.com/superseriousbusiness/gotosocial/internal/storage" | 
					
						
							| 
									
										
										
										
											2023-05-25 10:37:38 +02:00
										 |  |  | 	"github.com/superseriousbusiness/gotosocial/testrig" | 
					
						
							| 
									
										
										
										
											2022-01-09 18:41:22 +01:00
										 |  |  | ) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | type ManagerTestSuite struct { | 
					
						
							|  |  |  | 	MediaStandardTestSuite | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-09-24 11:11:47 +02:00
										 |  |  | func (suite *ManagerTestSuite) TestEmojiProcessBlocking() { | 
					
						
							|  |  |  | 	ctx := context.Background() | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-11-03 15:03:12 +01:00
										 |  |  | 	data := func(_ context.Context) (io.ReadCloser, int64, error) { | 
					
						
							| 
									
										
										
										
											2022-09-24 11:11:47 +02:00
										 |  |  | 		// load bytes from a test image | 
					
						
							|  |  |  | 		b, err := os.ReadFile("./test/rainbow-original.png") | 
					
						
							|  |  |  | 		if err != nil { | 
					
						
							|  |  |  | 			panic(err) | 
					
						
							|  |  |  | 		} | 
					
						
							| 
									
										
										
										
											2022-11-03 15:03:12 +01:00
										 |  |  | 		return io.NopCloser(bytes.NewBuffer(b)), int64(len(b)), nil | 
					
						
							| 
									
										
										
										
											2022-09-24 11:11:47 +02:00
										 |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	emojiID := "01GDQ9G782X42BAMFASKP64343" | 
					
						
							|  |  |  | 	emojiURI := "http://localhost:8080/emoji/01GDQ9G782X42BAMFASKP64343" | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2023-05-28 13:08:35 +01:00
										 |  |  | 	processingEmoji, err := suite.manager.ProcessEmoji(ctx, data, "rainbow_test", emojiID, emojiURI, nil, false) | 
					
						
							| 
									
										
										
										
											2022-09-24 11:11:47 +02:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// do a blocking call to fetch the emoji | 
					
						
							|  |  |  | 	emoji, err := processingEmoji.LoadEmoji(ctx) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotNil(emoji) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// make sure it's got the stuff set on it that we expect | 
					
						
							|  |  |  | 	suite.Equal(emojiID, emoji.ID) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// file meta should be correctly derived from the image | 
					
						
							|  |  |  | 	suite.Equal("image/png", emoji.ImageContentType) | 
					
						
							|  |  |  | 	suite.Equal("image/png", emoji.ImageStaticContentType) | 
					
						
							|  |  |  | 	suite.Equal(36702, emoji.ImageFileSize) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// now make sure the emoji is in the database | 
					
						
							|  |  |  | 	dbEmoji, err := suite.db.GetEmojiByID(ctx, emojiID) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotNil(dbEmoji) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// make sure the processed emoji file is in storage | 
					
						
							|  |  |  | 	processedFullBytes, err := suite.storage.Get(ctx, emoji.ImagePath) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// load the processed bytes from our test folder, to compare | 
					
						
							|  |  |  | 	processedFullBytesExpected, err := os.ReadFile("./test/rainbow-original.png") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// the bytes in storage should be what we expected | 
					
						
							|  |  |  | 	suite.Equal(processedFullBytesExpected, processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// now do the same for the thumbnail and make sure it's what we expected | 
					
						
							|  |  |  | 	processedStaticBytes, err := suite.storage.Get(ctx, emoji.ImageStaticPath) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedStaticBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	processedStaticBytesExpected, err := os.ReadFile("./test/rainbow-static.png") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedStaticBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	suite.Equal(processedStaticBytesExpected, processedStaticBytes) | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-10-13 15:16:24 +02:00
										 |  |  | func (suite *ManagerTestSuite) TestEmojiProcessBlockingRefresh() { | 
					
						
							|  |  |  | 	ctx := context.Background() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// we're going to 'refresh' the remote 'yell' emoji by changing the image url to the pixellated gts logo | 
					
						
							|  |  |  | 	originalEmoji := suite.testEmojis["yell"] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	emojiToUpdate := >smodel.Emoji{} | 
					
						
							|  |  |  | 	*emojiToUpdate = *originalEmoji | 
					
						
							|  |  |  | 	newImageRemoteURL := "http://fossbros-anonymous.io/some/image/path.png" | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	oldEmojiImagePath := emojiToUpdate.ImagePath | 
					
						
							|  |  |  | 	oldEmojiImageStaticPath := emojiToUpdate.ImageStaticPath | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-11-03 15:03:12 +01:00
										 |  |  | 	data := func(_ context.Context) (io.ReadCloser, int64, error) { | 
					
						
							| 
									
										
										
										
											2022-10-13 15:16:24 +02:00
										 |  |  | 		b, err := os.ReadFile("./test/gts_pixellated-original.png") | 
					
						
							|  |  |  | 		if err != nil { | 
					
						
							|  |  |  | 			panic(err) | 
					
						
							|  |  |  | 		} | 
					
						
							| 
									
										
										
										
											2022-11-03 15:03:12 +01:00
										 |  |  | 		return io.NopCloser(bytes.NewBuffer(b)), int64(len(b)), nil | 
					
						
							| 
									
										
										
										
											2022-10-13 15:16:24 +02:00
										 |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	emojiID := emojiToUpdate.ID | 
					
						
							|  |  |  | 	emojiURI := emojiToUpdate.URI | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2023-05-28 13:08:35 +01:00
										 |  |  | 	processingEmoji, err := suite.manager.ProcessEmoji(ctx, data, "yell", emojiID, emojiURI, &media.AdditionalEmojiInfo{ | 
					
						
							| 
									
										
										
										
											2022-10-13 15:16:24 +02:00
										 |  |  | 		CreatedAt:      &emojiToUpdate.CreatedAt, | 
					
						
							|  |  |  | 		Domain:         &emojiToUpdate.Domain, | 
					
						
							|  |  |  | 		ImageRemoteURL: &newImageRemoteURL, | 
					
						
							|  |  |  | 	}, true) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// do a blocking call to fetch the emoji | 
					
						
							|  |  |  | 	emoji, err := processingEmoji.LoadEmoji(ctx) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotNil(emoji) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// make sure it's got the stuff set on it that we expect | 
					
						
							|  |  |  | 	suite.Equal(emojiID, emoji.ID) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// file meta should be correctly derived from the image | 
					
						
							|  |  |  | 	suite.Equal("image/png", emoji.ImageContentType) | 
					
						
							|  |  |  | 	suite.Equal("image/png", emoji.ImageStaticContentType) | 
					
						
							|  |  |  | 	suite.Equal(10296, emoji.ImageFileSize) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// now make sure the emoji is in the database | 
					
						
							|  |  |  | 	dbEmoji, err := suite.db.GetEmojiByID(ctx, emojiID) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotNil(dbEmoji) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// make sure the processed emoji file is in storage | 
					
						
							|  |  |  | 	processedFullBytes, err := suite.storage.Get(ctx, emoji.ImagePath) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// load the processed bytes from our test folder, to compare | 
					
						
							|  |  |  | 	processedFullBytesExpected, err := os.ReadFile("./test/gts_pixellated-original.png") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// the bytes in storage should be what we expected | 
					
						
							|  |  |  | 	suite.Equal(processedFullBytesExpected, processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// now do the same for the thumbnail and make sure it's what we expected | 
					
						
							|  |  |  | 	processedStaticBytes, err := suite.storage.Get(ctx, emoji.ImageStaticPath) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedStaticBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	processedStaticBytesExpected, err := os.ReadFile("./test/gts_pixellated-static.png") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedStaticBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	suite.Equal(processedStaticBytesExpected, processedStaticBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// most fields should be different on the emoji now from what they were before | 
					
						
							|  |  |  | 	suite.Equal(originalEmoji.ID, dbEmoji.ID) | 
					
						
							|  |  |  | 	suite.NotEqual(originalEmoji.ImageRemoteURL, dbEmoji.ImageRemoteURL) | 
					
						
							|  |  |  | 	suite.NotEqual(originalEmoji.ImageURL, dbEmoji.ImageURL) | 
					
						
							|  |  |  | 	suite.NotEqual(originalEmoji.ImageStaticURL, dbEmoji.ImageStaticURL) | 
					
						
							|  |  |  | 	suite.NotEqual(originalEmoji.ImageFileSize, dbEmoji.ImageFileSize) | 
					
						
							|  |  |  | 	suite.NotEqual(originalEmoji.ImageStaticFileSize, dbEmoji.ImageStaticFileSize) | 
					
						
							|  |  |  | 	suite.NotEqual(originalEmoji.ImagePath, dbEmoji.ImagePath) | 
					
						
							|  |  |  | 	suite.NotEqual(originalEmoji.ImageStaticPath, dbEmoji.ImageStaticPath) | 
					
						
							|  |  |  | 	suite.NotEqual(originalEmoji.ImageStaticPath, dbEmoji.ImageStaticPath) | 
					
						
							|  |  |  | 	suite.NotEqual(originalEmoji.UpdatedAt, dbEmoji.UpdatedAt) | 
					
						
							|  |  |  | 	suite.NotEqual(originalEmoji.ImageUpdatedAt, dbEmoji.ImageUpdatedAt) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// the old image files should no longer be in storage | 
					
						
							|  |  |  | 	_, err = suite.storage.Get(ctx, oldEmojiImagePath) | 
					
						
							|  |  |  | 	suite.ErrorIs(err, storage.ErrNotFound) | 
					
						
							|  |  |  | 	_, err = suite.storage.Get(ctx, oldEmojiImageStaticPath) | 
					
						
							|  |  |  | 	suite.ErrorIs(err, storage.ErrNotFound) | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-09-24 11:11:47 +02:00
										 |  |  | func (suite *ManagerTestSuite) TestEmojiProcessBlockingTooLarge() { | 
					
						
							|  |  |  | 	ctx := context.Background() | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-11-03 15:03:12 +01:00
										 |  |  | 	data := func(_ context.Context) (io.ReadCloser, int64, error) { | 
					
						
							| 
									
										
										
										
											2022-09-24 11:11:47 +02:00
										 |  |  | 		// load bytes from a test image | 
					
						
							|  |  |  | 		b, err := os.ReadFile("./test/big-panda.gif") | 
					
						
							|  |  |  | 		if err != nil { | 
					
						
							|  |  |  | 			panic(err) | 
					
						
							|  |  |  | 		} | 
					
						
							| 
									
										
										
										
											2022-11-03 15:03:12 +01:00
										 |  |  | 		return io.NopCloser(bytes.NewBuffer(b)), int64(len(b)), nil | 
					
						
							| 
									
										
										
										
											2022-09-24 11:11:47 +02:00
										 |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	emojiID := "01GDQ9G782X42BAMFASKP64343" | 
					
						
							|  |  |  | 	emojiURI := "http://localhost:8080/emoji/01GDQ9G782X42BAMFASKP64343" | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2023-05-28 13:08:35 +01:00
										 |  |  | 	processingEmoji, err := suite.manager.ProcessEmoji(ctx, data, "big_panda", emojiID, emojiURI, nil, false) | 
					
						
							| 
									
										
										
										
											2022-09-24 11:11:47 +02:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// do a blocking call to fetch the emoji | 
					
						
							|  |  |  | 	emoji, err := processingEmoji.LoadEmoji(ctx) | 
					
						
							| 
									
										
										
										
											2023-06-22 20:46:36 +01:00
										 |  |  | 	suite.EqualError(err, "store: given emoji size 630kiB greater than max allowed 50.0kiB") | 
					
						
							| 
									
										
										
										
											2022-09-24 11:11:47 +02:00
										 |  |  | 	suite.Nil(emoji) | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | func (suite *ManagerTestSuite) TestEmojiProcessBlockingTooLargeNoSizeGiven() { | 
					
						
							|  |  |  | 	ctx := context.Background() | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-11-03 15:03:12 +01:00
										 |  |  | 	data := func(_ context.Context) (io.ReadCloser, int64, error) { | 
					
						
							| 
									
										
										
										
											2022-09-24 11:11:47 +02:00
										 |  |  | 		// load bytes from a test image | 
					
						
							|  |  |  | 		b, err := os.ReadFile("./test/big-panda.gif") | 
					
						
							|  |  |  | 		if err != nil { | 
					
						
							|  |  |  | 			panic(err) | 
					
						
							|  |  |  | 		} | 
					
						
							| 
									
										
										
										
											2023-01-11 11:13:13 +00:00
										 |  |  | 		return io.NopCloser(bytes.NewBuffer(b)), -1, nil | 
					
						
							| 
									
										
										
										
											2022-09-24 11:11:47 +02:00
										 |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	emojiID := "01GDQ9G782X42BAMFASKP64343" | 
					
						
							|  |  |  | 	emojiURI := "http://localhost:8080/emoji/01GDQ9G782X42BAMFASKP64343" | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2023-05-28 13:08:35 +01:00
										 |  |  | 	processingEmoji, err := suite.manager.ProcessEmoji(ctx, data, "big_panda", emojiID, emojiURI, nil, false) | 
					
						
							| 
									
										
										
										
											2022-09-24 11:11:47 +02:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// do a blocking call to fetch the emoji | 
					
						
							|  |  |  | 	emoji, err := processingEmoji.LoadEmoji(ctx) | 
					
						
							| 
									
										
										
										
											2023-06-22 20:46:36 +01:00
										 |  |  | 	suite.EqualError(err, "store: calculated emoji size 630kiB greater than max allowed 50.0kiB") | 
					
						
							| 
									
										
										
										
											2022-09-24 11:11:47 +02:00
										 |  |  | 	suite.Nil(emoji) | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | func (suite *ManagerTestSuite) TestEmojiProcessBlockingNoFileSizeGiven() { | 
					
						
							|  |  |  | 	ctx := context.Background() | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-11-03 15:03:12 +01:00
										 |  |  | 	data := func(_ context.Context) (io.ReadCloser, int64, error) { | 
					
						
							| 
									
										
										
										
											2022-09-24 11:11:47 +02:00
										 |  |  | 		// load bytes from a test image | 
					
						
							|  |  |  | 		b, err := os.ReadFile("./test/rainbow-original.png") | 
					
						
							|  |  |  | 		if err != nil { | 
					
						
							|  |  |  | 			panic(err) | 
					
						
							|  |  |  | 		} | 
					
						
							| 
									
										
										
										
											2022-11-03 15:03:12 +01:00
										 |  |  | 		return io.NopCloser(bytes.NewBuffer(b)), -1, nil | 
					
						
							| 
									
										
										
										
											2022-09-24 11:11:47 +02:00
										 |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	emojiID := "01GDQ9G782X42BAMFASKP64343" | 
					
						
							|  |  |  | 	emojiURI := "http://localhost:8080/emoji/01GDQ9G782X42BAMFASKP64343" | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// process the media with no additional info provided | 
					
						
							| 
									
										
										
										
											2023-05-28 13:08:35 +01:00
										 |  |  | 	processingEmoji, err := suite.manager.ProcessEmoji(ctx, data, "rainbow_test", emojiID, emojiURI, nil, false) | 
					
						
							| 
									
										
										
										
											2022-09-24 11:11:47 +02:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// do a blocking call to fetch the emoji | 
					
						
							|  |  |  | 	emoji, err := processingEmoji.LoadEmoji(ctx) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotNil(emoji) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// make sure it's got the stuff set on it that we expect | 
					
						
							|  |  |  | 	suite.Equal(emojiID, emoji.ID) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// file meta should be correctly derived from the image | 
					
						
							|  |  |  | 	suite.Equal("image/png", emoji.ImageContentType) | 
					
						
							|  |  |  | 	suite.Equal("image/png", emoji.ImageStaticContentType) | 
					
						
							|  |  |  | 	suite.Equal(36702, emoji.ImageFileSize) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// now make sure the emoji is in the database | 
					
						
							|  |  |  | 	dbEmoji, err := suite.db.GetEmojiByID(ctx, emojiID) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotNil(dbEmoji) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// make sure the processed emoji file is in storage | 
					
						
							|  |  |  | 	processedFullBytes, err := suite.storage.Get(ctx, emoji.ImagePath) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// load the processed bytes from our test folder, to compare | 
					
						
							|  |  |  | 	processedFullBytesExpected, err := os.ReadFile("./test/rainbow-original.png") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// the bytes in storage should be what we expected | 
					
						
							|  |  |  | 	suite.Equal(processedFullBytesExpected, processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// now do the same for the thumbnail and make sure it's what we expected | 
					
						
							|  |  |  | 	processedStaticBytes, err := suite.storage.Get(ctx, emoji.ImageStaticPath) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedStaticBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	processedStaticBytesExpected, err := os.ReadFile("./test/rainbow-static.png") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedStaticBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	suite.Equal(processedStaticBytesExpected, processedStaticBytes) | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | func (suite *ManagerTestSuite) TestSimpleJpegProcessBlocking() { | 
					
						
							|  |  |  | 	ctx := context.Background() | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-11-03 15:03:12 +01:00
										 |  |  | 	data := func(_ context.Context) (io.ReadCloser, int64, error) { | 
					
						
							| 
									
										
										
										
											2022-01-11 17:49:14 +01:00
										 |  |  | 		// load bytes from a test image | 
					
						
							| 
									
										
										
										
											2022-01-16 18:52:55 +01:00
										 |  |  | 		b, err := os.ReadFile("./test/test-jpeg.jpg") | 
					
						
							|  |  |  | 		if err != nil { | 
					
						
							|  |  |  | 			panic(err) | 
					
						
							|  |  |  | 		} | 
					
						
							| 
									
										
										
										
											2022-11-03 15:03:12 +01:00
										 |  |  | 		return io.NopCloser(bytes.NewBuffer(b)), int64(len(b)), nil | 
					
						
							| 
									
										
										
										
											2022-01-11 17:49:14 +01:00
										 |  |  | 	} | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	accountID := "01FS1X72SK9ZPW0J1QQ68BD264" | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// process the media with no additional info provided | 
					
						
							| 
									
										
										
										
											2023-05-28 13:08:35 +01:00
										 |  |  | 	processingMedia, err := suite.manager.ProcessMedia(ctx, data, accountID, nil) | 
					
						
							| 
									
										
										
										
											2022-09-24 11:11:47 +02:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	// fetch the attachment id from the processing media | 
					
						
							|  |  |  | 	attachmentID := processingMedia.AttachmentID() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// do a blocking call to fetch the attachment | 
					
						
							|  |  |  | 	attachment, err := processingMedia.LoadAttachment(ctx) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotNil(attachment) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// make sure it's got the stuff set on it that we expect | 
					
						
							|  |  |  | 	// the attachment ID and accountID we expect | 
					
						
							|  |  |  | 	suite.Equal(attachmentID, attachment.ID) | 
					
						
							|  |  |  | 	suite.Equal(accountID, attachment.AccountID) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// file meta should be correctly derived from the image | 
					
						
							|  |  |  | 	suite.EqualValues(gtsmodel.Original{ | 
					
						
							|  |  |  | 		Width: 1920, Height: 1080, Size: 2073600, Aspect: 1.7777777777777777, | 
					
						
							|  |  |  | 	}, attachment.FileMeta.Original) | 
					
						
							|  |  |  | 	suite.EqualValues(gtsmodel.Small{ | 
					
						
							|  |  |  | 		Width: 512, Height: 288, Size: 147456, Aspect: 1.7777777777777777, | 
					
						
							|  |  |  | 	}, attachment.FileMeta.Small) | 
					
						
							|  |  |  | 	suite.Equal("image/jpeg", attachment.File.ContentType) | 
					
						
							|  |  |  | 	suite.Equal("image/jpeg", attachment.Thumbnail.ContentType) | 
					
						
							|  |  |  | 	suite.Equal(269739, attachment.File.FileSize) | 
					
						
							|  |  |  | 	suite.Equal("LiBzRk#6V[WF_NvzV@WY_3rqV@a$", attachment.Blurhash) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// now make sure the attachment is in the database | 
					
						
							|  |  |  | 	dbAttachment, err := suite.db.GetAttachmentByID(ctx, attachmentID) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotNil(dbAttachment) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// make sure the processed file is in storage | 
					
						
							|  |  |  | 	processedFullBytes, err := suite.storage.Get(ctx, attachment.File.Path) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// load the processed bytes from our test folder, to compare | 
					
						
							|  |  |  | 	processedFullBytesExpected, err := os.ReadFile("./test/test-jpeg-processed.jpg") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// the bytes in storage should be what we expected | 
					
						
							|  |  |  | 	suite.Equal(processedFullBytesExpected, processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// now do the same for the thumbnail and make sure it's what we expected | 
					
						
							|  |  |  | 	processedThumbnailBytes, err := suite.storage.Get(ctx, attachment.Thumbnail.Path) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedThumbnailBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	processedThumbnailBytesExpected, err := os.ReadFile("./test/test-jpeg-thumbnail.jpg") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedThumbnailBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	suite.Equal(processedThumbnailBytesExpected, processedThumbnailBytes) | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-12-17 05:38:56 +01:00
										 |  |  | func (suite *ManagerTestSuite) TestSlothVineProcessBlocking() { | 
					
						
							|  |  |  | 	ctx := context.Background() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	data := func(_ context.Context) (io.ReadCloser, int64, error) { | 
					
						
							|  |  |  | 		// load bytes from a test video | 
					
						
							|  |  |  | 		b, err := os.ReadFile("./test/test-mp4-original.mp4") | 
					
						
							|  |  |  | 		if err != nil { | 
					
						
							|  |  |  | 			panic(err) | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		return io.NopCloser(bytes.NewBuffer(b)), int64(len(b)), nil | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	accountID := "01FS1X72SK9ZPW0J1QQ68BD264" | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// process the media with no additional info provided | 
					
						
							| 
									
										
										
										
											2023-05-28 13:08:35 +01:00
										 |  |  | 	processingMedia, err := suite.manager.ProcessMedia(ctx, data, accountID, nil) | 
					
						
							| 
									
										
										
										
											2022-12-17 05:38:56 +01:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	// fetch the attachment id from the processing media | 
					
						
							|  |  |  | 	attachmentID := processingMedia.AttachmentID() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// do a blocking call to fetch the attachment | 
					
						
							|  |  |  | 	attachment, err := processingMedia.LoadAttachment(ctx) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotNil(attachment) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// make sure it's got the stuff set on it that we expect | 
					
						
							|  |  |  | 	// the attachment ID and accountID we expect | 
					
						
							|  |  |  | 	suite.Equal(attachmentID, attachment.ID) | 
					
						
							|  |  |  | 	suite.Equal(accountID, attachment.AccountID) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// file meta should be correctly derived from the video | 
					
						
							| 
									
										
										
										
											2022-12-22 11:48:28 +01:00
										 |  |  | 	suite.Equal(338, attachment.FileMeta.Original.Width) | 
					
						
							|  |  |  | 	suite.Equal(240, attachment.FileMeta.Original.Height) | 
					
						
							|  |  |  | 	suite.Equal(81120, attachment.FileMeta.Original.Size) | 
					
						
							|  |  |  | 	suite.EqualValues(1.4083333, attachment.FileMeta.Original.Aspect) | 
					
						
							| 
									
										
										
										
											2023-01-16 16:19:17 +01:00
										 |  |  | 	suite.EqualValues(6.640907, *attachment.FileMeta.Original.Duration) | 
					
						
							| 
									
										
										
										
											2022-12-22 11:48:28 +01:00
										 |  |  | 	suite.EqualValues(29.000029, *attachment.FileMeta.Original.Framerate) | 
					
						
							| 
									
										
										
										
											2023-01-16 16:19:17 +01:00
										 |  |  | 	suite.EqualValues(0x59e74, *attachment.FileMeta.Original.Bitrate) | 
					
						
							| 
									
										
										
										
											2022-12-17 05:38:56 +01:00
										 |  |  | 	suite.EqualValues(gtsmodel.Small{ | 
					
						
							|  |  |  | 		Width: 338, Height: 240, Size: 81120, Aspect: 1.4083333333333334, | 
					
						
							|  |  |  | 	}, attachment.FileMeta.Small) | 
					
						
							|  |  |  | 	suite.Equal("video/mp4", attachment.File.ContentType) | 
					
						
							|  |  |  | 	suite.Equal("image/jpeg", attachment.Thumbnail.ContentType) | 
					
						
							|  |  |  | 	suite.Equal(312413, attachment.File.FileSize) | 
					
						
							| 
									
										
										
										
											2023-01-11 11:13:13 +00:00
										 |  |  | 	suite.Equal("L00000fQfQfQfQfQfQfQfQfQfQfQ", attachment.Blurhash) | 
					
						
							| 
									
										
										
										
											2022-12-17 05:38:56 +01:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	// now make sure the attachment is in the database | 
					
						
							|  |  |  | 	dbAttachment, err := suite.db.GetAttachmentByID(ctx, attachmentID) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotNil(dbAttachment) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// make sure the processed file is in storage | 
					
						
							|  |  |  | 	processedFullBytes, err := suite.storage.Get(ctx, attachment.File.Path) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// load the processed bytes from our test folder, to compare | 
					
						
							|  |  |  | 	processedFullBytesExpected, err := os.ReadFile("./test/test-mp4-processed.mp4") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// the bytes in storage should be what we expected | 
					
						
							|  |  |  | 	suite.Equal(processedFullBytesExpected, processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// now do the same for the thumbnail and make sure it's what we expected | 
					
						
							|  |  |  | 	processedThumbnailBytes, err := suite.storage.Get(ctx, attachment.Thumbnail.Path) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedThumbnailBytes) | 
					
						
							| 
									
										
										
										
											2022-12-21 11:17:43 +01:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-12-17 05:38:56 +01:00
										 |  |  | 	processedThumbnailBytesExpected, err := os.ReadFile("./test/test-mp4-thumbnail.jpg") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedThumbnailBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	suite.Equal(processedThumbnailBytesExpected, processedThumbnailBytes) | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-12-22 11:48:28 +01:00
										 |  |  | func (suite *ManagerTestSuite) TestLongerMp4ProcessBlocking() { | 
					
						
							|  |  |  | 	ctx := context.Background() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	data := func(_ context.Context) (io.ReadCloser, int64, error) { | 
					
						
							|  |  |  | 		// load bytes from a test video | 
					
						
							|  |  |  | 		b, err := os.ReadFile("./test/longer-mp4-original.mp4") | 
					
						
							|  |  |  | 		if err != nil { | 
					
						
							|  |  |  | 			panic(err) | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		return io.NopCloser(bytes.NewBuffer(b)), int64(len(b)), nil | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	accountID := "01FS1X72SK9ZPW0J1QQ68BD264" | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// process the media with no additional info provided | 
					
						
							| 
									
										
										
										
											2023-05-28 13:08:35 +01:00
										 |  |  | 	processingMedia, err := suite.manager.ProcessMedia(ctx, data, accountID, nil) | 
					
						
							| 
									
										
										
										
											2022-12-22 11:48:28 +01:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	// fetch the attachment id from the processing media | 
					
						
							|  |  |  | 	attachmentID := processingMedia.AttachmentID() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// do a blocking call to fetch the attachment | 
					
						
							|  |  |  | 	attachment, err := processingMedia.LoadAttachment(ctx) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotNil(attachment) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// make sure it's got the stuff set on it that we expect | 
					
						
							|  |  |  | 	// the attachment ID and accountID we expect | 
					
						
							|  |  |  | 	suite.Equal(attachmentID, attachment.ID) | 
					
						
							|  |  |  | 	suite.Equal(accountID, attachment.AccountID) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// file meta should be correctly derived from the video | 
					
						
							|  |  |  | 	suite.Equal(600, attachment.FileMeta.Original.Width) | 
					
						
							|  |  |  | 	suite.Equal(330, attachment.FileMeta.Original.Height) | 
					
						
							|  |  |  | 	suite.Equal(198000, attachment.FileMeta.Original.Size) | 
					
						
							|  |  |  | 	suite.EqualValues(1.8181819, attachment.FileMeta.Original.Aspect) | 
					
						
							|  |  |  | 	suite.EqualValues(16.6, *attachment.FileMeta.Original.Duration) | 
					
						
							|  |  |  | 	suite.EqualValues(10, *attachment.FileMeta.Original.Framerate) | 
					
						
							|  |  |  | 	suite.EqualValues(0xc8fb, *attachment.FileMeta.Original.Bitrate) | 
					
						
							|  |  |  | 	suite.EqualValues(gtsmodel.Small{ | 
					
						
							| 
									
										
										
										
											2023-01-11 11:13:13 +00:00
										 |  |  | 		Width: 512, Height: 281, Size: 143872, Aspect: 1.822064, | 
					
						
							| 
									
										
										
										
											2022-12-22 11:48:28 +01:00
										 |  |  | 	}, attachment.FileMeta.Small) | 
					
						
							|  |  |  | 	suite.Equal("video/mp4", attachment.File.ContentType) | 
					
						
							|  |  |  | 	suite.Equal("image/jpeg", attachment.Thumbnail.ContentType) | 
					
						
							|  |  |  | 	suite.Equal(109549, attachment.File.FileSize) | 
					
						
							| 
									
										
										
										
											2023-01-11 11:13:13 +00:00
										 |  |  | 	suite.Equal("L00000fQfQfQfQfQfQfQfQfQfQfQ", attachment.Blurhash) | 
					
						
							| 
									
										
										
										
											2022-12-22 11:48:28 +01:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	// now make sure the attachment is in the database | 
					
						
							|  |  |  | 	dbAttachment, err := suite.db.GetAttachmentByID(ctx, attachmentID) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotNil(dbAttachment) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// make sure the processed file is in storage | 
					
						
							|  |  |  | 	processedFullBytes, err := suite.storage.Get(ctx, attachment.File.Path) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// load the processed bytes from our test folder, to compare | 
					
						
							|  |  |  | 	processedFullBytesExpected, err := os.ReadFile("./test/longer-mp4-processed.mp4") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// the bytes in storage should be what we expected | 
					
						
							|  |  |  | 	suite.Equal(processedFullBytesExpected, processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// now do the same for the thumbnail and make sure it's what we expected | 
					
						
							|  |  |  | 	processedThumbnailBytes, err := suite.storage.Get(ctx, attachment.Thumbnail.Path) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedThumbnailBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	processedThumbnailBytesExpected, err := os.ReadFile("./test/longer-mp4-thumbnail.jpg") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedThumbnailBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	suite.Equal(processedThumbnailBytesExpected, processedThumbnailBytes) | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2023-01-16 16:19:17 +01:00
										 |  |  | func (suite *ManagerTestSuite) TestBirdnestMp4ProcessBlocking() { | 
					
						
							|  |  |  | 	ctx := context.Background() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	data := func(_ context.Context) (io.ReadCloser, int64, error) { | 
					
						
							|  |  |  | 		// load bytes from a test video | 
					
						
							|  |  |  | 		b, err := os.ReadFile("./test/birdnest-original.mp4") | 
					
						
							|  |  |  | 		if err != nil { | 
					
						
							|  |  |  | 			panic(err) | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		return io.NopCloser(bytes.NewBuffer(b)), int64(len(b)), nil | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	accountID := "01FS1X72SK9ZPW0J1QQ68BD264" | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// process the media with no additional info provided | 
					
						
							| 
									
										
										
										
											2023-05-28 13:08:35 +01:00
										 |  |  | 	processingMedia, err := suite.manager.ProcessMedia(ctx, data, accountID, nil) | 
					
						
							| 
									
										
										
										
											2023-01-16 16:19:17 +01:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	// fetch the attachment id from the processing media | 
					
						
							|  |  |  | 	attachmentID := processingMedia.AttachmentID() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// do a blocking call to fetch the attachment | 
					
						
							|  |  |  | 	attachment, err := processingMedia.LoadAttachment(ctx) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotNil(attachment) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// make sure it's got the stuff set on it that we expect | 
					
						
							|  |  |  | 	// the attachment ID and accountID we expect | 
					
						
							|  |  |  | 	suite.Equal(attachmentID, attachment.ID) | 
					
						
							|  |  |  | 	suite.Equal(accountID, attachment.AccountID) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// file meta should be correctly derived from the video | 
					
						
							|  |  |  | 	suite.Equal(404, attachment.FileMeta.Original.Width) | 
					
						
							|  |  |  | 	suite.Equal(720, attachment.FileMeta.Original.Height) | 
					
						
							|  |  |  | 	suite.Equal(290880, attachment.FileMeta.Original.Size) | 
					
						
							|  |  |  | 	suite.EqualValues(0.5611111, attachment.FileMeta.Original.Aspect) | 
					
						
							|  |  |  | 	suite.EqualValues(9.822041, *attachment.FileMeta.Original.Duration) | 
					
						
							|  |  |  | 	suite.EqualValues(30, *attachment.FileMeta.Original.Framerate) | 
					
						
							|  |  |  | 	suite.EqualValues(0x117c79, *attachment.FileMeta.Original.Bitrate) | 
					
						
							|  |  |  | 	suite.EqualValues(gtsmodel.Small{ | 
					
						
							|  |  |  | 		Width: 287, Height: 512, Size: 146944, Aspect: 0.5605469, | 
					
						
							|  |  |  | 	}, attachment.FileMeta.Small) | 
					
						
							|  |  |  | 	suite.Equal("video/mp4", attachment.File.ContentType) | 
					
						
							|  |  |  | 	suite.Equal("image/jpeg", attachment.Thumbnail.ContentType) | 
					
						
							|  |  |  | 	suite.Equal(1409577, attachment.File.FileSize) | 
					
						
							|  |  |  | 	suite.Equal("L00000fQfQfQfQfQfQfQfQfQfQfQ", attachment.Blurhash) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// now make sure the attachment is in the database | 
					
						
							|  |  |  | 	dbAttachment, err := suite.db.GetAttachmentByID(ctx, attachmentID) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotNil(dbAttachment) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// make sure the processed file is in storage | 
					
						
							|  |  |  | 	processedFullBytes, err := suite.storage.Get(ctx, attachment.File.Path) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// load the processed bytes from our test folder, to compare | 
					
						
							|  |  |  | 	processedFullBytesExpected, err := os.ReadFile("./test/birdnest-processed.mp4") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// the bytes in storage should be what we expected | 
					
						
							|  |  |  | 	suite.Equal(processedFullBytesExpected, processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// now do the same for the thumbnail and make sure it's what we expected | 
					
						
							|  |  |  | 	processedThumbnailBytes, err := suite.storage.Get(ctx, attachment.Thumbnail.Path) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedThumbnailBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	processedThumbnailBytesExpected, err := os.ReadFile("./test/birdnest-thumbnail.jpg") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedThumbnailBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	suite.Equal(processedThumbnailBytesExpected, processedThumbnailBytes) | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-12-22 11:48:28 +01:00
										 |  |  | func (suite *ManagerTestSuite) TestNotAnMp4ProcessBlocking() { | 
					
						
							|  |  |  | 	// try to load an 'mp4' that's actually an mkv in disguise | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	ctx := context.Background() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	data := func(_ context.Context) (io.ReadCloser, int64, error) { | 
					
						
							|  |  |  | 		// load bytes from a test video | 
					
						
							|  |  |  | 		b, err := os.ReadFile("./test/not-an.mp4") | 
					
						
							|  |  |  | 		if err != nil { | 
					
						
							|  |  |  | 			panic(err) | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		return io.NopCloser(bytes.NewBuffer(b)), int64(len(b)), nil | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	accountID := "01FS1X72SK9ZPW0J1QQ68BD264" | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// pre processing should go fine but... | 
					
						
							| 
									
										
										
										
											2023-05-28 13:08:35 +01:00
										 |  |  | 	processingMedia, err := suite.manager.ProcessMedia(ctx, data, accountID, nil) | 
					
						
							| 
									
										
										
										
											2022-12-22 11:48:28 +01:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// we should get an error while loading | 
					
						
							|  |  |  | 	attachment, err := processingMedia.LoadAttachment(ctx) | 
					
						
							| 
									
										
										
										
											2023-06-22 20:46:36 +01:00
										 |  |  | 	suite.EqualError(err, "finish: error decoding video: error determining video metadata: [width height framerate]") | 
					
						
							| 
									
										
										
										
											2022-12-22 11:48:28 +01:00
										 |  |  | 	suite.Nil(attachment) | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-09-24 11:11:47 +02:00
										 |  |  | func (suite *ManagerTestSuite) TestSimpleJpegProcessBlockingNoContentLengthGiven() { | 
					
						
							|  |  |  | 	ctx := context.Background() | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-11-03 15:03:12 +01:00
										 |  |  | 	data := func(_ context.Context) (io.ReadCloser, int64, error) { | 
					
						
							| 
									
										
										
										
											2022-09-24 11:11:47 +02:00
										 |  |  | 		// load bytes from a test image | 
					
						
							|  |  |  | 		b, err := os.ReadFile("./test/test-jpeg.jpg") | 
					
						
							|  |  |  | 		if err != nil { | 
					
						
							|  |  |  | 			panic(err) | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		// give length as -1 to indicate unknown | 
					
						
							| 
									
										
										
										
											2022-11-03 15:03:12 +01:00
										 |  |  | 		return io.NopCloser(bytes.NewBuffer(b)), -1, nil | 
					
						
							| 
									
										
										
										
											2022-09-24 11:11:47 +02:00
										 |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	accountID := "01FS1X72SK9ZPW0J1QQ68BD264" | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// process the media with no additional info provided | 
					
						
							| 
									
										
										
										
											2023-05-28 13:08:35 +01:00
										 |  |  | 	processingMedia, err := suite.manager.ProcessMedia(ctx, data, accountID, nil) | 
					
						
							| 
									
										
										
										
											2022-09-24 11:11:47 +02:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	// fetch the attachment id from the processing media | 
					
						
							|  |  |  | 	attachmentID := processingMedia.AttachmentID() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// do a blocking call to fetch the attachment | 
					
						
							|  |  |  | 	attachment, err := processingMedia.LoadAttachment(ctx) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotNil(attachment) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// make sure it's got the stuff set on it that we expect | 
					
						
							|  |  |  | 	// the attachment ID and accountID we expect | 
					
						
							|  |  |  | 	suite.Equal(attachmentID, attachment.ID) | 
					
						
							|  |  |  | 	suite.Equal(accountID, attachment.AccountID) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// file meta should be correctly derived from the image | 
					
						
							|  |  |  | 	suite.EqualValues(gtsmodel.Original{ | 
					
						
							|  |  |  | 		Width: 1920, Height: 1080, Size: 2073600, Aspect: 1.7777777777777777, | 
					
						
							|  |  |  | 	}, attachment.FileMeta.Original) | 
					
						
							|  |  |  | 	suite.EqualValues(gtsmodel.Small{ | 
					
						
							|  |  |  | 		Width: 512, Height: 288, Size: 147456, Aspect: 1.7777777777777777, | 
					
						
							|  |  |  | 	}, attachment.FileMeta.Small) | 
					
						
							|  |  |  | 	suite.Equal("image/jpeg", attachment.File.ContentType) | 
					
						
							|  |  |  | 	suite.Equal("image/jpeg", attachment.Thumbnail.ContentType) | 
					
						
							|  |  |  | 	suite.Equal(269739, attachment.File.FileSize) | 
					
						
							|  |  |  | 	suite.Equal("LiBzRk#6V[WF_NvzV@WY_3rqV@a$", attachment.Blurhash) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// now make sure the attachment is in the database | 
					
						
							|  |  |  | 	dbAttachment, err := suite.db.GetAttachmentByID(ctx, attachmentID) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotNil(dbAttachment) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// make sure the processed file is in storage | 
					
						
							|  |  |  | 	processedFullBytes, err := suite.storage.Get(ctx, attachment.File.Path) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// load the processed bytes from our test folder, to compare | 
					
						
							|  |  |  | 	processedFullBytesExpected, err := os.ReadFile("./test/test-jpeg-processed.jpg") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// the bytes in storage should be what we expected | 
					
						
							|  |  |  | 	suite.Equal(processedFullBytesExpected, processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// now do the same for the thumbnail and make sure it's what we expected | 
					
						
							|  |  |  | 	processedThumbnailBytes, err := suite.storage.Get(ctx, attachment.Thumbnail.Path) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedThumbnailBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	processedThumbnailBytesExpected, err := os.ReadFile("./test/test-jpeg-thumbnail.jpg") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedThumbnailBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	suite.Equal(processedThumbnailBytesExpected, processedThumbnailBytes) | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | func (suite *ManagerTestSuite) TestSimpleJpegProcessBlockingReadCloser() { | 
					
						
							|  |  |  | 	ctx := context.Background() | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-11-03 15:03:12 +01:00
										 |  |  | 	data := func(_ context.Context) (io.ReadCloser, int64, error) { | 
					
						
							| 
									
										
										
										
											2022-09-24 11:11:47 +02:00
										 |  |  | 		// open test image as a file | 
					
						
							|  |  |  | 		f, err := os.Open("./test/test-jpeg.jpg") | 
					
						
							|  |  |  | 		if err != nil { | 
					
						
							|  |  |  | 			panic(err) | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		// give length as -1 to indicate unknown | 
					
						
							|  |  |  | 		return f, -1, nil | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	accountID := "01FS1X72SK9ZPW0J1QQ68BD264" | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// process the media with no additional info provided | 
					
						
							| 
									
										
										
										
											2023-05-28 13:08:35 +01:00
										 |  |  | 	processingMedia, err := suite.manager.ProcessMedia(ctx, data, accountID, nil) | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	// fetch the attachment id from the processing media | 
					
						
							|  |  |  | 	attachmentID := processingMedia.AttachmentID() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// do a blocking call to fetch the attachment | 
					
						
							| 
									
										
										
										
											2022-01-11 17:49:14 +01:00
										 |  |  | 	attachment, err := processingMedia.LoadAttachment(ctx) | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotNil(attachment) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// make sure it's got the stuff set on it that we expect | 
					
						
							|  |  |  | 	// the attachment ID and accountID we expect | 
					
						
							|  |  |  | 	suite.Equal(attachmentID, attachment.ID) | 
					
						
							|  |  |  | 	suite.Equal(accountID, attachment.AccountID) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// file meta should be correctly derived from the image | 
					
						
							|  |  |  | 	suite.EqualValues(gtsmodel.Original{ | 
					
						
							|  |  |  | 		Width: 1920, Height: 1080, Size: 2073600, Aspect: 1.7777777777777777, | 
					
						
							|  |  |  | 	}, attachment.FileMeta.Original) | 
					
						
							|  |  |  | 	suite.EqualValues(gtsmodel.Small{ | 
					
						
							|  |  |  | 		Width: 512, Height: 288, Size: 147456, Aspect: 1.7777777777777777, | 
					
						
							|  |  |  | 	}, attachment.FileMeta.Small) | 
					
						
							|  |  |  | 	suite.Equal("image/jpeg", attachment.File.ContentType) | 
					
						
							| 
									
										
										
										
											2022-02-21 11:26:26 +01:00
										 |  |  | 	suite.Equal("image/jpeg", attachment.Thumbnail.ContentType) | 
					
						
							| 
									
										
										
										
											2022-01-23 15:52:18 +01:00
										 |  |  | 	suite.Equal(269739, attachment.File.FileSize) | 
					
						
							| 
									
										
										
										
											2022-08-10 14:05:14 +02:00
										 |  |  | 	suite.Equal("LiBzRk#6V[WF_NvzV@WY_3rqV@a$", attachment.Blurhash) | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	// now make sure the attachment is in the database | 
					
						
							|  |  |  | 	dbAttachment, err := suite.db.GetAttachmentByID(ctx, attachmentID) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotNil(dbAttachment) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// make sure the processed file is in storage | 
					
						
							| 
									
										
										
										
											2022-07-03 12:08:30 +02:00
										 |  |  | 	processedFullBytes, err := suite.storage.Get(ctx, attachment.File.Path) | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytes) | 
					
						
							| 
									
										
										
										
											2022-01-09 18:41:22 +01:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | 	// load the processed bytes from our test folder, to compare | 
					
						
							|  |  |  | 	processedFullBytesExpected, err := os.ReadFile("./test/test-jpeg-processed.jpg") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// the bytes in storage should be what we expected | 
					
						
							|  |  |  | 	suite.Equal(processedFullBytesExpected, processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// now do the same for the thumbnail and make sure it's what we expected | 
					
						
							| 
									
										
										
										
											2022-07-03 12:08:30 +02:00
										 |  |  | 	processedThumbnailBytes, err := suite.storage.Get(ctx, attachment.Thumbnail.Path) | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedThumbnailBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	processedThumbnailBytesExpected, err := os.ReadFile("./test/test-jpeg-thumbnail.jpg") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedThumbnailBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	suite.Equal(processedThumbnailBytesExpected, processedThumbnailBytes) | 
					
						
							| 
									
										
										
										
											2022-01-09 18:41:22 +01:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-04-25 14:45:44 +02:00
										 |  |  | func (suite *ManagerTestSuite) TestPngNoAlphaChannelProcessBlocking() { | 
					
						
							|  |  |  | 	ctx := context.Background() | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-11-03 15:03:12 +01:00
										 |  |  | 	data := func(_ context.Context) (io.ReadCloser, int64, error) { | 
					
						
							| 
									
										
										
										
											2022-04-25 14:45:44 +02:00
										 |  |  | 		// load bytes from a test image | 
					
						
							|  |  |  | 		b, err := os.ReadFile("./test/test-png-noalphachannel.png") | 
					
						
							|  |  |  | 		if err != nil { | 
					
						
							|  |  |  | 			panic(err) | 
					
						
							|  |  |  | 		} | 
					
						
							| 
									
										
										
										
											2022-11-03 15:03:12 +01:00
										 |  |  | 		return io.NopCloser(bytes.NewBuffer(b)), int64(len(b)), nil | 
					
						
							| 
									
										
										
										
											2022-04-25 14:45:44 +02:00
										 |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	accountID := "01FS1X72SK9ZPW0J1QQ68BD264" | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// process the media with no additional info provided | 
					
						
							| 
									
										
										
										
											2023-05-28 13:08:35 +01:00
										 |  |  | 	processingMedia, err := suite.manager.ProcessMedia(ctx, data, accountID, nil) | 
					
						
							| 
									
										
										
										
											2022-04-25 14:45:44 +02:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	// fetch the attachment id from the processing media | 
					
						
							|  |  |  | 	attachmentID := processingMedia.AttachmentID() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// do a blocking call to fetch the attachment | 
					
						
							|  |  |  | 	attachment, err := processingMedia.LoadAttachment(ctx) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotNil(attachment) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// make sure it's got the stuff set on it that we expect | 
					
						
							|  |  |  | 	// the attachment ID and accountID we expect | 
					
						
							|  |  |  | 	suite.Equal(attachmentID, attachment.ID) | 
					
						
							|  |  |  | 	suite.Equal(accountID, attachment.AccountID) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// file meta should be correctly derived from the image | 
					
						
							|  |  |  | 	suite.EqualValues(gtsmodel.Original{ | 
					
						
							|  |  |  | 		Width: 186, Height: 187, Size: 34782, Aspect: 0.9946524064171123, | 
					
						
							|  |  |  | 	}, attachment.FileMeta.Original) | 
					
						
							|  |  |  | 	suite.EqualValues(gtsmodel.Small{ | 
					
						
							|  |  |  | 		Width: 186, Height: 187, Size: 34782, Aspect: 0.9946524064171123, | 
					
						
							|  |  |  | 	}, attachment.FileMeta.Small) | 
					
						
							|  |  |  | 	suite.Equal("image/png", attachment.File.ContentType) | 
					
						
							|  |  |  | 	suite.Equal("image/jpeg", attachment.Thumbnail.ContentType) | 
					
						
							|  |  |  | 	suite.Equal(17471, attachment.File.FileSize) | 
					
						
							| 
									
										
										
										
											2022-08-10 14:05:14 +02:00
										 |  |  | 	suite.Equal("LFQT7e.A%O%4?co$M}M{_1W9~TxV", attachment.Blurhash) | 
					
						
							| 
									
										
										
										
											2022-04-25 14:45:44 +02:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	// now make sure the attachment is in the database | 
					
						
							|  |  |  | 	dbAttachment, err := suite.db.GetAttachmentByID(ctx, attachmentID) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotNil(dbAttachment) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// make sure the processed file is in storage | 
					
						
							| 
									
										
										
										
											2022-07-03 12:08:30 +02:00
										 |  |  | 	processedFullBytes, err := suite.storage.Get(ctx, attachment.File.Path) | 
					
						
							| 
									
										
										
										
											2022-04-25 14:45:44 +02:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// load the processed bytes from our test folder, to compare | 
					
						
							|  |  |  | 	processedFullBytesExpected, err := os.ReadFile("./test/test-png-noalphachannel-processed.png") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// the bytes in storage should be what we expected | 
					
						
							|  |  |  | 	suite.Equal(processedFullBytesExpected, processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// now do the same for the thumbnail and make sure it's what we expected | 
					
						
							| 
									
										
										
										
											2022-07-03 12:08:30 +02:00
										 |  |  | 	processedThumbnailBytes, err := suite.storage.Get(ctx, attachment.Thumbnail.Path) | 
					
						
							| 
									
										
										
										
											2022-04-25 14:45:44 +02:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedThumbnailBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	processedThumbnailBytesExpected, err := os.ReadFile("./test/test-png-noalphachannel-thumbnail.jpg") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedThumbnailBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	suite.Equal(processedThumbnailBytesExpected, processedThumbnailBytes) | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | func (suite *ManagerTestSuite) TestPngAlphaChannelProcessBlocking() { | 
					
						
							|  |  |  | 	ctx := context.Background() | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-11-03 15:03:12 +01:00
										 |  |  | 	data := func(_ context.Context) (io.ReadCloser, int64, error) { | 
					
						
							| 
									
										
										
										
											2022-04-25 14:45:44 +02:00
										 |  |  | 		// load bytes from a test image | 
					
						
							|  |  |  | 		b, err := os.ReadFile("./test/test-png-alphachannel.png") | 
					
						
							|  |  |  | 		if err != nil { | 
					
						
							|  |  |  | 			panic(err) | 
					
						
							|  |  |  | 		} | 
					
						
							| 
									
										
										
										
											2022-11-03 15:03:12 +01:00
										 |  |  | 		return io.NopCloser(bytes.NewBuffer(b)), int64(len(b)), nil | 
					
						
							| 
									
										
										
										
											2022-04-25 14:45:44 +02:00
										 |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	accountID := "01FS1X72SK9ZPW0J1QQ68BD264" | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// process the media with no additional info provided | 
					
						
							| 
									
										
										
										
											2023-05-28 13:08:35 +01:00
										 |  |  | 	processingMedia, err := suite.manager.ProcessMedia(ctx, data, accountID, nil) | 
					
						
							| 
									
										
										
										
											2022-04-25 14:45:44 +02:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	// fetch the attachment id from the processing media | 
					
						
							|  |  |  | 	attachmentID := processingMedia.AttachmentID() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// do a blocking call to fetch the attachment | 
					
						
							|  |  |  | 	attachment, err := processingMedia.LoadAttachment(ctx) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotNil(attachment) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// make sure it's got the stuff set on it that we expect | 
					
						
							|  |  |  | 	// the attachment ID and accountID we expect | 
					
						
							|  |  |  | 	suite.Equal(attachmentID, attachment.ID) | 
					
						
							|  |  |  | 	suite.Equal(accountID, attachment.AccountID) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// file meta should be correctly derived from the image | 
					
						
							|  |  |  | 	suite.EqualValues(gtsmodel.Original{ | 
					
						
							|  |  |  | 		Width: 186, Height: 187, Size: 34782, Aspect: 0.9946524064171123, | 
					
						
							|  |  |  | 	}, attachment.FileMeta.Original) | 
					
						
							|  |  |  | 	suite.EqualValues(gtsmodel.Small{ | 
					
						
							|  |  |  | 		Width: 186, Height: 187, Size: 34782, Aspect: 0.9946524064171123, | 
					
						
							|  |  |  | 	}, attachment.FileMeta.Small) | 
					
						
							|  |  |  | 	suite.Equal("image/png", attachment.File.ContentType) | 
					
						
							|  |  |  | 	suite.Equal("image/jpeg", attachment.Thumbnail.ContentType) | 
					
						
							|  |  |  | 	suite.Equal(18904, attachment.File.FileSize) | 
					
						
							| 
									
										
										
										
											2022-08-10 14:05:14 +02:00
										 |  |  | 	suite.Equal("LFQT7e.A%O%4?co$M}M{_1W9~TxV", attachment.Blurhash) | 
					
						
							| 
									
										
										
										
											2022-04-25 14:45:44 +02:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	// now make sure the attachment is in the database | 
					
						
							|  |  |  | 	dbAttachment, err := suite.db.GetAttachmentByID(ctx, attachmentID) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotNil(dbAttachment) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// make sure the processed file is in storage | 
					
						
							| 
									
										
										
										
											2022-07-03 12:08:30 +02:00
										 |  |  | 	processedFullBytes, err := suite.storage.Get(ctx, attachment.File.Path) | 
					
						
							| 
									
										
										
										
											2022-04-25 14:45:44 +02:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// load the processed bytes from our test folder, to compare | 
					
						
							|  |  |  | 	processedFullBytesExpected, err := os.ReadFile("./test/test-png-alphachannel-processed.png") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// the bytes in storage should be what we expected | 
					
						
							|  |  |  | 	suite.Equal(processedFullBytesExpected, processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// now do the same for the thumbnail and make sure it's what we expected | 
					
						
							| 
									
										
										
										
											2022-07-03 12:08:30 +02:00
										 |  |  | 	processedThumbnailBytes, err := suite.storage.Get(ctx, attachment.Thumbnail.Path) | 
					
						
							| 
									
										
										
										
											2022-04-25 14:45:44 +02:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedThumbnailBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	processedThumbnailBytesExpected, err := os.ReadFile("./test/test-png-alphachannel-thumbnail.jpg") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedThumbnailBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	suite.Equal(processedThumbnailBytesExpected, processedThumbnailBytes) | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-02-22 13:50:33 +01:00
										 |  |  | func (suite *ManagerTestSuite) TestSimpleJpegProcessBlockingWithCallback() { | 
					
						
							|  |  |  | 	ctx := context.Background() | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-11-03 15:03:12 +01:00
										 |  |  | 	data := func(_ context.Context) (io.ReadCloser, int64, error) { | 
					
						
							| 
									
										
										
										
											2022-02-22 13:50:33 +01:00
										 |  |  | 		// load bytes from a test image | 
					
						
							|  |  |  | 		b, err := os.ReadFile("./test/test-jpeg.jpg") | 
					
						
							|  |  |  | 		if err != nil { | 
					
						
							|  |  |  | 			panic(err) | 
					
						
							|  |  |  | 		} | 
					
						
							| 
									
										
										
										
											2022-11-03 15:03:12 +01:00
										 |  |  | 		return io.NopCloser(bytes.NewBuffer(b)), int64(len(b)), nil | 
					
						
							| 
									
										
										
										
											2022-02-22 13:50:33 +01:00
										 |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	accountID := "01FS1X72SK9ZPW0J1QQ68BD264" | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// process the media with no additional info provided | 
					
						
							| 
									
										
										
										
											2023-05-28 13:08:35 +01:00
										 |  |  | 	processingMedia, err := suite.manager.ProcessMedia(ctx, data, accountID, nil) | 
					
						
							| 
									
										
										
										
											2022-02-22 13:50:33 +01:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	// fetch the attachment id from the processing media | 
					
						
							|  |  |  | 	attachmentID := processingMedia.AttachmentID() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// do a blocking call to fetch the attachment | 
					
						
							|  |  |  | 	attachment, err := processingMedia.LoadAttachment(ctx) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotNil(attachment) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// make sure it's got the stuff set on it that we expect | 
					
						
							|  |  |  | 	// the attachment ID and accountID we expect | 
					
						
							|  |  |  | 	suite.Equal(attachmentID, attachment.ID) | 
					
						
							|  |  |  | 	suite.Equal(accountID, attachment.AccountID) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// file meta should be correctly derived from the image | 
					
						
							|  |  |  | 	suite.EqualValues(gtsmodel.Original{ | 
					
						
							|  |  |  | 		Width: 1920, Height: 1080, Size: 2073600, Aspect: 1.7777777777777777, | 
					
						
							|  |  |  | 	}, attachment.FileMeta.Original) | 
					
						
							|  |  |  | 	suite.EqualValues(gtsmodel.Small{ | 
					
						
							|  |  |  | 		Width: 512, Height: 288, Size: 147456, Aspect: 1.7777777777777777, | 
					
						
							|  |  |  | 	}, attachment.FileMeta.Small) | 
					
						
							|  |  |  | 	suite.Equal("image/jpeg", attachment.File.ContentType) | 
					
						
							|  |  |  | 	suite.Equal("image/jpeg", attachment.Thumbnail.ContentType) | 
					
						
							|  |  |  | 	suite.Equal(269739, attachment.File.FileSize) | 
					
						
							| 
									
										
										
										
											2022-08-10 14:05:14 +02:00
										 |  |  | 	suite.Equal("LiBzRk#6V[WF_NvzV@WY_3rqV@a$", attachment.Blurhash) | 
					
						
							| 
									
										
										
										
											2022-02-22 13:50:33 +01:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	// now make sure the attachment is in the database | 
					
						
							|  |  |  | 	dbAttachment, err := suite.db.GetAttachmentByID(ctx, attachmentID) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotNil(dbAttachment) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// make sure the processed file is in storage | 
					
						
							| 
									
										
										
										
											2022-07-03 12:08:30 +02:00
										 |  |  | 	processedFullBytes, err := suite.storage.Get(ctx, attachment.File.Path) | 
					
						
							| 
									
										
										
										
											2022-02-22 13:50:33 +01:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// load the processed bytes from our test folder, to compare | 
					
						
							|  |  |  | 	processedFullBytesExpected, err := os.ReadFile("./test/test-jpeg-processed.jpg") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// the bytes in storage should be what we expected | 
					
						
							|  |  |  | 	suite.Equal(processedFullBytesExpected, processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// now do the same for the thumbnail and make sure it's what we expected | 
					
						
							| 
									
										
										
										
											2022-07-03 12:08:30 +02:00
										 |  |  | 	processedThumbnailBytes, err := suite.storage.Get(ctx, attachment.Thumbnail.Path) | 
					
						
							| 
									
										
										
										
											2022-02-22 13:50:33 +01:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedThumbnailBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	processedThumbnailBytesExpected, err := os.ReadFile("./test/test-jpeg-thumbnail.jpg") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedThumbnailBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	suite.Equal(processedThumbnailBytesExpected, processedThumbnailBytes) | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | func (suite *ManagerTestSuite) TestSimpleJpegProcessAsync() { | 
					
						
							| 
									
										
										
										
											2023-01-11 11:13:13 +00:00
										 |  |  | 	ctx, cncl := context.WithTimeout(context.Background(), time.Second*30) | 
					
						
							|  |  |  | 	defer cncl() | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-11-03 15:03:12 +01:00
										 |  |  | 	data := func(_ context.Context) (io.ReadCloser, int64, error) { | 
					
						
							| 
									
										
										
										
											2022-01-11 17:49:14 +01:00
										 |  |  | 		// load bytes from a test image | 
					
						
							| 
									
										
										
										
											2022-01-16 18:52:55 +01:00
										 |  |  | 		b, err := os.ReadFile("./test/test-jpeg.jpg") | 
					
						
							|  |  |  | 		if err != nil { | 
					
						
							|  |  |  | 			panic(err) | 
					
						
							|  |  |  | 		} | 
					
						
							| 
									
										
										
										
											2022-11-03 15:03:12 +01:00
										 |  |  | 		return io.NopCloser(bytes.NewBuffer(b)), int64(len(b)), nil | 
					
						
							| 
									
										
										
										
											2022-01-11 17:49:14 +01:00
										 |  |  | 	} | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	accountID := "01FS1X72SK9ZPW0J1QQ68BD264" | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// process the media with no additional info provided | 
					
						
							| 
									
										
										
										
											2023-05-28 13:08:35 +01:00
										 |  |  | 	processingMedia, err := suite.manager.ProcessMedia(ctx, data, accountID, nil) | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | 	suite.NoError(err) | 
					
						
							| 
									
										
										
										
											2023-01-11 11:13:13 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | 	// fetch the attachment id from the processing media | 
					
						
							|  |  |  | 	attachmentID := processingMedia.AttachmentID() | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2023-05-25 10:37:38 +02:00
										 |  |  | 	// wait for processing to complete | 
					
						
							|  |  |  | 	var attachment *gtsmodel.MediaAttachment | 
					
						
							|  |  |  | 	if !testrig.WaitFor(func() bool { | 
					
						
							|  |  |  | 		attachment, err = suite.db.GetAttachmentByID(ctx, attachmentID) | 
					
						
							|  |  |  | 		return err == nil && attachment != nil | 
					
						
							|  |  |  | 	}) { | 
					
						
							|  |  |  | 		suite.FailNow("timed out waiting for attachment to process") | 
					
						
							|  |  |  | 	} | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	// make sure it's got the stuff set on it that we expect | 
					
						
							|  |  |  | 	// the attachment ID and accountID we expect | 
					
						
							|  |  |  | 	suite.Equal(attachmentID, attachment.ID) | 
					
						
							|  |  |  | 	suite.Equal(accountID, attachment.AccountID) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// file meta should be correctly derived from the image | 
					
						
							|  |  |  | 	suite.EqualValues(gtsmodel.Original{ | 
					
						
							|  |  |  | 		Width: 1920, Height: 1080, Size: 2073600, Aspect: 1.7777777777777777, | 
					
						
							|  |  |  | 	}, attachment.FileMeta.Original) | 
					
						
							|  |  |  | 	suite.EqualValues(gtsmodel.Small{ | 
					
						
							|  |  |  | 		Width: 512, Height: 288, Size: 147456, Aspect: 1.7777777777777777, | 
					
						
							|  |  |  | 	}, attachment.FileMeta.Small) | 
					
						
							|  |  |  | 	suite.Equal("image/jpeg", attachment.File.ContentType) | 
					
						
							| 
									
										
										
										
											2022-02-21 11:26:26 +01:00
										 |  |  | 	suite.Equal("image/jpeg", attachment.Thumbnail.ContentType) | 
					
						
							| 
									
										
										
										
											2022-01-23 15:52:18 +01:00
										 |  |  | 	suite.Equal(269739, attachment.File.FileSize) | 
					
						
							| 
									
										
										
										
											2022-08-10 14:05:14 +02:00
										 |  |  | 	suite.Equal("LiBzRk#6V[WF_NvzV@WY_3rqV@a$", attachment.Blurhash) | 
					
						
							| 
									
										
										
										
											2022-01-09 18:41:22 +01:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | 	// now make sure the attachment is in the database | 
					
						
							|  |  |  | 	dbAttachment, err := suite.db.GetAttachmentByID(ctx, attachmentID) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotNil(dbAttachment) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// make sure the processed file is in storage | 
					
						
							| 
									
										
										
										
											2022-07-03 12:08:30 +02:00
										 |  |  | 	processedFullBytes, err := suite.storage.Get(ctx, attachment.File.Path) | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// load the processed bytes from our test folder, to compare | 
					
						
							|  |  |  | 	processedFullBytesExpected, err := os.ReadFile("./test/test-jpeg-processed.jpg") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// the bytes in storage should be what we expected | 
					
						
							|  |  |  | 	suite.Equal(processedFullBytesExpected, processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// now do the same for the thumbnail and make sure it's what we expected | 
					
						
							| 
									
										
										
										
											2022-07-03 12:08:30 +02:00
										 |  |  | 	processedThumbnailBytes, err := suite.storage.Get(ctx, attachment.Thumbnail.Path) | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedThumbnailBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	processedThumbnailBytesExpected, err := os.ReadFile("./test/test-jpeg-thumbnail.jpg") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedThumbnailBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	suite.Equal(processedThumbnailBytesExpected, processedThumbnailBytes) | 
					
						
							| 
									
										
										
										
											2022-01-09 18:41:22 +01:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2022-01-08 17:17:01 +01:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | func (suite *ManagerTestSuite) TestSimpleJpegQueueSpamming() { | 
					
						
							|  |  |  | 	// in this test, we spam the manager queue with 50 new media requests, just to see how it holds up | 
					
						
							|  |  |  | 	ctx := context.Background() | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-01-16 18:52:55 +01:00
										 |  |  | 	b, err := os.ReadFile("./test/test-jpeg.jpg") | 
					
						
							|  |  |  | 	if err != nil { | 
					
						
							|  |  |  | 		panic(err) | 
					
						
							|  |  |  | 	} | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-11-03 15:03:12 +01:00
										 |  |  | 	data := func(_ context.Context) (io.ReadCloser, int64, error) { | 
					
						
							| 
									
										
										
										
											2022-01-16 18:52:55 +01:00
										 |  |  | 		// load bytes from a test image | 
					
						
							| 
									
										
										
										
											2022-11-03 15:03:12 +01:00
										 |  |  | 		return io.NopCloser(bytes.NewReader(b)), int64(len(b)), nil | 
					
						
							| 
									
										
										
										
											2022-01-11 17:49:14 +01:00
										 |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | 	accountID := "01FS1X72SK9ZPW0J1QQ68BD264" | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	spam := 50 | 
					
						
							| 
									
										
										
										
											2022-01-11 17:49:14 +01:00
										 |  |  | 	inProcess := []*media.ProcessingMedia{} | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | 	for i := 0; i < spam; i++ { | 
					
						
							|  |  |  | 		// process the media with no additional info provided | 
					
						
							| 
									
										
										
										
											2023-05-28 13:08:35 +01:00
										 |  |  | 		processingMedia, err := suite.manager.ProcessMedia(ctx, data, accountID, nil) | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | 		suite.NoError(err) | 
					
						
							|  |  |  | 		inProcess = append(inProcess, processingMedia) | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	for _, processingMedia := range inProcess { | 
					
						
							|  |  |  | 		// fetch the attachment id from the processing media | 
					
						
							|  |  |  | 		attachmentID := processingMedia.AttachmentID() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		// do a blocking call to fetch the attachment | 
					
						
							| 
									
										
										
										
											2022-01-11 17:49:14 +01:00
										 |  |  | 		attachment, err := processingMedia.LoadAttachment(ctx) | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | 		suite.NoError(err) | 
					
						
							|  |  |  | 		suite.NotNil(attachment) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		// make sure it's got the stuff set on it that we expect | 
					
						
							|  |  |  | 		// the attachment ID and accountID we expect | 
					
						
							|  |  |  | 		suite.Equal(attachmentID, attachment.ID) | 
					
						
							|  |  |  | 		suite.Equal(accountID, attachment.AccountID) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		// file meta should be correctly derived from the image | 
					
						
							|  |  |  | 		suite.EqualValues(gtsmodel.Original{ | 
					
						
							|  |  |  | 			Width: 1920, Height: 1080, Size: 2073600, Aspect: 1.7777777777777777, | 
					
						
							|  |  |  | 		}, attachment.FileMeta.Original) | 
					
						
							|  |  |  | 		suite.EqualValues(gtsmodel.Small{ | 
					
						
							|  |  |  | 			Width: 512, Height: 288, Size: 147456, Aspect: 1.7777777777777777, | 
					
						
							|  |  |  | 		}, attachment.FileMeta.Small) | 
					
						
							|  |  |  | 		suite.Equal("image/jpeg", attachment.File.ContentType) | 
					
						
							| 
									
										
										
										
											2022-02-21 11:26:26 +01:00
										 |  |  | 		suite.Equal("image/jpeg", attachment.Thumbnail.ContentType) | 
					
						
							| 
									
										
										
										
											2022-01-23 15:52:18 +01:00
										 |  |  | 		suite.Equal(269739, attachment.File.FileSize) | 
					
						
							| 
									
										
										
										
											2022-08-10 14:05:14 +02:00
										 |  |  | 		suite.Equal("LiBzRk#6V[WF_NvzV@WY_3rqV@a$", attachment.Blurhash) | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | 
 | 
					
						
							|  |  |  | 		// now make sure the attachment is in the database | 
					
						
							|  |  |  | 		dbAttachment, err := suite.db.GetAttachmentByID(ctx, attachmentID) | 
					
						
							|  |  |  | 		suite.NoError(err) | 
					
						
							|  |  |  | 		suite.NotNil(dbAttachment) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		// make sure the processed file is in storage | 
					
						
							| 
									
										
										
										
											2022-07-03 12:08:30 +02:00
										 |  |  | 		processedFullBytes, err := suite.storage.Get(ctx, attachment.File.Path) | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | 		suite.NoError(err) | 
					
						
							|  |  |  | 		suite.NotEmpty(processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		// load the processed bytes from our test folder, to compare | 
					
						
							|  |  |  | 		processedFullBytesExpected, err := os.ReadFile("./test/test-jpeg-processed.jpg") | 
					
						
							|  |  |  | 		suite.NoError(err) | 
					
						
							|  |  |  | 		suite.NotEmpty(processedFullBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		// the bytes in storage should be what we expected | 
					
						
							|  |  |  | 		suite.Equal(processedFullBytesExpected, processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		// now do the same for the thumbnail and make sure it's what we expected | 
					
						
							| 
									
										
										
										
											2022-07-03 12:08:30 +02:00
										 |  |  | 		processedThumbnailBytes, err := suite.storage.Get(ctx, attachment.Thumbnail.Path) | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | 		suite.NoError(err) | 
					
						
							|  |  |  | 		suite.NotEmpty(processedThumbnailBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		processedThumbnailBytesExpected, err := os.ReadFile("./test/test-jpeg-thumbnail.jpg") | 
					
						
							|  |  |  | 		suite.NoError(err) | 
					
						
							|  |  |  | 		suite.NotEmpty(processedThumbnailBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		suite.Equal(processedThumbnailBytesExpected, processedThumbnailBytes) | 
					
						
							|  |  |  | 	} | 
					
						
							| 
									
										
										
										
											2022-01-09 18:41:22 +01:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2022-01-08 17:17:01 +01:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-01-24 18:11:27 +01:00
										 |  |  | func (suite *ManagerTestSuite) TestSimpleJpegProcessBlockingWithDiskStorage() { | 
					
						
							|  |  |  | 	ctx := context.Background() | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-11-03 15:03:12 +01:00
										 |  |  | 	data := func(_ context.Context) (io.ReadCloser, int64, error) { | 
					
						
							| 
									
										
										
										
											2022-01-24 18:11:27 +01:00
										 |  |  | 		// load bytes from a test image | 
					
						
							|  |  |  | 		b, err := os.ReadFile("./test/test-jpeg.jpg") | 
					
						
							|  |  |  | 		if err != nil { | 
					
						
							|  |  |  | 			panic(err) | 
					
						
							|  |  |  | 		} | 
					
						
							| 
									
										
										
										
											2022-11-03 15:03:12 +01:00
										 |  |  | 		return io.NopCloser(bytes.NewBuffer(b)), int64(len(b)), nil | 
					
						
							| 
									
										
										
										
											2022-01-24 18:11:27 +01:00
										 |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	accountID := "01FS1X72SK9ZPW0J1QQ68BD264" | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-01-29 12:16:12 +01:00
										 |  |  | 	temp := fmt.Sprintf("%s/gotosocial-test", os.TempDir()) | 
					
						
							| 
									
										
										
										
											2022-01-24 18:11:27 +01:00
										 |  |  | 	defer os.RemoveAll(temp) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-11-24 08:35:46 +00:00
										 |  |  | 	disk, err := storage.OpenDisk(temp, &storage.DiskConfig{ | 
					
						
							| 
									
										
										
										
											2022-01-29 12:16:12 +01:00
										 |  |  | 		LockFile: path.Join(temp, "store.lock"), | 
					
						
							|  |  |  | 	}) | 
					
						
							| 
									
										
										
										
											2022-01-24 18:11:27 +01:00
										 |  |  | 	if err != nil { | 
					
						
							|  |  |  | 		panic(err) | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2023-02-13 18:40:48 +00:00
										 |  |  | 	var state state.State | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	state.Workers.Start() | 
					
						
							|  |  |  | 	defer state.Workers.Stop() | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-11-24 08:35:46 +00:00
										 |  |  | 	storage := >sstorage.Driver{ | 
					
						
							|  |  |  | 		Storage: disk, | 
					
						
							|  |  |  | 	} | 
					
						
							| 
									
										
										
										
											2023-02-13 18:40:48 +00:00
										 |  |  | 	state.Storage = storage | 
					
						
							|  |  |  | 	state.DB = suite.db | 
					
						
							| 
									
										
										
										
											2022-11-24 08:35:46 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2023-02-13 18:40:48 +00:00
										 |  |  | 	diskManager := media.NewManager(&state) | 
					
						
							| 
									
										
										
										
											2022-01-24 18:11:27 +01:00
										 |  |  | 	suite.manager = diskManager | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// process the media with no additional info provided | 
					
						
							| 
									
										
										
										
											2023-05-28 13:08:35 +01:00
										 |  |  | 	processingMedia, err := diskManager.ProcessMedia(ctx, data, accountID, nil) | 
					
						
							| 
									
										
										
										
											2022-01-24 18:11:27 +01:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	// fetch the attachment id from the processing media | 
					
						
							|  |  |  | 	attachmentID := processingMedia.AttachmentID() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// do a blocking call to fetch the attachment | 
					
						
							|  |  |  | 	attachment, err := processingMedia.LoadAttachment(ctx) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotNil(attachment) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// make sure it's got the stuff set on it that we expect | 
					
						
							|  |  |  | 	// the attachment ID and accountID we expect | 
					
						
							|  |  |  | 	suite.Equal(attachmentID, attachment.ID) | 
					
						
							|  |  |  | 	suite.Equal(accountID, attachment.AccountID) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// file meta should be correctly derived from the image | 
					
						
							|  |  |  | 	suite.EqualValues(gtsmodel.Original{ | 
					
						
							|  |  |  | 		Width: 1920, Height: 1080, Size: 2073600, Aspect: 1.7777777777777777, | 
					
						
							|  |  |  | 	}, attachment.FileMeta.Original) | 
					
						
							|  |  |  | 	suite.EqualValues(gtsmodel.Small{ | 
					
						
							|  |  |  | 		Width: 512, Height: 288, Size: 147456, Aspect: 1.7777777777777777, | 
					
						
							|  |  |  | 	}, attachment.FileMeta.Small) | 
					
						
							|  |  |  | 	suite.Equal("image/jpeg", attachment.File.ContentType) | 
					
						
							| 
									
										
										
										
											2022-02-21 11:26:26 +01:00
										 |  |  | 	suite.Equal("image/jpeg", attachment.Thumbnail.ContentType) | 
					
						
							| 
									
										
										
										
											2022-01-24 18:11:27 +01:00
										 |  |  | 	suite.Equal(269739, attachment.File.FileSize) | 
					
						
							| 
									
										
										
										
											2022-08-10 14:05:14 +02:00
										 |  |  | 	suite.Equal("LiBzRk#6V[WF_NvzV@WY_3rqV@a$", attachment.Blurhash) | 
					
						
							| 
									
										
										
										
											2022-01-24 18:11:27 +01:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	// now make sure the attachment is in the database | 
					
						
							|  |  |  | 	dbAttachment, err := suite.db.GetAttachmentByID(ctx, attachmentID) | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotNil(dbAttachment) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// make sure the processed file is in storage | 
					
						
							| 
									
										
										
										
											2022-11-24 08:35:46 +00:00
										 |  |  | 	processedFullBytes, err := storage.Get(ctx, attachment.File.Path) | 
					
						
							| 
									
										
										
										
											2022-01-24 18:11:27 +01:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// load the processed bytes from our test folder, to compare | 
					
						
							|  |  |  | 	processedFullBytesExpected, err := os.ReadFile("./test/test-jpeg-processed.jpg") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedFullBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// the bytes in storage should be what we expected | 
					
						
							|  |  |  | 	suite.Equal(processedFullBytesExpected, processedFullBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// now do the same for the thumbnail and make sure it's what we expected | 
					
						
							| 
									
										
										
										
											2022-11-24 08:35:46 +00:00
										 |  |  | 	processedThumbnailBytes, err := storage.Get(ctx, attachment.Thumbnail.Path) | 
					
						
							| 
									
										
										
										
											2022-01-24 18:11:27 +01:00
										 |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedThumbnailBytes) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	processedThumbnailBytesExpected, err := os.ReadFile("./test/test-jpeg-thumbnail.jpg") | 
					
						
							|  |  |  | 	suite.NoError(err) | 
					
						
							|  |  |  | 	suite.NotEmpty(processedThumbnailBytesExpected) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	suite.Equal(processedThumbnailBytesExpected, processedThumbnailBytes) | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-01-10 18:36:09 +01:00
										 |  |  | func TestManagerTestSuite(t *testing.T) { | 
					
						
							|  |  |  | 	suite.Run(t, &ManagerTestSuite{}) | 
					
						
							| 
									
										
										
										
											2022-01-09 18:41:22 +01:00
										 |  |  | } |