103 lines
1.6 KiB
Go
103 lines
1.6 KiB
Go
package automation
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"errors"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// Hello! This file is auto-generated.
|
|
|
|
func TestScriptSetWalk(t *testing.T) {
|
|
var (
|
|
value = make(ScriptSet, 3)
|
|
req = require.New(t)
|
|
)
|
|
|
|
// check walk with no errors
|
|
{
|
|
err := value.Walk(func(*Script) error {
|
|
return nil
|
|
})
|
|
req.NoError(err)
|
|
}
|
|
|
|
// check walk with error
|
|
req.Error(value.Walk(func(*Script) error { return errors.New("walk error") }))
|
|
|
|
}
|
|
|
|
func TestScriptSetFilter(t *testing.T) {
|
|
var (
|
|
value = make(ScriptSet, 3)
|
|
req = require.New(t)
|
|
)
|
|
|
|
// filter nothing
|
|
{
|
|
set, err := value.Filter(func(*Script) (bool, error) {
|
|
return true, nil
|
|
})
|
|
req.NoError(err)
|
|
req.Equal(len(set), len(value))
|
|
}
|
|
|
|
// filter one item
|
|
{
|
|
found := false
|
|
set, err := value.Filter(func(*Script) (bool, error) {
|
|
if !found {
|
|
found = true
|
|
return found, nil
|
|
}
|
|
return false, nil
|
|
})
|
|
req.NoError(err)
|
|
req.Len(set, 1)
|
|
}
|
|
|
|
// filter error
|
|
{
|
|
_, err := value.Filter(func(*Script) (bool, error) {
|
|
return false, errors.New("filter error")
|
|
})
|
|
req.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestScriptSetIDs(t *testing.T) {
|
|
var (
|
|
value = make(ScriptSet, 3)
|
|
req = require.New(t)
|
|
)
|
|
|
|
// construct objects
|
|
value[0] = new(Script)
|
|
value[1] = new(Script)
|
|
value[2] = new(Script)
|
|
// set ids
|
|
value[0].ID = 1
|
|
value[1].ID = 2
|
|
value[2].ID = 3
|
|
|
|
// Find existing
|
|
{
|
|
val := value.FindByID(2)
|
|
req.Equal(uint64(2), val.ID)
|
|
}
|
|
|
|
// Find non-existing
|
|
{
|
|
val := value.FindByID(4)
|
|
req.Nil(val)
|
|
}
|
|
|
|
// List IDs from set
|
|
{
|
|
val := value.IDs()
|
|
req.Equal(len(val), len(value))
|
|
}
|
|
}
|