fix: make change in capture response method to record tests (#106)

Signed-off-by: re-Tick <jain.ritik.1001@gmail.com> Signed-off-by: re-Tick <jain.ritik.1001@gmail.com>
keploy · Aug 18, 2022 · 1c99aed · 1c99aed
1 parent 02725df
commit 1c99aed
Show file tree

Hide file tree

Showing 4 changed files with 52 additions and 30 deletions.
diff --git a/integrations/ksql/connBeginTx.go b/integrations/ksql/connBeginTx.go
@@ -36,7 +36,9 @@ func (t Tx) Commit() error {
 	case "test":
 		// don't run
 	case "capture":
-		err = t.Tx.Commit()
+		if t.Tx != nil {
+			err = t.Tx.Commit()
+		}
 	default:
 		return errors.New("integrations: Not in a valid sdk mode")
 	}
@@ -79,7 +81,9 @@ func (t Tx) Rollback() error {
 	case "test":
 		// don't run
 	case "capture":
-		err = t.Tx.Rollback()
+		if t.Tx != nil {
+			err = t.Tx.Rollback()
+		}
 	default:
 		return errors.New("integrations: Not in a valid sdk mode")
 	}

diff --git a/integrations/ksql/connPrepareContext.go b/integrations/ksql/connPrepareContext.go
@@ -45,13 +45,17 @@ func (s Stmt) Exec(args []driver.Value) (driver.Result, error) {
 	case "test":
 		//don't call method
 	case "capture":
-		result, err = s.Stmt.Exec(args)
-		l, e := result.LastInsertId()
-		drResult.LastInserted = l
-		drResult.LError = e.Error()
-		ra, e := result.RowsAffected()
-		drResult.RowsAff = ra
-		drResult.RError = e.Error()
+		if s.Stmt != nil {
+			result, err = s.Stmt.Exec(args)
+			if result != nil {
+				l, e := result.LastInsertId()
+				drResult.LastInserted = l
+				drResult.LError = e.Error()
+				ra, e := result.RowsAffected()
+				drResult.RowsAff = ra
+				drResult.RError = e.Error()
+			}
+		}
 	default:
 		return nil, err
 	}
@@ -94,8 +98,10 @@ func (s Stmt) Query(args []driver.Value) (driver.Rows, error) {
 	case "test":
 		// don't run
 	case "capture":
-		rows, err = s.Stmt.Query(args)
-		drRows.Rows = rows
+		if s.Stmt != nil {
+			rows, err = s.Stmt.Query(args)
+			drRows.Rows = rows
+		}
 	default:
 		return nil, errors.New("integrations: Not in a valid sdk mode")
 	}
@@ -138,8 +144,10 @@ func (s Stmt) NumInput() int {
 	case "test":
 		// don't run
 	case "capture":
-		o := s.Stmt.NumInput()
-		output = &o
+		if s.Stmt != nil {
+			o := s.Stmt.NumInput()
+			output = &o
+		}
 	default:
 		return 0
 	}
@@ -171,7 +179,9 @@ func (s Stmt) Close() error {
 	case "test":
 		// don't run
 	case "capture":
-		err = s.Stmt.Close()
+		if s.Stmt != nil {
+			err = s.Stmt.Close()
+		}
 	default:
 		return errors.New("integrations: Not in a valid sdk mode")
 	}

diff --git a/integrations/ksql/execerContext.go b/integrations/ksql/execerContext.go
@@ -61,17 +61,19 @@ func (c Conn) ExecContext(ctx context.Context, query string, args []driver.Named
 		//don't call Find method
 	case "capture":
 		result, err = execerContext.ExecContext(ctx, query, args)
-		// calls LastInsertId to capture their outputs
-		li, e := result.LastInsertId()
-		driverResult.LastInserted = li
-		if e != nil {
-			driverResult.LError = e.Error()
-		}
-		// calls RowsAffected to capture their outputs
-		ra, e := result.RowsAffected()
-		driverResult.RowsAff = ra
-		if e != nil {
-			driverResult.RError = e.Error()
+		if result != nil {
+			// calls LastInsertId to capture their outputs
+			li, e := result.LastInsertId()
+			driverResult.LastInserted = li
+			if e != nil {
+				driverResult.LError = e.Error()
+			}
+			// calls RowsAffected to capture their outputs
+			ra, e := result.RowsAffected()
+			driverResult.RowsAff = ra
+			if e != nil {
+				driverResult.RError = e.Error()
+			}
 		}
 	default:
 		return result, err

diff --git a/integrations/ksql/queryerContext.go b/integrations/ksql/queryerContext.go
@@ -40,8 +40,10 @@ func (r Rows) Columns() []string {
 	case "test":
 		// don't run
 	case "capture":
-		o := r.Rows.Columns()
-		output = &o
+		if r.Rows != nil {
+			o := r.Rows.Columns()
+			output = &o
+		}
 	default:
 		return nil
 	}
@@ -80,7 +82,9 @@ func (r Rows) Close() error {
 			return nil
 		}
 	case "capture":
-		err = r.Rows.Close()
+		if r.Rows != nil {
+			err = r.Rows.Close()
+		}
 	default:
 		return errors.New("integrations: Not in a valid sdk mode")
 	}
@@ -237,8 +241,10 @@ func (r Rows) Next(dest []driver.Value) error {
 	case "test":
 		// don't run
 	case "capture":
-		err = r.Rows.Next(dest)
-		output.Value = dest
+		if r.Rows != nil {
+			err = r.Rows.Next(dest)
+			output.Value = dest
+		}
 	default:
 		return errors.New("integrations: Not in a valid sdk mode")
 	}