chore: fix tests

flanksource · Dec 28, 2023 · 293e650 · 293e650
1 parent d9a318b
commit 293e650
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/tests/job_test.go b/tests/job_test.go
@@ -4,7 +4,6 @@ import (
 	"sync/atomic"
 	"time"
 
-	"github.com/flanksource/duty"
 	"github.com/flanksource/duty/context"
 	"github.com/flanksource/duty/job"
 	"github.com/flanksource/duty/models"
@@ -30,8 +29,8 @@ var _ = Describe("Job", Ordered, func() {
 				return nil
 			},
 		}
-		_ = duty.UpdateProperty(ctx, "test.trace", "true")
-		_ = duty.UpdateProperty(ctx, "test.db.level", "trace")
+		_ = context.UpdateProperty(ctx, "test.trace", "true")
+		_ = context.UpdateProperty(ctx, "test.db.level", "trace")
 
 		sampleJob.Run()
 		Expect(sampleJob.Retention.Success).To(Equal(3))

diff --git a/tests/properties_test.go b/tests/properties_test.go
@@ -1,14 +1,14 @@
 package tests
 
 import (
-	"github.com/flanksource/duty"
+	"github.com/flanksource/duty/context"
 	. "github.com/onsi/ginkgo/v2"
 	. "github.com/onsi/gomega"
 )
 
 var _ = Describe("Properties", func() {
 	It("Should save properties to db", func() {
-		err := duty.UpdateProperties(DefaultContext, map[string]string{
+		err := context.UpdateProperties(DefaultContext, map[string]string{
 			"john":  "doe",
 			"hello": "world",
 		})