fix the double click for next page

Author: shuoweil

packages/bigframes/bigframes/display/anywidget.py

@@ -256,6 +256,7 @@ def update_ui():
                             self._batch_iter = batch_iter
                             self._cached_batches = cached_batches
                             self._all_data_loaded = all_data_loaded
+                            self._last_sort_state = _SortState((), ())
                             self.row_count = total_rows
                             self.table_html = initial_html
                             self.is_deferred_mode = False

packages/bigframes/notebooks/streaming/streaming_dataframe.ipynb

@@ -4,8 +4,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "# BigFrames StreamingDataFrame",
-    "bigframes.streaming.StreamingDataFrame is a special DataFrame type that allows simple operations and can create streaming jobs to process real-time data and reverse ETL output to Bigtable and Pub/Sub using [BigQuery continuous queries](https://cloud.google.com/bigquery/docs/continuous-queries-introduction).\n",
+    "# BigFrames StreamingDataFramebigframes.streaming.StreamingDataFrame is a special DataFrame type that allows simple operations and can create streaming jobs to process real-time data and reverse ETL output to Bigtable and Pub/Sub using [BigQuery continuous queries](https://cloud.google.com/bigquery/docs/continuous-queries-introduction).\n",
     "\n",
     "In this notebook, we will:\n",
     "* Create a StreamingDataFrame from a BigQuery table\n",
@@ -31,6 +30,7 @@
    ],
    "source": [
     "import bigframes\n",
+    "\n",
     "# make sure bigframes version >= 1.12.0\n",
     "bigframes.__version__"
    ]
@@ -43,6 +43,7 @@
    "source": [
     "import bigframes.pandas as bpd\n",
     "import bigframes.streaming as bst\n",
+    "\n",
     "bigframes.options._bigquery_options.project = \"bigframes-load-testing\" # Change to your own project ID\n",
     "job_id_prefix = \"test_streaming_\""
    ]
@@ -97,8 +98,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "## Create, select, filter and preview",
-    "Create the StreamingDataFrame from a BigQuery table, select certain columns, filter rows and preview the output"
+    "## Create, select, filter and previewCreate the StreamingDataFrame from a BigQuery table, select certain columns, filter rows and preview the output"
    ]
   },
   {
@@ -559,7 +559,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "venv",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
    "name": "python3"
   },
@@ -573,9 +573,9 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.12.1"
+   "version": "3.10.19"
   }
  },
  "nbformat": 4,
- "nbformat_minor": 2
+ "nbformat_minor": 4
 }

packages/bigframes/tests/js/table_widget_angular.test.js

@@ -141,6 +141,7 @@ describe('TableWidgetAngular', () => {
 
       // Act (Click Run Query)
       runButton.click();
+      await new Promise((resolve) => setTimeout(resolve, 50));
 
       // Assert (Execution requested)
       expect(model.set).toHaveBeenCalledWith('start_execution', true);

packages/bigframes/tests/unit/display/test_anywidget.py

@@ -485,3 +485,30 @@ def test_deferred_mode_execution_error(mock_deferred_df):
 
         assert widget.is_deferred_mode is True
         assert widget._error_message == "Query Failed"
+
+
+def test_deferred_mode_execution_does_not_reset_page_on_navigation(
+    mock_deferred_df, mock_df_deferred
+):
+    from bigframes.display.anywidget import TableWidget
+
+    mock_deferred_df.execute.return_value = mock_df_deferred
+
+    batches = mock.MagicMock()
+    batch_df = pd.DataFrame({"A": [1], "B": ["a"], "C": [1.0], "D": [True]})
+    batches.__iter__.return_value = iter([batch_df])
+    batches.total_rows = 50
+    mock_df_deferred.to_pandas_batches.return_value = batches
+
+    with bigframes.option_context("display.render_mode", "anywidget"):
+        widget = TableWidget(mock_deferred_df)
+        widget.page_size = 10
+        widget.start_execution = True
+
+        thread = getattr(widget, "_execution_thread", None)
+        if thread is not None:
+            thread.join(timeout=5)
+
+        assert widget.page == 0
+        widget.page = 1
+        assert widget.page == 1