Add python support, move test to python unittest

python/tests/test_app.py

@@ -0,0 +1,103 @@
+import os
+import unittest
+import json
+
+import web3
+import eth_tester
+
+from eth_accounts_index import Registry
+
+testdir = os.path.dirname(__file__)
+
+
+class Test(unittest.TestCase):
+
+    def setUp(self):
+        eth_params = eth_tester.backends.pyevm.main.get_default_genesis_params({
+            'gas_limit': 9000000,
+            })
+
+        f = open(os.path.join(testdir, '../eth_accounts_index/data/registry.bin'), 'r')
+        bytecode = f.read()
+        f.close()
+
+        f = open(os.path.join(testdir, '../eth_accounts_index/data/registry.abi.json'), 'r')
+        abi = json.load(f)
+        f.close()
+
+        backend = eth_tester.PyEVMBackend(eth_params)
+        self.eth_tester = instance = eth_tester.EthereumTester(backend)
+        provider = web3.Web3.EthereumTesterProvider(instance)
+        self.w3 = web3.Web3(provider)
+        c = self.w3.eth.contract(abi=abi, bytecode=bytecode)
+        tx_hash = c.constructor().transact({'from': self.w3.eth.accounts[0]})
+
+        r = self.w3.eth.getTransactionReceipt(tx_hash)
+
+        self.address = r.contractAddress
+        c = self.w3.eth.contract(abi=abi, address=self.address)
+
+        c.functions.addWriter(self.w3.eth.accounts[1]).transact()
+
+
+    def tearDown(self):
+        pass
+
+
+    def test_basic(self):
+        registry = Registry(self.w3, self.address)
+        self.assertEqual(registry.count(), 1); # count starts at 1, first addess is always 0x0
+
+
+    def test_access(self):
+        registry = Registry(self.w3, self.address, self.w3.eth.accounts[1])
+        registry.add(self.w3.eth.accounts[2])
+        self.eth_tester.mine_block()
+        self.assertEqual(registry.count(), 2)
+
+        # account 0 does not have access
+        registry = Registry(self.w3, self.address, self.w3.eth.accounts[2])
+        registry.add(self.w3.eth.accounts[2])
+        self.eth_tester.mine_block()
+        self.assertEqual(registry.count(), 2)
+
+        # after this account 2 has access
+        registry.contract.functions.addWriter(self.w3.eth.accounts[2]).transact()
+        registry.add(self.w3.eth.accounts[3])
+        self.eth_tester.mine_block()
+        self.assertEqual(registry.count(), 3)
+
+        # after this account 2 no longer has access
+        registry.contract.functions.deleteWriter(self.w3.eth.accounts[2]).transact()
+        registry.add(self.w3.eth.accounts[3])
+        self.eth_tester.mine_block()
+        self.assertEqual(registry.count(), 3)
+
+
+    def test_indices(self):
+        registry = Registry(self.w3, self.address, self.w3.eth.accounts[1])
+        registry.add(self.w3.eth.accounts[2])
+
+        self.assertTrue(registry.have(self.w3.eth.accounts[2]))
+        self.assertFalse(registry.have(self.w3.eth.accounts[3]))
+
+
+    def test_list(self):
+        registry = Registry(self.w3, self.address, self.w3.eth.accounts[1])
+
+        for i in range(2, 10):
+            registry.add(self.w3.eth.accounts[i])
+
+        self.assertEqual(registry.count(), 9)
+
+        accounts_reverse = []
+        for i in range(9, 1, -1):
+            accounts_reverse.append(self.w3.eth.accounts[i])
+
+        accounts_list = registry.last(8)
+        for i in range(8):
+            self.assertEqual(accounts_list[i], accounts_reverse[i])
+
+
+if __name__ == '__main__':
+    unittest.main()