Skip to content
This repository has been archived by the owner on Jan 13, 2023. It is now read-only.

add tests for iter_used_addresses #299

Merged
merged 2 commits into from
Jan 30, 2020
Merged

add tests for iter_used_addresses #299

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
dd190e1
add tests for `iter_used_addresses`
lzpap Jan 29, 2020
feef769
fix typos in `util_test.py`
lzpap Jan 30, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
241 changes: 241 additions & 0 deletions test/commands/extended/utils_test.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,241 @@
# coding=utf-8
from __future__ import absolute_import, division, print_function, \
unicode_literals

from unittest import TestCase
from iota.commands.extended.utils import iter_used_addresses
from iota import MockAdapter
from iota.crypto.types import Seed
from test import mock


class IterUsedAddressesTestCase(TestCase):
def setUp(self):
super(IterUsedAddressesTestCase, self).setUp()

self.adapter = MockAdapter()
self.seed = Seed(trytes='S' * 81)
self.address0 = 'A' * 81
self.address1 = 'B' * 81
self.address2 = 'C' * 81
self.address3 = 'D' * 81

# To speed up the tests, we will mock the address generator.
def address_generator(ag, start, step=1):
for addy in [self.address0, self.address1, self.address2,
self.address3][start::step]:
yield addy
self.mock_address_generator = address_generator

def seed_unused_address(self):
self.adapter.seed_response('findTransactions', {
'hashes': [],
})
self.adapter.seed_response('wereAddressesSpentFrom', {
'states': [False],
})

def get_all_used_addresses(self, start=0):
return [address for address, _
in iter_used_addresses(self.adapter, self.seed, start)]

def test_first_address_is_not_used(self):
"""
The very first address is not used. No address is returned.
"""
# Address 0
self.seed_unused_address()

with mock.patch(
'iota.crypto.addresses.AddressGenerator.create_iterator',
self.mock_address_generator,
):
self.assertEqual([], self.get_all_used_addresses())

self.assertListEqual(
self.adapter.requests,
[
{
'command': 'findTransactions',
'addresses': [self.address0],
},
{
'command': 'wereAddressesSpentFrom',
'addresses': [self.address0],
},
]
)

def test_transactions_are_considered_used(self):
"""
An address with a transaction is considered used.
"""
# Address 0
self.adapter.seed_response('findTransactions', {
'hashes': ['T' * 81],
})

# Address 1
self.seed_unused_address()

with mock.patch(
'iota.crypto.addresses.AddressGenerator.create_iterator',
self.mock_address_generator,
):
self.assertEqual([self.address0], self.get_all_used_addresses())

self.assertListEqual(
self.adapter.requests,
[
{
'command': 'findTransactions',
'addresses': [self.address0],
},
{
'command': 'findTransactions',
'addresses': [self.address1],
},
{
'command': 'wereAddressesSpentFrom',
'addresses': [self.address1],
},
]
)

def test_spent_from_is_considered_used(self):
"""
An address that was spent from is considered used.
"""
# Address 0
self.adapter.seed_response('findTransactions', {
'hashes': [],
})
self.adapter.seed_response('wereAddressesSpentFrom', {
'states': [True],
})

# Address 1
self.seed_unused_address()

with mock.patch(
'iota.crypto.addresses.AddressGenerator.create_iterator',
self.mock_address_generator,
):
self.assertEqual([self.address0], self.get_all_used_addresses())

self.assertListEqual(
self.adapter.requests,
[
{
'command': 'findTransactions',
'addresses': [self.address0],
},
{
'command': 'wereAddressesSpentFrom',
'addresses': [self.address0],
},
{
'command': 'findTransactions',
'addresses': [self.address1],
},
{
'command': 'wereAddressesSpentFrom',
'addresses': [self.address1],
},
]
)

def test_start_parameter_is_given(self):
"""
The correct address is returned if a start parameter is given.
"""
# Address 1
self.adapter.seed_response('findTransactions', {
'hashes': ['T' * 81],
})

# Address 2
self.seed_unused_address()

with mock.patch(
'iota.crypto.addresses.AddressGenerator.create_iterator',
self.mock_address_generator,
):
self.assertEqual([self.address1],
self.get_all_used_addresses(start=1))

self.assertListEqual(
self.adapter.requests,
[
{
'command': 'findTransactions',
'addresses': [self.address1],
},
{
'command': 'findTransactions',
'addresses': [self.address2],
},
{
'command': 'wereAddressesSpentFrom',
'addresses': [self.address2],
},
]
)

def test_multiple_addresses_return(self):
"""
A larger test that combines multiple cases and more than one address
should be returned.
Address 0: Was spent from
Address 1: Has a transaction
Address 2: Is not used. Should not be returned
"""

# Address 0
self.adapter.seed_response('findTransactions', {
'hashes': [],
})
self.adapter.seed_response('wereAddressesSpentFrom', {
'states': [True],
})

# Address 1
self.adapter.seed_response('findTransactions', {
'hashes': ['T' * 81],
})

# Address 2
self.seed_unused_address()

with mock.patch(
'iota.crypto.addresses.AddressGenerator.create_iterator',
self.mock_address_generator,
):
self.assertEqual([self.address0, self.address1],
self.get_all_used_addresses())

self.assertListEqual(
self.adapter.requests,
[
{
'command': 'findTransactions',
'addresses': [self.address0],
},
{
'command': 'wereAddressesSpentFrom',
'addresses': [self.address0],
},
{
'command': 'findTransactions',
'addresses': [self.address1],
},
{
'command': 'findTransactions',
'addresses': [self.address2],
},
{
'command': 'wereAddressesSpentFrom',
'addresses': [self.address2],
},
]
)