gcode/macro_test.go at main · lestrrat-go/gcode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
package gcode_test

import (
	"fmt"
	"testing"

	"github.com/lestrrat-go/gcode"
	"github.com/stretchr/testify/require"
)

func TestSimpleMacroExpand(t *testing.T) {
	t.Parallel()
	m := gcode.NewSimpleMacro("home-and-move",
		gcode.NewLine("G28"),
		gcode.NewLine("G1").ArgF("X", 10).ArgF("Y", 20),
	)
	require.Equal(t, "home-and-move", m.Name())

	expanded, err := m.Expand(nil)
	require.NoError(t, err)
	require.Len(t, expanded, 2)
	require.Equal(t, "G28", expanded[0].Command.Name)
	require.Equal(t, "G1", expanded[1].Command.Name)
	require.Len(t, expanded[1].Command.Args, 2)
}

func TestSimpleMacroExpandIsDeepCopy(t *testing.T) {
	t.Parallel()
	m := gcode.NewSimpleMacro("move",
		gcode.NewLine("G1").ArgF("X", 10),
	)

	first, err := m.Expand(nil)
	require.NoError(t, err)
	first[0].Command.Args[0].Number = 999

	second, err := m.Expand(nil)
	require.NoError(t, err)
	require.InDelta(t, 10.0, second[0].Command.Args[0].Number, 1e-9)
}

func TestMacroRegistryLookupExpand(t *testing.T) {
	t.Parallel()
	reg := gcode.NewMacroRegistry().
		Register(gcode.NewSimpleMacro("preheat",
			gcode.NewLine("M104").ArgF("S", 200),
		))

	m, ok := reg.Lookup("preheat")
	require.True(t, ok)
	require.Equal(t, "preheat", m.Name())

	expanded, err := reg.Expand("preheat", nil)
	require.NoError(t, err)
	require.Len(t, expanded, 1)
	require.Equal(t, "M104", expanded[0].Command.Name)
}

func TestMacroRegistryUnknown(t *testing.T) {
	t.Parallel()
	reg := gcode.NewMacroRegistry()
	_, err := reg.Expand("nope", nil)
	require.Error(t, err)
}

type customMacro struct{ name string }

func (m *customMacro) Name() string { return m.name }
func (m *customMacro) Expand(args map[string]float64) ([]gcode.Line, error) {
	x, ok := args["X"]
	if !ok {
		return nil, fmt.Errorf("missing required arg X")
	}
	return []gcode.Line{gcode.NewLine("G1").ArgF("X", x)}, nil
}

func TestCustomMacro(t *testing.T) {
	t.Parallel()
	reg := gcode.NewMacroRegistry().Register(&customMacro{name: "move-x"})

	expanded, err := reg.Expand("move-x", map[string]float64{"X": 42})
	require.NoError(t, err)
	require.InDelta(t, 42.0, expanded[0].Command.Args[0].Number, 1e-9)

	_, err = reg.Expand("move-x", nil)
	require.Error(t, err)
}