mirror of
https://github.com/yzernik/squeaknode.git
synced 2026-08-14 12:43:24 +02:00
5198467889
* Got all tests in core dir passing * Got all tests in db dir passing * Got tests for admin dir passing * Got all unit tests passing * Added migration for squeak table and profile table using new pubkey * Re-start db migrations with new data-v2.db file * Fix column type for profile public key * Got all itests passing * Log the public key in hex in create contact profile handle fn * Rename rpc methods to use pubkey instead of address * Got frontend mostly working with new profile pubkey * Update frontend to use pubkey instead of address * Change size of squeak profile detail item * Update create contact profile dialog to use pubkey * Update twitter account item to use pubkey * Include label for pubkey in profile detail item * Decrease size of profile detail item * Update frontend build * Improve logging of requests in handler
136 lines
4.4 KiB
Python
136 lines
4.4 KiB
Python
# MIT License
|
|
#
|
|
# Copyright (c) 2020 Jonathan Zernik
|
|
#
|
|
# Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
# of this software and associated documentation files (the "Software"), to deal
|
|
# in the Software without restriction, including without limitation the rights
|
|
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
# copies of the Software, and to permit persons to whom the Software is
|
|
# furnished to do so, subject to the following conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be included in all
|
|
# copies or substantial portions of the Software.
|
|
#
|
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
# SOFTWARE.
|
|
import mock
|
|
import pytest
|
|
from squeak.messages import msg_getdata
|
|
from squeak.messages import msg_getsqueaks
|
|
from squeak.net import CInterested
|
|
from squeak.net import CInv
|
|
from squeak.net import CSqueakLocator
|
|
|
|
from squeaknode.core.download_result import DownloadResult
|
|
from squeaknode.node.active_download_manager import HashDownload
|
|
from squeaknode.node.active_download_manager import InterestDownload
|
|
from squeaknode.node.downloaded_object import DownloadedSqueak
|
|
from tests.utils import gen_squeak
|
|
|
|
|
|
@pytest.fixture()
|
|
def download_hash(squeak_hash):
|
|
yield HashDownload(squeak_hash=squeak_hash)
|
|
|
|
|
|
@pytest.fixture()
|
|
def interest(public_key, block_count):
|
|
yield CInterested(
|
|
pubkeys=(public_key,),
|
|
nMinBlockHeight=block_count - 100,
|
|
nMaxBlockHeight=block_count + 100,
|
|
)
|
|
|
|
|
|
@pytest.fixture()
|
|
def download_interest(interest):
|
|
yield InterestDownload(limit=10, interest=interest)
|
|
|
|
|
|
def test_download_hash_is_interested(download_hash, squeak):
|
|
downloaded_squeak = DownloadedSqueak(squeak)
|
|
|
|
assert download_hash.is_interested(downloaded_squeak)
|
|
|
|
|
|
def test_download_hash_is_not_interested(download_hash, private_key, block_count):
|
|
other_squeak = gen_squeak(private_key, block_count)
|
|
downloaded_squeak = DownloadedSqueak(other_squeak)
|
|
|
|
assert not download_hash.is_interested(downloaded_squeak)
|
|
|
|
|
|
def test_download_interest_is_interested(download_interest, squeak):
|
|
downloaded_squeak = DownloadedSqueak(squeak)
|
|
|
|
assert download_interest.is_interested(downloaded_squeak)
|
|
|
|
|
|
def test_download_interest_is_not_interested(download_interest, private_key, block_count):
|
|
other_squeak = gen_squeak(private_key, block_count + 200)
|
|
downloaded_squeak = DownloadedSqueak(other_squeak)
|
|
|
|
assert not download_interest.is_interested(downloaded_squeak)
|
|
|
|
|
|
def test_download_hash_initiate(download_hash, squeak_hash):
|
|
broadcast_fn = mock.Mock()
|
|
download_hash.initiate_download(broadcast_fn)
|
|
|
|
expected_msg = msg_getdata(
|
|
inv=[CInv(type=1, hash=squeak_hash)]
|
|
)
|
|
|
|
broadcast_fn.assert_called_once_with(expected_msg)
|
|
|
|
|
|
def test_download_interest_initiate(download_interest, interest):
|
|
broadcast_fn = mock.Mock()
|
|
download_interest.initiate_download(broadcast_fn)
|
|
|
|
expected_msg = msg_getsqueaks(
|
|
locator=CSqueakLocator(
|
|
vInterested=[interest],
|
|
)
|
|
)
|
|
|
|
broadcast_fn.assert_called_once_with(expected_msg)
|
|
|
|
|
|
def test_download_hash_mark_complete(download_hash, squeak):
|
|
with mock.patch.object(download_hash, 'mark_complete', autospec=True) as mock_mark_complete:
|
|
mock_mark_complete.return_value = None
|
|
download_hash.increment()
|
|
|
|
mock_mark_complete.assert_called_once_with()
|
|
|
|
|
|
def test_download_hash_mark_complete_not_called(download_hash, squeak):
|
|
with mock.patch.object(download_hash, 'mark_complete', autospec=True) as mock_mark_complete:
|
|
mock_mark_complete.return_value = None
|
|
|
|
mock_mark_complete.assert_not_called()
|
|
|
|
|
|
def test_download_hash_wait_for_complete(download_hash):
|
|
download_hash.mark_complete()
|
|
|
|
download_hash.wait_for_complete(50)
|
|
|
|
|
|
def test_download_hash_get_result(download_hash, squeak):
|
|
download_hash.increment()
|
|
download_result = download_hash.get_result()
|
|
|
|
assert download_result == DownloadResult(
|
|
number_downloaded=1,
|
|
number_requested=1,
|
|
elapsed_time_ms=0,
|
|
number_peers=0,
|
|
)
|