RDBC-769 BulkInsertCountersTest::addDocumentAfterIncrementCounter

Author: poissoncorp

ravendb/documents/bulk_insert_operation.py

@@ -540,7 +540,7 @@ def increment(self, id_: str, name: str, delta: int = 1) -> None:
 
                         self._operation._first = False
 
-                        _id = self._id
+                        self._id = id_
                         self._operation._in_progress_command = CommandType.COUNTERS
 
                         self._write_prefix_for_new_command()

ravendb/tests/jvm_migrated_tests/client_tests/counters_tests/test_bulk_insert_counters.py

@@ -0,0 +1,43 @@
+from ravendb.documents.operations.counters import GetCountersOperation
+from ravendb.infrastructure.entities import User
+from ravendb.tests.test_base import TestBase
+
+
+class TestBulkInsertCounters(TestBase):
+    def setUp(self):
+        super().setUp()
+
+    def test_add_document_after_increment_counter(self):
+        with self.store.bulk_insert() as bulk_insert:
+            user1 = User()
+            user1.name = "Grisha"
+            bulk_insert.store(user1)
+
+            user_id_1 = user1.Id
+
+        with self.store.bulk_insert() as bulk_insert:
+            counter = bulk_insert.counters_for(user_id_1)
+
+            counter.increment("likes", 100)
+            counter.increment("downloads", 500)
+
+            user2 = User(name="Kotler")
+            bulk_insert.store(user2)
+
+            user_id_2 = user2.Id
+
+            bulk_insert.counters_for(user_id_2).increment("votes", 1000)
+
+        counters = self.store.operations.send(GetCountersOperation(user_id_1, ["likes", "downloads"])).counters
+
+        self.assertEqual(2, len(counters))
+
+        counters.sort(key=lambda counter: counter.counter_name)
+
+        self.assertEqual(500, counters[0].total_value)
+
+        self.assertEqual(100, counters[1].total_value)
+
+        val = self.store.operations.send(GetCountersOperation(user_id_2, "votes"))
+
+        self.assertEqual(1000, val.counters[0].total_value)