developgo
diff --git a/‎client/driver_test.go‎
Lines changed: 7 additions & 2 deletions b/‎client/driver_test.go‎
Lines changed: 7 additions & 2 deletions
diff --git a/‎cmd/cql-observer/observation_test.go‎
Lines changed: 0 additions & 3 deletions b/‎cmd/cql-observer/observation_test.go‎
Lines changed: 0 additions & 3 deletions
@@ -23,6 +23,7 @@ import (
 	"testing"
 	"time"
 
+	bp "github.com/CovenantSQL/CovenantSQL/blockproducer"
 	"github.com/CovenantSQL/CovenantSQL/crypto"
 	"github.com/CovenantSQL/CovenantSQL/crypto/asymmetric"
 	"github.com/CovenantSQL/CovenantSQL/crypto/kms"
@@ -68,10 +69,13 @@ func TestCreate(t *testing.T) {
 		var dsn string
 		dsn, err = Create(ResourceMeta{})
 		So(err, ShouldBeNil)
+		dsnCfg, err := ParseDSN(dsn)
+		So(err, ShouldBeNil)
 
 		waitCtx, cancelWait := context.WithTimeout(context.Background(), time.Nanosecond)
 		defer cancelWait()
-		err = WaitDBCreation(waitCtx, dsn)
+		// should not use client.WaitDBCreation, sql.Open is not supported in this test case
+		err = bp.WaitDatabaseCreation(waitCtx, proto.DatabaseID(dsnCfg.DatabaseID), nil, 3*time.Second)
 		So(err, ShouldResemble, context.DeadlineExceeded)
 
 		// Calculate database ID
@@ -92,7 +96,8 @@ func TestCreate(t *testing.T) {
 
 		waitCtx2, cancelWait2 := context.WithTimeout(context.Background(), 5*time.Minute)
 		defer cancelWait2()
-		err = WaitDBCreation(waitCtx2, dsn)
+		// should not use client.WaitDBCreation, sql.Open is not supported in this test case
+		err = bp.WaitDatabaseCreation(waitCtx2, proto.DatabaseID(dsnCfg.DatabaseID), nil, 3*time.Second)
 		So(err, ShouldBeNil)
 	})
 }
 
@@ -304,9 +304,6 @@ func TestFullProcess(t *testing.T) {
 		err = client.WaitDBCreation(ctx2, dsn)
 		So(err, ShouldBeNil)
 
-		_, err = db.ExecContext(ctx2, "SHOW TABLES")
-		So(err, ShouldBeNil)
-
 		_, err = db.Exec("CREATE TABLE test (test int)")
 		So(err, ShouldBeNil)