Lithium/pipeline_test.go

package main

import (
	"github.com/stretchr/testify/assert"
	"testing"
)

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

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

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

		assert.Equal(t, 1, len(steps), "Output steps should contain one element")
		assert.Equal(t, "resize image", steps[0].GetName())
		assert.Equal(t, TypeResizeImageStep, steps[0].GetType())

		translated, err := steps[0].Translate()
		assert.Equal(t, err, nil)
		assert.IsType(t, ResizeImageStep{}, translated)
	})
}

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

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

		assert.Equal(t, 1, len(steps), "Output steps should contain one element")
		assert.Equal(t, "resize image", steps[0].GetName())
		assert.Equal(t, StepType(99999), steps[0].GetType())

		translated, err := steps[0].Translate()
		assert.EqualError(t, err, "invalid type")
		assert.Nil(t, translated)
	})
}