types_test.go

// SPDX-License-Identifier: Apache-2.0
// SPDX-FileCopyrightText: 2024-Present Harry Randazzo

package vai

import (
	"encoding/json"
	"os"
	"testing"

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

// do not make changes to this variable within tests
var helloWorldWorkflow = Workflow{
	"default": {Step{Run: "echo 'Hello World!'"}},
	"a-task":  {Step{Run: "echo 'task a'"}},
	"task-b":  {Step{Run: "echo 'task b'"}},
}

func TestWorkflowFind(t *testing.T) {
	task, ok := helloWorldWorkflow.Find(DefaultTaskName)
	require.True(t, ok)

	require.Len(t, task, 1)
	require.Equal(t, "echo 'Hello World!'", task[0].Run)

	task, ok = helloWorldWorkflow.Find("foo")
	require.Nil(t, task)
	require.False(t, ok)
}

func TestOrderedTaskNames(t *testing.T) {
	names := helloWorldWorkflow.OrderedTaskNames()
	expected := []string{"default", "a-task", "task-b"}
	require.ElementsMatch(t, expected, names)

	wf := Workflow{"foo": nil, "bar": nil, "baz": nil, "default": nil}
	names = wf.OrderedTaskNames()
	expected = []string{"default", "bar", "baz", "foo"}
	require.ElementsMatch(t, expected, names)

	delete(wf, "default")

	names = wf.OrderedTaskNames()
	expected = []string{"bar", "baz", "foo"}
	require.ElementsMatch(t, expected, names)
}

func TestWorkflowSchemaGen(t *testing.T) {
	schema := WorkFlowSchema()

	require.NotNil(t, schema)

	b, err := json.Marshal(schema)
	require.NoError(t, err)

	current, err := os.ReadFile("vai.schema.json")
	require.NoError(t, err)

	require.JSONEq(t, string(current), string(b))
}