Add dry_run support to transaction broadcasting

emre · Oct 11, 2018 · 6492045 · 6492045
1 parent 5754f03
commit 6492045
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/lightsteem/broadcast/transaction_builder.py b/lightsteem/broadcast/transaction_builder.py
@@ -111,7 +111,8 @@ def _is_canonical(self, sig):
                 and not (sig[32] & 0x80)
                 and not (sig[32] == 0 and not (sig[33] & 0x80)))
 
-    def broadcast(self, operations, chain=None):
+    def broadcast(self, operations, chain=None, dry_run=False):
+        preferred_api_type = self.client.api_type
         if not isinstance(operations, list):
             operations = [operations, ]
 
@@ -192,5 +193,9 @@ def broadcast(self, operations, chain=None):
             sigs.append(hexlify(sigstr).decode('ascii'))
 
         self.transaction["signatures"] = sigs
+        self.client.api_type = preferred_api_type
+
+        if dry_run:
+            return self.transaction
 
         return self.client.broadcast_transaction(self.transaction)
diff --git a/lightsteem/client.py b/lightsteem/client.py
@@ -161,8 +161,9 @@ def process_batch(self):
             self.queue = []
         return resp
 
-    def broadcast(self, op):
-        return self.transaction_builder.broadcast(op, chain=self.chain)
+    def broadcast(self, op, dry_run=False):
+        return self.transaction_builder.broadcast(
+            op, chain=self.chain, dry_run=dry_run)
 
     def account(self, username):
         return Account(self, username)