from golem_messages.message.concents import ClientAuthorization
from golem_messages.shortcuts import dump
from golem_messages.shortcuts import load
from golem_messages.utils import encode_hex
import datetime
import json
import requests
import http.client
from protocol_constants import get_protocol_constants
from protocol_constants import print_protocol_constants
from utils.helpers import sign_message
from utils.testing_helpers import generate_ecc_key_pair
(PROVIDER_PRIVATE_KEY, PROVIDER_PUBLIC_KEY) = generate_ecc_key_pair()
(REQUESTOR_PRIVATE_KEY, REQUESTOR_PUBLIC_KEY) = generate_ecc_key_pair()
REQUESTOR_ETHEREUM_PUBLIC_KEY = '46de69a1c010c84dceafc0700d8e8fb160208df73da6b6ba7a4cbb59d63ad5068d50e723b9bbf854c733eb487f32ca76b1d877dedb89f0dd44a5bf0330f3963b'
PROVIDER_ETHEREUM_PUBLIC_KEY = '05a777c69b893cf8527aefc441774e7da00e7b70c8a74146fcd236c129db67708b5d39fdaa5dd548403f4614db558b938df1fc2f7b73338cc7802d2c9d1934758d'
class TestAssertionException(Exception):
pass
class count_fails(object):
"""
Decorator that wraps a test functions for intercepting assertions and counting them.
"""
instances = [] # type: ignore
return mocked_object
def mock_send_incorrect_request_to_cluster_incorrect_response(_headers, _request_http):
mocked_object = mock.Mock()
mocked_object.status_code = 404
return mocked_object
def mock_send_incorrect_request_to_cluster_unexpected_response(_headers, _request_http):
mocked_object = mock.Mock()
mocked_object.status_code = 500
return mocked_object
(CONCENT_PRIVATE_KEY, CONCENT_PUBLIC_KEY) = generate_ecc_key_pair()
@override_settings(
CONCENT_PRIVATE_KEY = CONCENT_PRIVATE_KEY,
CONCENT_PUBLIC_KEY = CONCENT_PUBLIC_KEY,
)
class RequestUploadStatusTest(ConcentIntegrationTestCase):
def test_properly_work_of_request_upload_status_function(self):
report_computed_task=self._get_deserialized_report_computed_task(
subtask_id='1',
task_to_compute=self._get_deserialized_task_to_compute(
task_id='1/1',
subtask_id='1',
)
def setUp(self):
self.message = "Missing 'Authorization' header."
self.path = 'blender/benchmark/test_task/scene-Helicopter-27-cycles.blend'
self.client_key = b64encode(generate_ecc_key_pair()[1]).decode("ascii")
from golem_messages import settings
from golem_messages.factories import tasks
from golem_messages.shortcuts import dump
from golem_messages.shortcuts import load
from core.models import Client
from core.models import StoredMessage
from core.models import PendingResponse
from core.tests.utils import ConcentIntegrationTestCase
from core.models import Subtask
from utils.constants import ErrorCode
from utils.helpers import get_current_utc_timestamp
from utils.testing_helpers import generate_ecc_key_pair
(CONCENT_PRIVATE_KEY, CONCENT_PUBLIC_KEY) = generate_ecc_key_pair()
(PROVIDER_PRIVATE_KEY, PROVIDER_PUBLIC_KEY) = generate_ecc_key_pair()
(REQUESTOR_PRIVATE_KEY, REQUESTOR_PUBLIC_KEY) = generate_ecc_key_pair()
(DIFFERENT_PRIVATE_KEY, DIFFERENT_PUBLIC_KEY) = generate_ecc_key_pair()
@override_settings(
CONCENT_PRIVATE_KEY = CONCENT_PRIVATE_KEY,
CONCENT_PUBLIC_KEY = CONCENT_PUBLIC_KEY,
CONCENT_MESSAGING_TIME = 3600,
)
class CoreViewSendTest(ConcentIntegrationTestCase):
@freeze_time("2017-11-17 10:00:00")
def setUp(self):
super().setUp()
from django.test import TestCase
from golem_messages import message
from golem_messages.factories import tasks
from golem_messages.shortcuts import dump
from golem_messages.shortcuts import load
from core.exceptions import Http400
from core.validation import validate_all_messages_identical
from core.validation import validate_golem_message_client_authorization
from core.validation import validate_golem_message_signed_with_key
from utils.shortcuts import load_without_public_key
from utils.testing_helpers import generate_ecc_key_pair
(CONCENT_PRIVATE_KEY, CONCENT_PUBLIC_KEY) = generate_ecc_key_pair()
(REQUESTOR_PRIVATE_KEY, REQUESTOR_PUBLIC_KEY) = generate_ecc_key_pair()
@override_settings(
CONCENT_PRIVATE_KEY = CONCENT_PRIVATE_KEY,
CONCENT_PUBLIC_KEY = CONCENT_PUBLIC_KEY,
)
class LoadWithoutPublicKeyUnitTest(TestCase):
def test_load_without_public_key_should_load_message(self):
"""
Test that message loaded with load_without_public_key function will be the same as load
with golem_messages load function.
"""
def _get_or_generate_keys(party):
if are_keys_predifined(party):
return _get_predefined_keys(party)
return generate_ecc_key_pair()
def setUp(self):
super().setUp()
# Keys
(self.PROVIDER_PRIVATE_KEY, self.PROVIDER_PUBLIC_KEY) = generate_ecc_key_pair()
(self.REQUESTOR_PRIVATE_KEY, self.REQUESTOR_PUBLIC_KEY) = generate_ecc_key_pair()
(self.DIFFERENT_PROVIDER_PRIVATE_KEY, self.DIFFERENT_PROVIDER_PUBLIC_KEY) = generate_ecc_key_pair()
(self.DIFFERENT_REQUESTOR_PRIVATE_KEY, self.DIFFERENT_REQUESTOR_PUBLIC_KEY) = generate_ecc_key_pair()
(self.PROVIDER_PRIV_ETH_KEY, self.PROVIDER_PUB_ETH_KEY) = generate_priv_and_pub_eth_account_key()
(self.REQUESTOR_PRIV_ETH_KEY, self.REQUESTOR_PUB_ETH_KEY) = generate_priv_and_pub_eth_account_key()
(self.DIFFERENT_PROVIDER_PRIV_ETH_KEY, self.DIFFERENT_PROVIDER_PUB_ETH_KEY) = generate_priv_and_pub_eth_account_key()
(self.DIFFERENT_REQUESTOR_PRIV_ETH_KEY, self.DIFFERENT_REQUESTOR_PUB_ETH_KEY) = generate_priv_and_pub_eth_account_key()
# StoredMessage
self.stored_message_counter = 0
# Auth
self.auth_message_counter = 0
