diff --git a/pkg/sources/forward/data_forward_test.go b/pkg/sources/forward/data_forward_test.go index c90e13fb0d..6a10b3db7d 100644 --- a/pkg/sources/forward/data_forward_test.go +++ b/pkg/sources/forward/data_forward_test.go @@ -26,6 +26,9 @@ import ( "go.uber.org/goleak" + "github.com/prometheus/client_golang/prometheus/testutil" + "github.com/stretchr/testify/assert" + dfv1 "github.com/numaproj/numaflow/pkg/apis/numaflow/v1alpha1" "github.com/numaproj/numaflow/pkg/forward" "github.com/numaproj/numaflow/pkg/isb" @@ -36,9 +39,6 @@ import ( "github.com/numaproj/numaflow/pkg/watermark/generic" wmstore "github.com/numaproj/numaflow/pkg/watermark/store" "github.com/numaproj/numaflow/pkg/watermark/wmb" - - "github.com/prometheus/client_golang/prometheus/testutil" - "github.com/stretchr/testify/assert" ) const ( diff --git a/pkg/sources/forward/shutdown_test.go b/pkg/sources/forward/shutdown_test.go index 02ea94b9ee..4dfb21f8fa 100644 --- a/pkg/sources/forward/shutdown_test.go +++ b/pkg/sources/forward/shutdown_test.go @@ -21,14 +21,14 @@ import ( "testing" "time" + "github.com/stretchr/testify/assert" + dfv1 "github.com/numaproj/numaflow/pkg/apis/numaflow/v1alpha1" "github.com/numaproj/numaflow/pkg/forward" "github.com/numaproj/numaflow/pkg/isb" "github.com/numaproj/numaflow/pkg/isb/stores/simplebuffer" "github.com/numaproj/numaflow/pkg/isb/testutils" "github.com/numaproj/numaflow/pkg/watermark/generic" - - "github.com/stretchr/testify/assert" ) type myShutdownTest struct { diff --git a/pkg/sources/generator/tickgen.go b/pkg/sources/generator/tickgen.go index d969753cd5..ed7f3c3c15 100644 --- a/pkg/sources/generator/tickgen.go +++ b/pkg/sources/generator/tickgen.go @@ -27,14 +27,13 @@ import ( "go.uber.org/zap" - sourceforward "github.com/numaproj/numaflow/pkg/sources/forward" - applier2 "github.com/numaproj/numaflow/pkg/sources/forward/applier" - dfv1 "github.com/numaproj/numaflow/pkg/apis/numaflow/v1alpha1" "github.com/numaproj/numaflow/pkg/forward" "github.com/numaproj/numaflow/pkg/isb" "github.com/numaproj/numaflow/pkg/metrics" "github.com/numaproj/numaflow/pkg/shared/logging" + sourceforward "github.com/numaproj/numaflow/pkg/sources/forward" + applier2 "github.com/numaproj/numaflow/pkg/sources/forward/applier" "github.com/numaproj/numaflow/pkg/watermark/fetch" "github.com/numaproj/numaflow/pkg/watermark/processor" "github.com/numaproj/numaflow/pkg/watermark/publish" diff --git a/pkg/sources/generator/tickgen_test.go b/pkg/sources/generator/tickgen_test.go index c0e016ba89..8449d48bf6 100644 --- a/pkg/sources/generator/tickgen_test.go +++ b/pkg/sources/generator/tickgen_test.go @@ -21,6 +21,9 @@ import ( "testing" "time" + "github.com/stretchr/testify/assert" + v1 "k8s.io/apimachinery/pkg/apis/meta/v1" + dfv1 "github.com/numaproj/numaflow/pkg/apis/numaflow/v1alpha1" "github.com/numaproj/numaflow/pkg/forward" "github.com/numaproj/numaflow/pkg/isb" @@ -28,9 +31,6 @@ import ( "github.com/numaproj/numaflow/pkg/sources/forward/applier" "github.com/numaproj/numaflow/pkg/watermark/generic" "github.com/numaproj/numaflow/pkg/watermark/store" - - "github.com/stretchr/testify/assert" - v1 "k8s.io/apimachinery/pkg/apis/meta/v1" ) type myForwardToAllTest struct { diff --git a/pkg/sources/kafka/handler.go b/pkg/sources/kafka/handler.go index 885a4b22f1..ba6354b85c 100644 --- a/pkg/sources/kafka/handler.go +++ b/pkg/sources/kafka/handler.go @@ -22,6 +22,7 @@ import ( "go.uber.org/zap" "github.com/IBM/sarama" + "github.com/numaproj/numaflow/pkg/shared/logging" )