ui/dataset: Only reset collection index on actual dataset update

CHANGELOG.md

@@ -38,6 +38,10 @@ The types of changes are:
 
 * Add unlinked docs and fix any remaining broken links [#1266](https://github.com/ethyca/fides/pull/1266)
 
+### Fixed
+
+* After editing a dataset, the table will stay on the previously selected collection instead of resetting to the first one. [#1511](https://github.com/ethyca/fides/pull/1511)
+
 ## [1.9.2](https://github.com/ethyca/fides/compare/1.9.1...1.9.2)
 
 ### Deprecated

clients/admin-ui/cypress/e2e/datasets.cy.ts

@@ -160,15 +160,31 @@ describe("Dataset", () => {
     it("Can edit dataset fields", () => {
       const newDescription = "new description";
       cy.visit("/dataset/demo_users_dataset");
-      cy.getByTestId("field-row-uuid").click();
+      cy.getByTestId("collection-select").select("products");
+      cy.getByTestId("field-row-name").click();
       cy.getByTestId("input-description").clear().type(newDescription);
+
+      // Updating the dataset will trigger a refresh of the GET request.
+      cy.fixture("dataset.json").then((draftDataset) => {
+        draftDataset.collections[1].fields[1].description = newDescription;
+        cy.intercept("GET", "/api/v1/dataset/*", {
+          body: { ...draftDataset },
+        }).as("getDataset");
+      });
+
       cy.getByTestId("save-btn").click({ force: true });
+
       cy.wait("@putDataset").then((interception) => {
         const { body } = interception.request;
-        expect(body.collections[0].fields[5].description).to.eql(
+        expect(body.collections[1].fields[1].description).to.eql(
           newDescription
         );
       });
+      cy.wait("@getDataset");
+
+      // The same dataset that was being edited should be shown in the table, updated.
+      cy.getByTestId("edit-drawer-content").should("not.exist");
+      cy.getByTestId("field-row-name").should("contain", newDescription);
     });
 
     it("Can edit dataset collections", () => {

clients/admin-ui/src/features/dataset/DatasetCollectionView.tsx

@@ -71,7 +71,6 @@ const DatasetCollectionView = ({ fidesKey }: Props) => {
   useEffect(() => {
     if (dataset) {
       dispatch(setActiveDatasetFidesKey(dataset.fides_key));
-      dispatch(setActiveCollectionIndex(0));
     }
   }, [dispatch, dataset]);
 

clients/admin-ui/src/features/dataset/dataset.slice.ts

@@ -100,7 +100,7 @@ export const datasetSlice = createSlice({
 
       // Clear out the related fields when the dataset is changed.
       draftState.activeDatasetFidesKey = action.payload;
-      draftState.activeCollectionIndex = undefined;
+      draftState.activeCollectionIndex = 0;
       draftState.activeFieldIndex = undefined;
     },
     setActiveCollectionIndex: (