Merge pull request #19 from rocklabs-io/unit-test

Unit test

Author: Myse1f

.github/workflows/test.yml

@@ -0,0 +1,28 @@
+name: test
+on
+  # Trigger the workflow on push or pull request,
+  # but only for the main branch
+  push:
+    branches:
+      - main
+  pull_request:
+    branches:
+      - main
+
+jobs:
+  test:
+    strategy:
+      matrix:
+        os: [ubuntu-latest, macos-latest]
+        python-version: [3.6, 3.7, 3.8, 3.9]
+    runs-on: ${{ matrix.os }}
+    steps:
+      - run: git config --global core.autocrlf input
+      - uses: actions/checkout@v2
+        with:
+          fetch-depth: 0
+      - uses: actions/setup-python@v2
+        with:
+          python-version: ${{ matrix.python-version }}
+      - run: pip install --upgrade tox
+      - run: tox

.gitignore

@@ -5,4 +5,6 @@ __pycache__
 test.py
 
 build/
-*.egg-info
+*.egg-info
+
+.tox/

readme.md renamed to README.md

@@ -0,0 +1,6 @@
+[build-system]
+requires = [
+  "setuptools",
+  "wheel",
+]
+build-backend = "setuptools.build_meta"

pyproject.toml

@@ -3,7 +3,7 @@
 from setuptools import setup
 from os import path
 this_directory = path.abspath(path.dirname(__file__))
-with open(path.join(this_directory, 'readme.md'), 'r') as f:
+with open(path.join(this_directory, 'README.md'), 'r') as f:
     long_description = f.read()
 
 setup(

setup.py

@@ -0,0 +1,31 @@
+from ic.agent import *
+from ic.identity import *
+from ic.client import *
+from ic.candid import Types, encode
+
+class TestAgent:
+
+    def setup_class(self):
+        client = Client()
+        iden = Identity(privkey="833fe62409237b9d62ec77587520911e9a759cec1d19755b7da901b96dca3d42")
+        self.agent = Agent(iden, client)
+
+    def test_query(self):
+        # query token totalSupply
+        ret = self.agent.query_raw("gvbup-jyaaa-aaaah-qcdwa-cai", "totalSupply", encode([]))
+        assert ret[0]['value'] == 10000000000000000
+
+        # query token name
+        ret = self.agent.query_raw("gvbup-jyaaa-aaaah-qcdwa-cai", "name", encode([]))
+        assert ret[0]['value'] == 'XTC Test'
+
+    def test_update(self):
+        ret = self.agent.update_raw(
+            "gvbup-jyaaa-aaaah-qcdwa-cai",
+            "transfer",
+            encode([
+                {'type': Types.Principal, 'value': 'aaaaa-aa'},
+                {'type': Types.Nat, 'value': 10000000000}
+                ])
+            )
+        assert ret[0]['type'] == 'rec_0'

tests/test_agent.py

@@ -0,0 +1,99 @@
+from ic.candid import *
+
+# TODO
+class TestCandid:
+
+    def test_nat_encode(self):
+        nat = NatClass()
+        res = encode([{'type':nat, 'value':10000000000}])
+        assert res.hex() == "4449444c00017d80c8afa025"
+
+    def test_nat_decode(self):
+        data = bytes.fromhex("4449444c00017d80c8afa025")
+        res = decode(data)
+        assert len(res) == 1
+        assert res[0]["type"] == 'nat'
+        assert res[0]["value"] == 10000000000
+
+    def test_principal_encode(self):
+        principal = PrincipalClass()
+        res = encode([{'type': principal, 'value':'aaaaa-aa'}])
+        assert res.hex() == "4449444c0001680100"
+    
+    def test_principal_decode(self):
+        data = bytes.fromhex("4449444c0001680100")
+        res = decode(data)
+        assert len(res) == 1
+        assert res[0]["type"] == 'principal'
+        assert res[0]["value"].to_str() == 'aaaaa-aa'
+
+    # data = b'DIDL\x00\x01q\x08XTC Test'
+    # print('decode data: {}'.format(data))
+    # out = decode(data)
+    # print(out)
+
+    # data = b'DIDL\x00\x01}\xe2\x82\xac\xe2\x82\xac\xe2\x80'
+    # print('decode data: {}'.format(data))
+    # out = decode(data)
+    # print(out)
+
+    def test_record_encode(self):
+        record = Types.Record({'foo':Types.Text, 'bar': Types.Int})
+        res = encode([{'type': record, 'value':{'foo': '💩', 'bar': 42}}])
+        assert res.hex() == '4449444c016c02d3e3aa027c868eb7027101002a04f09f92a9'
+        
+    def test_record_decode(self):
+        data = bytes.fromhex('4449444c016c02d3e3aa027c868eb7027101002a04f09f92a9')
+        res = decode(data)
+        assert len(res) == 1
+        assert res[0]["type"].startswith('rec')
+        assert res[0]['value'] == {'_4895187': 42, '_5097222': '💩'}
+
+    # def test_tuple_encode(self):
+    #     tup = Types.Tuple(Types.Int, Types.Text)
+    #     res = encode([{'type': tup, 'value': [42, '💩']}])
+    #     assert res.hex() == '4449444c016c02007c017101002a04f09f92a9'
+        
+
+    # # variant
+    # tup = Types.Variant({'ok': Types.Text, 'err': Types.Text})
+    # res = encode([{'type': tup, 'value': {'ok': 'good'} }])
+    # print('expected:', '4449444c016b03017e9cc20171e58eb4027101000104676f6f64')
+    # print('current:', res.hex())
+    # print(decode(res, tup))
+    
+    # # tuple(variant)
+    # tup = Types.Tuple(Types.Variant({'ok': Types.Text, 'err': Types.Text}))
+    # res = encode([{'type': tup, 'value': [{'ok': 'good'}] }])
+    # print('expected:', '4449444c026b029cc20171e58eb402716c01000001010004676f6f64')
+    # print('current:', res.hex())
+    # print(decode(res, tup))
+
+    # # Vec
+    # vec = Types.Vec(Types.Nat64)
+    # param = [0, 1, 2, 3]
+    # res = encode([{'type': vec, 'value': param}])
+    # print('expected:', '4449444c016d7c01000400010203')
+    # print('current :', res.hex())
+    # print('decode Vec:', decode(res, vec))
+ 
+    # # Principle
+    # Prin = Types.Principal
+    # param = 'expmt-gtxsw-inftj-ttabj-qhp5s-nozup-n3bbo-k7zvn-dg4he-knac3-lae'
+    # res = encode([{'type': Prin, 'value': param}])
+    # print('current :', res.hex())
+    # print('decode Principal:', decode(res))
+
+    # # Opt principal
+    # Prin = Types.Opt(Types.Principal)
+    # param = ['expmt-gtxsw-inftj-ttabj-qhp5s-nozup-n3bbo-k7zvn-dg4he-knac3-lae']
+    # res = encode([{'type': Prin, 'value': param}])
+    # print('current :', res.hex())
+    # print('decode Principal:', decode(res, Prin))
+    
+    # # NULL
+    # Prin = Types.Null
+    # param = None
+    # res = encode([{'type': Prin, 'value': param}])
+    # print('current :', res.hex())
+    # print('decode Null:', decode(res, Prin))

tests/test_candid.py

@@ -0,0 +1,25 @@
+from ic.identity import *
+
+class TestIdentity:
+
+    def test_ed25519_privatekey(self):
+        iden = Identity(privkey="833fe62409237b9d62ec77587520911e9a759cec1d19755b7da901b96dca3d42")
+        assert iden.key_type == 'ed25519'
+        assert iden.pubkey == 'ec172b93ad5e563bf4932c70e1245034c35467ef2efd4d64ebf819683467e2bf'
+
+    def test_secp256k1_privatekey(self):
+        pass
+
+    def test_ed25519_frompem(self):
+        pem = """
+        -----BEGIN PRIVATE KEY-----
+        MFMCAQEwBQYDK2VwBCIEIGQqNAZlORmn1k4QrYz1FvO4fOQowS3GXQMqRKDzmx9P
+        oSMDIQCrO5iGM5hnLWrHavywoXekAoXPpYRuB0Dr6DjZF6FZkg==
+        -----END PRIVATE KEY-----"""
+        iden = Identity.from_pem(pem)
+        assert iden.key_type == 'ed25519'
+        assert iden.privkey == '642a3406653919a7d64e10ad8cf516f3b87ce428c12dc65d032a44a0f39b1f4f'
+        assert iden.pubkey == 'ab3b98863398672d6ac76afcb0a177a40285cfa5846e0740ebe838d917a15992'
+
+    def test_secp256k1_frompem(self):
+        pass

tests/test_identity.py

@@ -0,0 +1,19 @@
+from ic.principal import Principal
+
+class TestPrincipal:
+
+    def test_default(self):
+        p = Principal()
+        assert p.to_str() == 'aaaaa-aa'
+
+    def test_anonymous(self):
+        p = Principal.anonymous();
+        assert p.to_str() == '2vxsx-fae'
+
+    def test_pubkey(self):
+        p = Principal.self_authenticating("ec172b93ad5e563bf4932c70e1245034c35467ef2efd4d64ebf819683467e2bf") # create a principal from public key
+        assert p.to_str() == 'zbtml-m23rk-szoaa-5x6p5-co5in-yylx6-463xu-zbjp4-4oizn-cqaij-tae'
+
+    def test_fromstr(self):
+        p = Principal.from_str("zbtml-m23rk-szoaa-5x6p5-co5in-yylx6-463xu-zbjp4-4oizn-cqaij-tae")
+        assert p.to_str() == 'zbtml-m23rk-szoaa-5x6p5-co5in-yylx6-463xu-zbjp4-4oizn-cqaij-tae'

tests/test_principal.py

@@ -0,0 +1,8 @@
+[tox]
+envlist = py
+isolated_build = true
+
+[testenv]
+changedir = tests
+deps = pytest
+commands = pytest {posargs}