From 5183459c36ca027bb27ec424dc9f2b47378f90c4 Mon Sep 17 00:00:00 2001
From: Wroud <wrouds@gmail.com>
Date: Fri, 19 Feb 2021 20:03:23 +0300
Subject: [PATCH] fix(plugin-data-viewer): CB-762 cell revert behavior

---
 .../src/DatabaseDataModel/DatabaseDataEditor.ts             | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/webapp/packages/plugin-data-viewer/src/DatabaseDataModel/DatabaseDataEditor.ts b/webapp/packages/plugin-data-viewer/src/DatabaseDataModel/DatabaseDataEditor.ts
index 4b73cc30b3..cbf0d44fef 100644
--- a/webapp/packages/plugin-data-viewer/src/DatabaseDataModel/DatabaseDataEditor.ts
+++ b/webapp/packages/plugin-data-viewer/src/DatabaseDataModel/DatabaseDataEditor.ts
@@ -200,7 +200,11 @@ export class DatabaseDataEditor<TResult extends IDatabaseDataResult> implements
     }
 
     if (!this.isRowEdited(result, row)) {
-      this.editedResults.delete(result.id);
+      update?.diff.delete(row);
+
+      if (update?.diff.size === 0) {
+        this.editedResults.delete(result.id);
+      }
     }
   }