imports issue fix

rudderlabs · Mar 1, 2023 · 40d69cb · 40d69cb
1 parent fa45b5d
commit 40d69cb
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 7 deletions.
diff --git a/runner/runner.go b/runner/runner.go
@@ -18,6 +18,7 @@ import (
 	"github.com/rudderlabs/rudder-server/warehouse/integrations/deltalake"
 	"github.com/rudderlabs/rudder-server/warehouse/integrations/mssql"
 	"github.com/rudderlabs/rudder-server/warehouse/integrations/postgres"
+	postgreslegacy "github.com/rudderlabs/rudder-server/warehouse/integrations/postgres-legacy"
 	"github.com/rudderlabs/rudder-server/warehouse/integrations/redshift"
 	"github.com/rudderlabs/rudder-server/warehouse/integrations/snowflake"
 
@@ -345,6 +346,7 @@ func runAllInit() {
 	azuresynapse.Init()
 	mssql.Init()
 	postgres.Init()
+	postgreslegacy.Init()
 	redshift.Init()
 	snowflake.Init()
 	deltalake.Init()

diff --git a/warehouse/integrations/postgres-legacy/postgres_test.go b/warehouse/integrations/postgres-legacy/postgres_test.go
@@ -5,9 +5,9 @@ import (
 	"strings"
 	"testing"
 
-	"github.com/rudderlabs/rudder-server/warehouse/integrations/testhelper"
+	postgreslegacy "github.com/rudderlabs/rudder-server/warehouse/integrations/postgres-legacy"
 
-	"github.com/rudderlabs/rudder-server/warehouse/integrations/postgres"
+	"github.com/rudderlabs/rudder-server/warehouse/integrations/testhelper"
 
 	"github.com/rudderlabs/rudder-server/warehouse/tunnelling"
 
@@ -29,10 +29,10 @@ func TestIntegrationPostgresThroughTunnelling(t *testing.T) {
 	misc.Init()
 	validations.Init()
 	warehouseutils.Init()
-	postgres.Init()
+	postgreslegacy.Init()
 
 	configurations := testhelper.PopulateTemplateConfigurations()
-	db, err := postgres.Connect(postgres.Credentials{
+	db, err := postgreslegacy.Connect(postgreslegacy.Credentials{
 		DBName:   configurations["privatePostgresDatabase"],
 		Password: configurations["privatePostgresPassword"],
 		User:     configurations["privatePostgresUser"],
@@ -125,9 +125,9 @@ func TestIntegrationPostgres(t *testing.T) {
 
 	t.Parallel()
 
-	postgres.Init()
+	postgreslegacy.Init()
 
-	db, err := postgres.Connect(postgres.Credentials{
+	db, err := postgreslegacy.Connect(postgreslegacy.Credentials{
 		DBName:   "rudderdb",
 		Password: "rudder-password",
 		User:     "rudder",
@@ -234,7 +234,7 @@ func TestConfigurationValidationPostgres(t *testing.T) {
 	misc.Init()
 	validations.Init()
 	warehouseutils.Init()
-	postgres.Init()
+	postgreslegacy.Init()
 
 	configurations := testhelper.PopulateTemplateConfigurations()
 	destination := backendconfig.DestinationT{

diff --git a/warehouse/warehouse_test.go b/warehouse/warehouse_test.go
@@ -7,6 +7,8 @@ import (
 	"testing"
 	"time"
 
+	postgreslegacy "github.com/rudderlabs/rudder-server/warehouse/integrations/postgres-legacy"
+
 	"github.com/rudderlabs/rudder-server/warehouse/integrations/postgres"
 	"github.com/rudderlabs/rudder-server/warehouse/internal/repo"
 
@@ -61,6 +63,7 @@ func initWarehouse() {
 	validations.Init()
 	misc.Init()
 	postgres.Init()
+	postgreslegacy.Init()
 }
 
 func getMockStats(g GinkgoTInterface) (*mock_stats.MockStats, *mock_stats.MockMeasurement) {