test: fixing unittests

PaloAltoNetworks · Dec 31, 2021 · 50fc449 · 50fc449
1 parent c13eb9a
commit 50fc449
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/testdata/testdata.go b/testdata/testdata.go
@@ -206,7 +206,8 @@ func (c *MockClient) VsysUnimport(ns, tmpl, ts string, names []string) error {
 	return c.UnimportError
 }
 
-func (c *MockClient) WaitForJob(a uint, d time.Duration, resp interface{}) error {
+func (c *MockClient) WaitForJob(a uint, d time.Duration, extras, resp interface{}) error {
+	c.Extras = extras
 	_, err := c.finalize(resp)
 	return err
 }