Lithium/pipelines/pipeline_test.go

package pipelines

import (
	"testing"

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

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())
	})
}