def setUp(self):
super().setUp()
self.concent_service = client.ConcentClientService(
keys_auth=keysauth.KeysAuth(
datadir=self.path,
private_key_name='priv_key',
password='******',
),
variant=variables.CONCENT_CHOICES['dev'],
)
self.msg = message.concents.ForceReportComputedTask()
def setUp(self):
super().setUp()
self.keys_auth = keysauth.KeysAuth(
datadir=self.path,
private_key_name='priv_key',
password='******',
)
self.cfs = filetransfers.ConcentFiletransferService(
keys_auth=self.keys_auth,
variant=variables.CONCENT_CHOICES['dev'],
)
def setUp(self):
super().setUp()
gc.collect()
# unfortunately dispatcher.disconnect won't do the job
dispatcher.connections = {}
dispatcher.senders = {}
dispatcher.sendersBack = {}
self.concent_service = client.ConcentClientService(
keys_auth=keysauth.KeysAuth(
datadir=self.path,
private_key_name='priv_key',
password='******',
),
variant=variables.CONCENT_CHOICES['dev'],
)
from golem.ethereum.incomeskeeper import IncomesKeeper
self.incomes_keeper = IncomesKeeper()
self.history = history.MessageHistoryService()
def setUp(self):
super().setUp()
self.client.concent_filetransfers = ConcentFiletransferService(
keys_auth=keysauth.KeysAuth(
datadir=self.path,
private_key_name='priv_key',
password='******',
),
variant=variables.CONCENT_CHOICES['dev'],
)
self.cft = self.client.concent_filetransfers
cft_patch = mock.patch(
'golem.network.concent.filetransfers'
'.ConcentFiletransferService.running',
mock.Mock(return_value=True))
cft_patch.start()
self.addCleanup(cft_patch.stop)
def setUp(self):
super().setUp()
self.task_session = tasksession.TaskSession(mock.MagicMock())
self.task_session.task_server.keys_auth = keys_auth = \
keysauth.KeysAuth(
datadir=self.tempdir,
private_key_name='priv_key',
password='******',
)
self.task_session.key_id = "KEY_ID"
self.msg = factories.tasks.ReportComputedTaskFactory()
self.msg._fake_sign()
self.now = datetime.datetime.utcnow()
now_ts = calendar.timegm(self.now.utctimetuple())
self.msg.task_to_compute.compute_task_def['deadline'] = now_ts + 60
self.msg.task_to_compute.sig = keys_auth.ecc.sign(
inputb=self.msg.task_to_compute.get_short_hash(),
)
task_id = self.msg.task_to_compute.compute_task_def['task_id']
task_header = taskbase.TaskHeader(
task_id='task_id',
environment='env',
task_owner=Node()
)
task_header.deadline = now_ts + 3600
task = mock.Mock()
task.header = task_header
self.task_session.task_manager.tasks = {
task_id: task,
}
self.task_session.task_manager.tasks_states = {}
self.task_session.task_manager.tasks_states[task_id] = task_state = \
taskstate.TaskState()
ctk = self.task_session.task_manager.comp_task_keeper
ctk.get_node_for_task_id.return_value = "KEY_ID"
self.task_session.task_manager.get_node_id_for_subtask.return_value = \
"KEY_ID"
task_state.subtask_states[self.msg.subtask_id] = subtask_state = \
taskstate.SubtaskState()
subtask_state.deadline = self.msg.task_to_compute.compute_task_def[
'deadline'
]