Function assertRows

output/sqlite_test.go:108–154

(t *testing.T, fname string, want [][]string)

}

func assertRows(t *testing.T, fname string, want [][]string) {
	// this helper asserts we're going to find these rows and records in
	// the sqlite database at the given path, in a table called 'lines'.
	t.Helper()

	conn, err := sql.Open("sqlite3", fname)
	if err != nil {
		t.Fatalf("sqlite database %q not found: %s", fname, err)
	}
	defer conn.Close()

	rows, err := conn.Query("SELECT * FROM lines")
	if err != nil {
		t.Fatalf("failed select: %s", err)
	}
	defer rows.Close()

	cols := make([]interface{}, len(want[0]))
	for i := range cols {
		cols[i] = new(string)
	}

	irow := 0
	for rows.Next() {
		if irow+1 >= len(want) {
			// There are more rows than we expected, anyway consume and count
			// them all to report the actual number in the error message.
			irow++
			continue
		}

		if err := rows.Scan(cols...); err != nil {
			t.Fatalf("failed scan: %s", err)
		}
		for i := range cols {
			col := *(cols[i].(*string))
			if want[irow][i] != col {
				t.Errorf("row[%d][%d] = %q, want %q", irow, i, col, want[irow][i])
			}
		}
		irow++
	}

	if irow != len(want) {
		t.Errorf("got %d total rows, want %d", irow, len(want))
	}
}

func TestSQLiteInsertRows(t *testing.T) { testSQLiteInsertRows(t, false) }