• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• django-eth-events-master
  • LICENSE
  • django_eth_events
    • apps.py
    • reorgs.py
    • chainevents.py
    • exceptions.py
    • decoder.py
    • event_listener.py
    • factories.py
    • version.py
    • models.py
    • tasks.py
    • web3_service.py
    • __init__.py
    • utils.py
    • migrations
      • 0009_daemon_last_error_date_time.py
      • 0008_daemon_listener_lock.py
      • 0002_daemon_last_error_block_number.py
      • 0006_block_timestamp.py
      • 0005_daemon_status.py
      • 0007_auto_20171207_0759.py
      • 0001_initial.py
      • __init__.py
      • 0004_auto_20171127_0501.py
      • 0003_block.py
      • 0010_remove_daemon_status.py
    • tests
      • test_utils.py
      • test_decoder.py
      • test_reorg_detector.py
      • test_event_listener_exec.py
      • test_singleton.py
      • test_daemon_model.py
      • test_event_listener_functions.py
      • test_celery.py
      • __init__.py
      • utils.py
      • test_chainevents.py
      • mocked_testrpc_reorg.py
      • test_web3_service.py
    • singleton.py
    • admin.py
    • management
      • commands
        • resync_daemon.py
        • __init__.py
      • __init__.py
  • manage.py
  • config
    • settings
      • local.py
      • __init__.py
      • test.py
      • base.py
  • run_tests.py
  • setup.py
  • setup.cfg
  • .travis.yml
  • README.md
  • scripts
    • deploy.sh
  • requirements.txt
  • .gitignore

import errno
from django.test import TestCase
from hexbytes import HexBytes
from urllib3.exceptions import HTTPError, LocationValueError, PoolError
from json import dumps, loads

from ..utils import normalize_address_without_0x, remove_0x_head, is_network_error, JsonBytesEncoder
from ..exceptions import Web3ConnectionException


class TestUtils(TestCase):

    def test_remove_0x_head(self):
        self.assertEqual('b58d5491D17ebF46E9DB7F18CeA7C556AE80d53B',
                         remove_0x_head('0xb58d5491D17ebF46E9DB7F18CeA7C556AE80d53B'))

        self.assertEqual('b58d5491d17ebf46e9db7f18cea7c556ae80d53b',
                         remove_0x_head('0xb58d5491d17ebf46e9db7f18cea7c556ae80d53b'))

        self.assertEqual('b58d5491d17ebf46e9db7f18cea7c556ae80d53B',
                         remove_0x_head('0xb58d5491d17ebf46e9db7f18cea7c556ae80d53B'))

        self.assertEqual('b58d5491d17ebf46e9db7f18cea7c556ae80d53b',
                         remove_0x_head(HexBytes('0xb58d5491D17ebF46E9DB7F18CeA7C556AE80d53B')))

        self.assertEqual('b58d5491d17ebf46e9db7f18cea7c556ae80d53b',
                         remove_0x_head(HexBytes('0xb58d5491d17ebf46e9db7f18cea7c556ae80d53B')))

        self.assertEqual('b58d5491d17ebf46e9db7f18cea7c556ae80d53b',
                         remove_0x_head(HexBytes('0xb58d5491d17ebf46e9db7f18cea7c556ae80d53B')))

        self.assertEqual('b58d5491d17ebf46e9db7f18cea7c556ae80d53b',
                         remove_0x_head(HexBytes('b58d5491d17ebf46e9db7f18cea7c556ae80d53B')))

    def test_normalize_address_without_0x(self):
        self.assertEqual('b58d5491d17ebf46e9db7f18cea7c556ae80d53b',
                         normalize_address_without_0x('0xb58d5491D17ebF46E9DB7F18CeA7C556AE80d53B'))

        self.assertEqual('b58d5491d17ebf46e9db7f18cea7c556ae80d53b',
                         normalize_address_without_0x('0xb58d5491d17ebf46e9db7f18cea7c556ae80d53b'))

        self.assertEqual('b58d5491d17ebf46e9db7f18cea7c556ae80d53b',
                         normalize_address_without_0x(HexBytes('0xb58d5491d17ebf46e9db7f18cea7c556ae80d53b')))

        self.assertEqual('b58d5491d17ebf46e9db7f18cea7c556ae80d53b',
                         normalize_address_without_0x(HexBytes('b58d5491d17ebf46e9db7f18cea7c556ae80d53b')))

    def test_network_error_exception_detector(self):
        http_error = HTTPError()
        self.assertTrue(is_network_error(http_error))

        location_value_error = LocationValueError()
        self.assertTrue(is_network_error(location_value_error))

        pool_error = PoolError(None, 'an error')
        self.assertTrue(is_network_error(pool_error))

        exception = Exception()
        self.assertFalse(is_network_error(exception))

        w3_conn_error = Web3ConnectionException()
        self.assertFalse(is_network_error(w3_conn_error))

        setattr(w3_conn_error, 'errno', errno.ECONNABORTED)
        self.assertTrue(is_network_error(w3_conn_error))

        setattr(w3_conn_error, 'errno', errno.EPERM)
        self.assertFalse(is_network_error(w3_conn_error))

    def test_json_encoder(self):
        base_address = 'b58d5491d17ebf46e9db7f18cea7c556ae80d53B'
        ipfs_hash_string = 'Qme4GBhwNJharbu83iNEsd5WnUhQYM1rBAgCgsSuFMdjcS'
        ipfs_hash_bytes = ipfs_hash_string.encode()  # b'...'

        json = {'ipfs_hash': ipfs_hash_bytes}
        # Simulate string encoding and convert back to dict
        encoded_json = loads(dumps(json, cls=JsonBytesEncoder))
        self.assertEqual(ipfs_hash_bytes.decode(), encoded_json['ipfs_hash'])

        json = {'ipfs_hash': ipfs_hash_string}
        # Simulate string encoding and convert back to dict
        encoded_json = loads(dumps(json, cls=JsonBytesEncoder))
        self.assertEqual(ipfs_hash_string, encoded_json['ipfs_hash'])

        hex_bytes_address = HexBytes(base_address)
        json = {'address': hex_bytes_address}
        encoded_json = loads(dumps(json, cls=JsonBytesEncoder))
        self.assertEqual(hex_bytes_address.hex(), encoded_json['address'])

        bytes_address = bytes.fromhex(base_address)
        json = {'address': bytes_address}
        encoded_json = loads(dumps(json, cls=JsonBytesEncoder))
        self.assertEqual('0x' + bytes_address.hex(), encoded_json['address'])