RDBC-768 NextAndSeedIdentitiesTest::seedIdentityFor

Author: poissoncorp

ravendb/documents/operations/identities.py

@@ -64,3 +64,55 @@ def __init__(self, name: str):
 
     def get_command(self, conventions: "DocumentConventions") -> "RavenCommand[int]":
         return NextIdentityForCommand(self._identity_name)
+
+
+class SeedIdentityForCommand(RavenCommand[int], RaftCommand):
+    def __init__(self, id_: str, value: int = None, forced: bool = False):
+        super().__init__(int)
+        if id_ is None:
+            raise ValueError("Id cannot be None")
+
+        self._id = id_
+        self._value = value
+        self._forced = forced
+
+    def is_read_request(self) -> bool:
+        return False
+
+    def create_request(self, node: ServerNode) -> requests.Request:
+        self.ensure_is_not_null_or_string(self._id, "id")
+
+        url = f"{node.url}/databases/{node.database}/identity/seed?name={self._id}&value={self._value}"
+
+        if self._forced:
+            url += "&force=true"
+
+        return requests.Request("POST", url)
+
+    def set_response(self, response: Optional[str], from_cache: bool) -> None:
+        if response is None:
+            self._throw_invalid_response()
+
+        json_node = json.loads(response)
+
+        if "NewSeedValue" not in json_node:
+            self._throw_invalid_response()
+
+        self.result = json_node["NewSeedValue"]
+
+    def get_raft_unique_request_id(self) -> str:
+        return RaftIdGenerator.new_id()
+
+
+class SeedIdentityForOperation(MaintenanceOperation[int]):
+    def __init__(self, name: str, value: int, force_update: bool = False):
+        super().__init__()
+        if not name or name.isspace():
+            raise ValueError("The field name cannot be None or whitespace")
+
+        self._identity_name = name
+        self._identity_value = value
+        self._force_update = force_update
+
+    def get_command(self, conventions: "DocumentConventions") -> "RavenCommand[_T]":
+        return SeedIdentityForCommand(self._identity_name, self._identity_value, self._force_update)

ravendb/tests/jvm_migrated_tests/client_tests/test_next_and_seed_identities.py

@@ -1,4 +1,5 @@
-from ravendb.documents.operations.identities import NextIdentityForOperation
+from ravendb.documents.operations.identities import NextIdentityForOperation, SeedIdentityForOperation
+from ravendb.infrastructure.entities import User
 from ravendb.tests.test_base import TestBase
 
 
@@ -11,3 +12,39 @@ def test_next_identity_for_operation_should_create_a_new_identify_if_there_is_no
             result = self.store.maintenance.send(NextIdentityForOperation("person|"))
 
             self.assertEqual(1, result)
+
+    def test_seed_identity_for(self):
+        with self.store.open_session() as session:
+            user = User(last_name="Adi")
+            session.store(user, "users|")
+            session.save_changes()
+
+        result1 = self.store.maintenance.send(SeedIdentityForOperation("users", 1990))
+        self.assertEqual(1990, result1)
+
+        with self.store.open_session() as session:
+            user = User(last_name="Avivi")
+            session.store(user, "users|")
+            session.save_changes()
+
+        with self.store.open_session() as session:
+            entity_with_id_1 = session.load("users/1", User)
+            entity_with_id_2 = session.load("users/2", User)
+            entity_with_id_1990 = session.load("users/1990", User)
+            entity_with_id_1991 = session.load("users/1991", User)
+            entity_with_id_1992 = session.load("users/1992", User)
+
+            self.assertIsNotNone(entity_with_id_1)
+            self.assertIsNotNone(entity_with_id_1991)
+            self.assertIsNone(entity_with_id_2)
+            self.assertIsNone(entity_with_id_1990)
+            self.assertIsNone(entity_with_id_1992)
+
+            self.assertEqual("Adi", entity_with_id_1.last_name)
+            self.assertEqual("Avivi", entity_with_id_1991.last_name)
+
+        result2 = self.store.maintenance.send(SeedIdentityForOperation("users", 1975))
+        self.assertEqual(1991, result2)
+
+        result3 = self.store.maintenance.send(SeedIdentityForOperation("users", 1975, True))
+        self.assertEqual(1975, result3)