store reference to manager in all txns; add origin_information()

1 files changed, 6 insertions, 3 deletions

diff --git a/tests/test_transaction.py b/tests/test_transaction.py
index 3dcaba0..bf7b130 100644
--- a/tests/test_transaction.py
+++ b/tests/test_transaction.py
@@ -18,14 +18,17 @@ class DB(dns.transaction.TransactionManager):
         self.rdatasets = {}
 
     def reader(self):
-        return Transaction(False, True, self)
+        return Transaction(self, False, True)
 
     def writer(self, replacement=False):
-        return Transaction(replacement, False, self)
+        return Transaction(self, replacement, False)
+
+    def origin_information(self):
+        return (None, True)
 
 
 class Transaction(dns.transaction.Transaction):
-    def __init__(self, replacement, read_only, db):
+    def __init__(self, db, replacement, read_only):
         super().__init__(replacement)
         self.db = db
         self.rdatasets = {}