b441f3f3de
If there is a `scripts` directory next to the current executable, load scripts from that directory as well. It is still required to restart the Manager in order to pick up changes to those scripts (including new/removed files), PLUS a refresh in the add-on.
75 lines
1.7 KiB
Go
75 lines
1.7 KiB
Go
package job_compilers
|
|
|
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/rs/zerolog"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestLoadScriptsFrom_skip_nonjs(t *testing.T) {
|
|
thisDirFS := os.DirFS(".")
|
|
compilers, err := loadScriptsFrom(thisDirFS)
|
|
assert.NoError(t, err, "input without JS files should not cause errors")
|
|
assert.Empty(t, compilers)
|
|
}
|
|
|
|
func TestLoadScriptsFrom_on_disk_js(t *testing.T) {
|
|
scriptsFS := os.DirFS("scripts-for-unittest")
|
|
compilers, err := loadScriptsFrom(scriptsFS)
|
|
|
|
assert.NoError(t, err)
|
|
expectKeys := map[string]bool{
|
|
"echo-and-sleep": true,
|
|
"simple-blender-render": true,
|
|
// Should NOT contain an entry for 'empty.js'.
|
|
}
|
|
assert.Equal(t, expectKeys, keys(compilers))
|
|
}
|
|
|
|
func TestLoadScriptsFrom_embedded(t *testing.T) {
|
|
initEmbeddedFS()
|
|
compilers, err := loadScriptsFrom(embeddedScriptsFS)
|
|
|
|
assert.NoError(t, err)
|
|
expectKeys := map[string]bool{
|
|
"echo-sleep-test": true,
|
|
"simple-blender-render": true,
|
|
}
|
|
assert.Equal(t, expectKeys, keys(compilers))
|
|
}
|
|
|
|
func BenchmarkLoadScripts_fromEmbedded(b *testing.B) {
|
|
zerolog.SetGlobalLevel(zerolog.Disabled)
|
|
initEmbeddedFS()
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
compilers, err := loadScriptsFrom(embeddedScriptsFS)
|
|
assert.NoError(b, err)
|
|
assert.NotEmpty(b, compilers)
|
|
}
|
|
}
|
|
|
|
func BenchmarkLoadScripts_fromDisk(b *testing.B) {
|
|
zerolog.SetGlobalLevel(zerolog.Disabled)
|
|
|
|
onDiskFS := os.DirFS("scripts-for-unittest")
|
|
for i := 0; i < b.N; i++ {
|
|
compilers, err := loadScriptsFrom(onDiskFS)
|
|
assert.NoError(b, err)
|
|
assert.NotEmpty(b, compilers)
|
|
}
|
|
}
|
|
|
|
// keys returns the set of keys of the mapping.
|
|
func keys[K comparable, V any](mapping map[K]V) map[K]bool {
|
|
keys := map[K]bool{}
|
|
for k := range mapping {
|
|
keys[k] = true
|
|
}
|
|
return keys
|
|
}
|