fix(sample): change update entity sample to use transaction (#1633)

googleapis · Oct 28, 2024 · 17b411f · 17b411f
1 parent 4461f5b
commit 17b411f
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/com/google/datastore/snippets/ConceptsTest.java b/com/google/datastore/snippets/ConceptsTest.java
@@ -278,8 +278,17 @@ public void testLookup() {
   public void testUpdate() {
     datastore.put(testEntity);
     // [START datastore_update]
-    Entity task = Entity.newBuilder(datastore.get(taskKey)).set("priority", 5).build();
-    datastore.update(task);
+    Entity task;
+    Transaction txn = datastore.newTransaction();
+    try {
+      task = Entity.newBuilder(txn.get(taskKey)).set("priority", 5).build();
+      txn.put(task);
+      txn.commit();
+    } finally {
+      if (txn.isActive()) {
+        txn.rollback();
+      }
+    }
     // [END datastore_update]
     assertEquals(task, datastore.get(taskKey));
   }