lang: core, funcs, types: Add ctx to simple func
Plumb through the standard context.Context so that a function can be cancelled if someone requests this. It makes it less awkward to write simple functions that might depend on io or network access.
This commit is contained in:
@@ -30,6 +30,7 @@
|
||||
package corenet
|
||||
|
||||
import (
|
||||
"context"
|
||||
"testing"
|
||||
|
||||
"github.com/purpleidea/mgmt/lang/types"
|
||||
@@ -51,7 +52,7 @@ func TestMacFmt(t *testing.T) {
|
||||
for _, tt := range tests {
|
||||
tt := tt
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
m, err := MacFmt([]types.Value{&types.StrValue{V: tt.in}})
|
||||
m, err := MacFmt(context.Background(), []types.Value{&types.StrValue{V: tt.in}})
|
||||
if (err != nil) != tt.wantErr {
|
||||
t.Errorf("func MacFmt() error = %v, wantErr %v", err, tt.wantErr)
|
||||
return
|
||||
@@ -81,7 +82,7 @@ func TestOldMacFmt(t *testing.T) {
|
||||
for _, tt := range tests {
|
||||
tt := tt
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
m, err := OldMacFmt([]types.Value{&types.StrValue{V: tt.in}})
|
||||
m, err := OldMacFmt(context.Background(), []types.Value{&types.StrValue{V: tt.in}})
|
||||
if (err != nil) != tt.wantErr {
|
||||
t.Errorf("func MacFmt() error = %v, wantErr %v", err, tt.wantErr)
|
||||
return
|
||||
|
||||
Reference in New Issue
Block a user