Lithium/pipelines/pipeline_test.go

package pipelines

import (
	"github.com/geplauder/lithium/storage"
	"image"
	"os"
	"path/filepath"
	"testing"

	"github.com/stretchr/testify/assert"
)

// pipeline deserialization

func TestImagePipelineDeserialization(t *testing.T) {
	const Payload string = `{
		"name": "example pipeline",
		"type": 0,
		"removeMetadata": false,
		"steps": [
			{
				"name": "resize image",
				"type": 0
			},
			{
				"name": "compress image",
				"type": 1
			}
		]
	}`

	t.Run("Image pipeline deserialization is successful", func(t *testing.T) {
		values := DeserializePipelines([][]byte{[]byte(Payload)})

		assert.Equal(t, 1, len(values), "Output should contain one element")
		assert.Equal(t, "example pipeline", values[0].GetName())
		assert.Equal(t, Image, values[0].GetType())
	})
}

func TestVideoPipelineDeserialization(t *testing.T) {
	const Payload string = `{
		"name": "example pipeline",
		"type": 1,
		"removeMetadata": false,
		"steps": [
			{
				"name": "resize image",
				"type": 0
			},
			{
				"name": "compress image",
				"type": 1
			}
		]
	}`

	t.Run("Video pipelines deserialization is successful", func(t *testing.T) {
		values := DeserializePipelines([][]byte{[]byte(Payload)})

		assert.Equal(t, 1, len(values), "Output should contain one element")
		assert.Equal(t, "example pipeline", values[0].GetName())
		assert.Equal(t, Video, values[0].GetType())
	})
}

// image pipeline steps

func TestNoSteps(t *testing.T) {
	const Payload string = `{
		"name": "example pipeline",
		"type": 1,
		"removeMetadata": false,
		"steps": []
	}`
	const Bucket string = "pipeline_test_04"

	t.Run("Pipeline execution is successful", func(t *testing.T) {
		wd, _ := os.Getwd()
		pipe := DeserializePipelines([][]byte{[]byte(Payload)})[0]

		storageProvider := storage.GetFileSystemStorageProvider("test", "..")

		// copy test file to storage bucket
		_, err := storageProvider.StoreExisting(Bucket, "source.jpg", filepath.Join(wd, "../tests/files/900x900.jpg"))
		assert.Nil(t, err, "Test file should be readable")
		assert.FileExists(t, storageProvider.GetPath(Bucket, "source.jpg"))

		// run pipeline steps
		dest, err := pipe.Run("source.jpg", Bucket, storageProvider)
		assert.Nil(t, err)
		assert.FileExists(t, storageProvider.GetPath(Bucket, dest))

		// clean up
		os.Remove(storageProvider.GetPath(Bucket, "source.jpg"))
		os.Remove(storageProvider.GetPath(Bucket, dest))
	})
}

func TestResizeImage(t *testing.T) {
	const Payload string = `{
		"name": "example pipeline",
		"type": 1,
		"removeMetadata": false,
		"steps": [
			{
				"name": "resize image",
				"type": 0,
				"options": {
					"width": 1280,
					"height": 720,
					"upscale": false
				}
			}
		]
	}`
	const Bucket string = "pipeline_test_01"

	t.Run("Image resizing is successful", func(t *testing.T) {
		wd, _ := os.Getwd()
		pipe := DeserializePipelines([][]byte{[]byte(Payload)})[0]

		storageProvider := storage.GetFileSystemStorageProvider("test", "..")

		// copy test file to storage bucket
		_, err := storageProvider.StoreExisting(Bucket, "source.jpg", filepath.Join(wd, "../tests/files/900x900.jpg"))
		assert.Nil(t, err, "Test file should be readable")
		assert.FileExists(t, storageProvider.GetPath(Bucket, "source.jpg"))

		// run pipeline steps
		dest, err := pipe.Run("source.jpg", Bucket, storageProvider)
		assert.Nil(t, err)
		assert.FileExists(t, storageProvider.GetPath(Bucket, dest))

		// read image config
		file, err := storageProvider.OpenFile(Bucket, dest)
		assert.Nil(t, err)

		imgConf, _, err := image.DecodeConfig(file)
		assert.Nil(t, err)

		assert.Equal(t, 1280, imgConf.Width)
		assert.Equal(t, 720, imgConf.Height)

		// clean up
		os.Remove(storageProvider.GetPath(Bucket, "source.jpg"))
		os.Remove(storageProvider.GetPath(Bucket, dest))
	})
}

func TestRotateImage(t *testing.T) {
	const Payload string = `{
		"name": "example pipeline",
		"type": 1,
		"removeMetadata": false,
		"steps": [
			{
				"name": "rotate image",
				"type": 1,
				"options": {
					"angle": 90.0
				}
			}
		]
	}`
	const Bucket string = "pipeline_test_03"

	t.Run("Image rotation step is successful", func(t *testing.T) {
		wd, _ := os.Getwd()
		pipe := DeserializePipelines([][]byte{[]byte(Payload)})[0]

		storageProvider := storage.GetFileSystemStorageProvider("test", "..")

		// copy test file to storage bucket
		_, err := storageProvider.StoreExisting(Bucket, "source.jpg", filepath.Join(wd, "../tests/files/800x500.jpg"))
		assert.Nil(t, err, "Test file should be readable")
		assert.FileExists(t, storageProvider.GetPath(Bucket, "source.jpg"))

		// run pipeline steps
		dest, err := pipe.Run("source.jpg", Bucket, storageProvider)
		assert.Nil(t, err)
		assert.FileExists(t, storageProvider.GetPath(Bucket, dest))

		// read image config
		file, err := storageProvider.OpenFile(Bucket, dest)
		assert.Nil(t, err)

		imgConf, _, err := image.DecodeConfig(file)
		assert.Nil(t, err)

		assert.Equal(t, 500, imgConf.Width)
		assert.Equal(t, 800, imgConf.Height)

		// clean up
		os.Remove(storageProvider.GetPath(Bucket, "source.jpg"))
		os.Remove(storageProvider.GetPath(Bucket, dest))
	})
}

// output options

func TestEncodeImageWithJpegQuality(t *testing.T) {
	const Payload string = `{
		"name": "example pipeline",
		"type": 1,
		"removeMetadata": false,
		"steps": [
			{
				"name": "resize image",
				"type": 0,
				"options": {
					"width": 1280,
					"height": 720,
					"upscale": false
				}
			}
		],
		"output": {
			"quality": 50
		}
	}`
	const Bucket string = "pipeline_test_02"

	t.Run("Image encoding with jpeg quality is successful", func(t *testing.T) {
		wd, _ := os.Getwd()
		pipe := DeserializePipelines([][]byte{[]byte(Payload)})[0]

		storageProvider := storage.GetFileSystemStorageProvider("test", "..")

		// copy test file to storage bucket
		_, err := storageProvider.StoreExisting(Bucket, "source.jpg", filepath.Join(wd, "../tests/files/900x900.jpg"))
		assert.Nil(t, err, "Test file should be readable")
		assert.FileExists(t, storageProvider.GetPath(Bucket, "source.jpg"))

		// run pipeline steps
		dest, err := pipe.Run("source.jpg", Bucket, storageProvider)
		assert.Nil(t, err)
		assert.FileExists(t, storageProvider.GetPath(Bucket, dest))

		// clean up
		os.Remove(storageProvider.GetPath(Bucket, "source.jpg"))
		os.Remove(storageProvider.GetPath(Bucket, dest))
	})
}