HCE Project Python language Distributed Tasks Manager Application, Distributed Crawler Application and client API bindings.  2.0.0-chaika
Hierarchical Cluster Engine Python language binding
Public Member Functions | Public Attributes | List of all members
tests.test_dtm_TasksManager.TestTasksManager Class Reference
Inheritance diagram for tests.test_dtm_TasksManager.TestTasksManager:
Inheritance graph
[legend]
Collaboration diagram for tests.test_dtm_TasksManager.TestTasksManager:
Collaboration graph
[legend]

Public Member Functions

def setUp (self)
 
def tearDown (self)
 
def check_task_in_dbi (self, task, full_compare=True)
 
def test_add_new_task (self)
 
def test_new_task_tasksDataManager_response (self)
 
def test_fetch_available_tasks (self)
 
def test_new_task_tasksDataManager_response_err (self)
 
def test_add_new_task_dbi_exception (self)
 
def test_add_new_task_scheduler_exception (self)
 
def test_add_new_task_scheduler_ok (self)
 
def test_update_task (self)
 
def test_update_task_not_procssing_task (self)
 
def test_delete_tasks (self)
 
def test_delete_task_not_processing_task (self)
 
def test_delete_task_dbi_error (self)
 
def test_fetch_result_cache (self)
 
def test_fetch_result_cache_no_task (self)
 
def test_get_status_request (self)
 
def test_get_task_fields (self)
 
def test_update_task_field1 (self)
 
def test_fetch_result_response (self)
 
def test_fetch_result_response_wrong_event_uid (self)
 
def test_update_task_field (self)
 

Public Attributes

 config
 
 adminServerFake
 
 pollerManager_mock
 
 connectionBuilderLight_mock
 
 dbi_mock
 
 serverConnection_mock
 
 tasksDataManager_mock
 
 scheduler_mock
 
 connections_mock
 
 tasksManager
 
 eventBuilder
 
 taskId
 
 newTask
 
 generalResponse
 
 updateTask
 
 deleteTasks
 
 fetchTasksResultsFromCache
 

Detailed Description

Definition at line 61 of file test_dtm_TasksManager.py.

Member Function Documentation

◆ check_task_in_dbi()

def tests.test_dtm_TasksManager.TestTasksManager.check_task_in_dbi (   self,
  task,
  full_compare = True 
)

Definition at line 108 of file test_dtm_TasksManager.py.

108  def check_task_in_dbi(self, task, full_compare=True):
109  lookBackTaskLogScheme = TaskBackLogScheme(task)
110  resBackTaskLogScheme = self.tasksManager.dbi.fetch(lookBackTaskLogScheme, "id=%s" % task.id)[0]
111  resTaskLog = resBackTaskLogScheme._getTaskLog()
112 
113  if not full_compare:
114  if task.id == resTaskLog.id:
115  return True
116  else:
117  return False
118 
119  if task.__dict__ == resTaskLog.__dict__:
120  return True
121  return False
122 
123 
Here is the caller graph for this function:

◆ setUp()

def tests.test_dtm_TasksManager.TestTasksManager.setUp (   self)

Definition at line 64 of file test_dtm_TasksManager.py.

64  def setUp(self):
65  self.config = ConfigParser.RawConfigParser()
66  cfg_section = "TasksManager"
67 
68  tf = tempfile.NamedTemporaryFile()
69  sql_name = "sqlite:///" + os.path.basename(tf.name) + ".db"
70 
71  self.config.add_section(cfg_section)
72  self.config.set(cfg_section, TasksManager.SERVER, "TasksManager")
73  self.config.set(cfg_section, TasksManager.TASKS_DATA_MANAGER_CLIENT, "TasksDataManager")
74  self.config.set(cfg_section, TasksManager.SCHEDULER_CLIENT, "Scheduler")
75  self.config.set(cfg_section, "db_name", sql_name)
76 
77  connectionBuilderLight = ConnectionBuilderLight()
78  self.adminServerFake = connectionBuilderLight.build(tr_consts.SERVER_CONNECT, BaseServerManager.ADMIN_CONNECT_ENDPOINT)
79 
80  self.pollerManager_mock = MagicMock(spec=PollerManager)
81  self.connectionBuilderLight_mock = MagicMock(spec=ConnectionBuilderLight )
82 
83  self.dbi_mock = MagicMock(DBI)
84  self.serverConnection_mock = MagicMock(spec=ConnectionLight)
85  self.tasksDataManager_mock = MagicMock(spec=ConnectionLight)
86  self.scheduler_mock = MagicMock(spec=ConnectionLight)
87  self.connections_mock = [self.serverConnection_mock, self.tasksDataManager_mock, self.scheduler_mock]
88 
89  self.connectionBuilderLight_mock.build.side_effect = self.connections_mock
90 
91  self.tasksManager = TasksManager(self.config, self.connectionBuilderLight_mock, self.pollerManager_mock)
92  self.eventBuilder = EventBuilder()
93 
94  self.taskId = 1
95  self.newTask = NewTask("ls", self.taskId)
96  self.generalResponse = GeneralResponse()
97  self.updateTask = UpdateTask(self.taskId)
98  self.deleteTasks = DeleteTask(self.taskId)
99  self.fetchTasksResultsFromCache = FetchTasksResultsFromCache([self.taskId])
100 
101 

◆ tearDown()

def tests.test_dtm_TasksManager.TestTasksManager.tearDown (   self)

Definition at line 102 of file test_dtm_TasksManager.py.

102  def tearDown(self):
103  lookBackTaskLogScheme = TaskBackLogScheme(self.newTask)
104  self.tasksManager.dbi.delete(lookBackTaskLogScheme, "id=%s" % self.newTask.id)
105  self.adminServerFake.close()
106 
107 

◆ test_add_new_task()

def tests.test_dtm_TasksManager.TestTasksManager.test_add_new_task (   self)

Definition at line 124 of file test_dtm_TasksManager.py.

124  def test_add_new_task(self):
125  newTaskEvent = self.eventBuilder.build(EVENT_TYPES.NEW_TASK, self.newTask)
126  self.pollerManager_mock.poll.return_value = [self.tasksManager.SERVER]
127  self.serverConnection_mock.recv.return_value = newTaskEvent
128 
129  match_event = Matcher(matchShedulerNewEvents, newTaskEvent)
130 
131  self.tasksManager.poll()
132 
133  self.assertEqual(len(self.tasksManager.tasksQueue), 0, "task is in tasksQueue")
134  self.assertEqual(len(self.tasksManager.pendingTasks), 1, "")
135  self.tasksDataManager_mock.send.assert_called_once_with(match_event)
136 
137 
138 

◆ test_add_new_task_dbi_exception()

def tests.test_dtm_TasksManager.TestTasksManager.test_add_new_task_dbi_exception (   self)

Definition at line 199 of file test_dtm_TasksManager.py.

199  def test_add_new_task_dbi_exception(self):
200 
201  newTaskEvent = self.eventBuilder.build(EVENT_TYPES.NEW_TASK, self.newTask)
202  newTaskEvent.connect_name = "server"
203  self.tasksManager.onNewTask(newTaskEvent)
204 
205  generalResponseEvent = self.eventBuilder.build(EVENT_TYPES.NEW_TASK_RESPONSE, self.generalResponse)
206  generalResponseEvent.uid = newTaskEvent.uid
207  self.pollerManager_mock.poll.return_value = [self.tasksManager.SERVER]
208  self.serverConnection_mock.recv.return_value = generalResponseEvent
209 
210  self.tasksManager.addConnection("server", self.serverConnection_mock)
211 
212  self.tasksManager.tasksQueue[self.newTask.id] = self.newTask
213 
214  self.tasksManager.dbi = self.dbi_mock
215  self.dbi_mock.insert.side_effect = DBIErr(11, "some err")
216 
217  #self.tasksManager.poll()
218  deleteTaskData = DeleteTaskData(self.taskId)
219  deleteEvent = self.eventBuilder.build(EVENT_TYPES.DELETE_TASK_DATA, deleteTaskData)
220 
221  match_event = Matcher(matchDeleteTasksEvent, deleteEvent)
222 
223  self.tasksManager.poll()
224 
225  self.assertEqual(len(self.tasksManager.tasksQueue), 0, "task is in tasksQueue")
226  self.assertEqual(len(self.tasksManager.pendingTasks), 0, "")
227  self.tasksDataManager_mock.send.assert_called_with(match_event)
228  self.serverConnection_mock.send.assert_called_once_with(ANY) #send response
229 
230 

◆ test_add_new_task_scheduler_exception()

def tests.test_dtm_TasksManager.TestTasksManager.test_add_new_task_scheduler_exception (   self)

Definition at line 231 of file test_dtm_TasksManager.py.

231  def test_add_new_task_scheduler_exception(self):
232 
233  newTaskEvent = self.eventBuilder.build(EVENT_TYPES.NEW_TASK, self.newTask)
234  newTaskEvent.connect_name = "server"
235  self.tasksManager.onNewTask(newTaskEvent)
236 
237  generalResponseEvent = self.eventBuilder.build(EVENT_TYPES.NEW_TASK_RESPONSE, self.generalResponse)
238  generalResponseEvent.uid = newTaskEvent.uid
239  self.tasksManager.onTasksManagerGeneralResponse(generalResponseEvent)
240 
241  self.pollerManager_mock.poll.return_value = [self.tasksManager.SERVER]
242  generalResponseErr = GeneralResponse(GeneralResponse.ERROR_OK + 1, "")
243  generalResponseErrEvent = self.eventBuilder.build(EVENT_TYPES.SCHEDULE_TASK_RESPONSE, generalResponseErr)
244  generalResponseErrEvent.uid = newTaskEvent.uid
245  self.serverConnection_mock.recv.return_value = generalResponseErrEvent
246 
247  self.tasksManager.addConnection("server", self.serverConnection_mock)
248 
249  self.tasksManager.poll()
250 
251  self.assertEqual(len(self.tasksManager.tasksQueue), 0, "task is in tasksQueue")
252  self.assertEqual(len(self.tasksManager.pendingTasks), 0, "")
253  self.serverConnection_mock.send.assert_called_once_with(ANY) #send response
254 
255 

◆ test_add_new_task_scheduler_ok()

def tests.test_dtm_TasksManager.TestTasksManager.test_add_new_task_scheduler_ok (   self)

Definition at line 256 of file test_dtm_TasksManager.py.

256  def test_add_new_task_scheduler_ok(self):
257 
258  newTaskEvent = self.eventBuilder.build(EVENT_TYPES.NEW_TASK, self.newTask)
259  newTaskEvent.connect_name = "server"
260  self.tasksManager.onNewTask(newTaskEvent)
261 
262  generalResponseEvent = self.eventBuilder.build(EVENT_TYPES.SCHEDULE_TASK_RESPONSE, self.generalResponse)
263  generalResponseEvent.uid = newTaskEvent.uid
264  self.tasksManager.onTasksManagerGeneralResponse(generalResponseEvent)
265 
266  self.pollerManager_mock.poll.return_value = [self.tasksManager.SERVER]
267  self.serverConnection_mock.recv.return_value = generalResponseEvent
268 
269  self.tasksManager.addConnection("server", self.serverConnection_mock)
270  taskLog = self.tasksManager.createTaskLog(self.newTask)
271 
272  self.tasksManager.poll()
273 
274  self.assertEqual(len(self.tasksManager.tasksQueue), 1, "task is in tasksQueue")
275  self.assertEqual(len(self.tasksManager.pendingTasks), 0, "")
276  self.serverConnection_mock.send.assert_called_once_with(ANY) #send response
277  self.assertTrue(self.check_task_in_dbi(taskLog, False), "insertion in dbi is broken")
278 
279 
Here is the call graph for this function:

◆ test_delete_task_dbi_error()

def tests.test_dtm_TasksManager.TestTasksManager.test_delete_task_dbi_error (   self)

Definition at line 342 of file test_dtm_TasksManager.py.

342  def test_delete_task_dbi_error(self):
343  #add task
344  self.tasksManager.tasksQueue[self.newTask.id] = self.newTask
345 
346  deleteTaskEvent = self.eventBuilder.build(EVENT_TYPES.DELETE_TASK, self.deleteTasks)
347  self.pollerManager_mock.poll.return_value = [self.tasksManager.SERVER]
348  self.serverConnection_mock.recv.return_value = deleteTaskEvent
349 
350  old_dbi = self.tasksManager.dbi
351  self.tasksManager.dbi = self.dbi_mock
352  self.dbi_mock.delete.side_effect = DBIErr(11, "some err")
353 
354  match_event = Matcher(matchDeleteTasksEvent, deleteTaskEvent)
355 
356  self.tasksManager.poll()
357 
358  self.serverConnection_mock.send.assert_called_once_with(ANY)
359 
360  self.tasksManager.dbi = old_dbi
361 
362 

◆ test_delete_task_not_processing_task()

def tests.test_dtm_TasksManager.TestTasksManager.test_delete_task_not_processing_task (   self)

Definition at line 332 of file test_dtm_TasksManager.py.

332  def test_delete_task_not_processing_task(self):
333  deleteTaskEvent = self.eventBuilder.build(EVENT_TYPES.DELETE_TASK, self.deleteTasks)
334  self.pollerManager_mock.poll.return_value = [self.tasksManager.SERVER]
335  self.serverConnection_mock.recv.return_value = deleteTaskEvent
336 
337  self.tasksManager.poll()
338 
339  self.serverConnection_mock.send.assert_called_once_with(ANY)
340 
341 

◆ test_delete_tasks()

def tests.test_dtm_TasksManager.TestTasksManager.test_delete_tasks (   self)

Definition at line 309 of file test_dtm_TasksManager.py.

309  def test_delete_tasks(self):
310  #add task
311  newTaskEvent = self.eventBuilder.build(EVENT_TYPES.NEW_TASK, self.newTask)
312  self.tasksManager.addNewTaskData(newTaskEvent)
313 
314  print self.check_task_in_dbi(self.newTask, False)
315  #self.tasksManager.tasksQueue[self.newTask.id] = self.newTask
316 
317  deleteTaskEvent = self.eventBuilder.build(EVENT_TYPES.DELETE_TASK, self.deleteTasks)
318  self.pollerManager_mock.poll.return_value = [self.tasksManager.SERVER]
319  self.serverConnection_mock.recv.return_value = deleteTaskEvent
320 
321  sendTaskEvent = self.eventBuilder.build(EVENT_TYPES.NEW_TASK, self.deleteTasks)
322  match_event = Matcher(matchDeleteTasksEvent, sendTaskEvent)
323 
324  self.tasksManager.poll()
325 
326  self.assertEqual(len(self.tasksManager.tasksQueue), 1, "task is in tasksQueue")
327  self.tasksDataManager_mock.send.assert_called_once_with(match_event)
328  self.assertEqual(self.scheduler_mock.send.call_count, 0, "")
329  self.assertEqual(self.serverConnection_mock.send.call_count, 0, "")
330 
331 
Here is the call graph for this function:

◆ test_fetch_available_tasks()

def tests.test_dtm_TasksManager.TestTasksManager.test_fetch_available_tasks (   self)

Definition at line 159 of file test_dtm_TasksManager.py.

159  def test_fetch_available_tasks(self):
160  req = FetchAvailabelTaskIds(100)
161  event = self.eventBuilder.build(EVENT_TYPES.FETCH_AVAILABLE_TASK_IDS, req)
162  event.connect_identity = "333"
163  taskLog1 = TaskLog()
164  taskLog1.id = "1"
165  taskLog2 = TaskLog()
166  taskLog2.id = "2"
167  taskLog3 = TaskLog()
168  taskLog3.id = "3"
169  self.tasksManager.dbi = self.dbi_mock
170  self.dbi_mock.sql.return_value = [TaskLogScheme(taskLog1), TaskLogScheme(taskLog2), TaskLogScheme(taskLog3)]
171  event.connect_name = TasksManager.SERVER
172  self.tasksManager.onFetchAvailableTasks(event)
173  args = self.serverConnection_mock.send.call_args
174  self.assertEqual(args[0][0].eventObj.ids, ["1", "2", "3"])
175 

◆ test_fetch_result_cache()

def tests.test_dtm_TasksManager.TestTasksManager.test_fetch_result_cache (   self)

Definition at line 363 of file test_dtm_TasksManager.py.

363  def test_fetch_result_cache(self):
364  #add task
365  self.tasksManager.tasksQueue[self.newTask.id] = self.newTask
366 
367  fetchTasksResultsFromCacheEvent = self.eventBuilder.build(EVENT_TYPES.FETCH_RESULTS_CACHE, self.fetchTasksResultsFromCache)
368  self.pollerManager_mock.poll.return_value = [self.tasksManager.SERVER]
369  self.serverConnection_mock.recv.return_value = fetchTasksResultsFromCacheEvent
370 
371  self.tasksManager.poll()
372 
373  self.assertEqual(len(self.tasksManager.fetchEvents), 1, "")
374  self.tasksDataManager_mock.send.assert_called_once_with(fetchTasksResultsFromCacheEvent)
375 
376 

◆ test_fetch_result_cache_no_task()

def tests.test_dtm_TasksManager.TestTasksManager.test_fetch_result_cache_no_task (   self)

Definition at line 377 of file test_dtm_TasksManager.py.

377  def test_fetch_result_cache_no_task(self):
378  fetchTasksResultsFromCacheEvent = self.eventBuilder.build(EVENT_TYPES.FETCH_RESULTS_CACHE, self.fetchTasksResultsFromCache)
379  self.pollerManager_mock.poll.return_value = [self.tasksManager.SERVER]
380  self.serverConnection_mock.recv.return_value = fetchTasksResultsFromCacheEvent
381 
382  self.tasksManager.poll()
383 
384  self.assertEqual(len(self.tasksManager.fetchEvents), 0, "")
385  self.assertEqual(self.tasksDataManager_mock.send.call_count, 0, "event send to tasksDataManager")
386  self.serverConnection_mock.send.assert_called_once_with(ANY)
387 
388 

◆ test_fetch_result_response()

def tests.test_dtm_TasksManager.TestTasksManager.test_fetch_result_response (   self)

Definition at line 418 of file test_dtm_TasksManager.py.

418  def test_fetch_result_response(self):
419  event = self.eventBuilder.build(EVENT_TYPES.FETCH_RESULTS_CACHE, "")
420  event.uid = "2"
421  event.connect_name = self.tasksManager.SERVER
422  self.tasksManager.fetchEvents[event.uid] = event
423 
424  eeResponseData = EEResponseData(self.taskId)
425  eeResponseDataEvent = self.eventBuilder.build(EVENT_TYPES.FETCH_TASK_RESULTS_RESPONSE, eeResponseData)
426  self.pollerManager_mock.poll.return_value = [self.tasksManager.SERVER]
427  self.serverConnection_mock.recv.return_value = eeResponseDataEvent
428 
429  self.tasksManager.poll()
430 
431  self.assertEqual(len(self.tasksManager.fetchEvents), 0, "event is still here")
432  self.serverConnection_mock.send.assert_called_once_with(ANY)
433 
434 

◆ test_fetch_result_response_wrong_event_uid()

def tests.test_dtm_TasksManager.TestTasksManager.test_fetch_result_response_wrong_event_uid (   self)

Definition at line 435 of file test_dtm_TasksManager.py.

435  def test_fetch_result_response_wrong_event_uid(self):
436  eeResponseData = EEResponseData(self.taskId)
437  eeResponseDataEvent = self.eventBuilder.build(EVENT_TYPES.FETCH_TASK_RESULTS_RESPONSE, eeResponseData)
438  self.pollerManager_mock.poll.return_value = [self.tasksManager.SERVER]
439  self.serverConnection_mock.recv.return_value = eeResponseDataEvent
440 
441  self.tasksManager.poll()
442 
443  self.assertEqual(len(self.tasksManager.fetchEvents), 0, "event is still here")
444  self.assertEqual(self.serverConnection_mock.send.call_count, 0, "send reply")
445 
446 

◆ test_get_status_request()

def tests.test_dtm_TasksManager.TestTasksManager.test_get_status_request (   self)

Definition at line 389 of file test_dtm_TasksManager.py.

389  def test_get_status_request(self):
390  getTasksStatus = GetTasksStatus([self.taskId])
391  getTasksStatusEvent = self.eventBuilder.build(EVENT_TYPES.GET_TASK_STATUS, getTasksStatus)
392  self.pollerManager_mock.poll.return_value = [self.tasksManager.SERVER]
393  self.serverConnection_mock.recv.return_value = getTasksStatusEvent
394 
395  self.tasksManager.poll()
396  self.serverConnection_mock.send.assert_called_once_with(ANY)
397 
398 

◆ test_get_task_fields()

def tests.test_dtm_TasksManager.TestTasksManager.test_get_task_fields (   self)

Definition at line 399 of file test_dtm_TasksManager.py.

399  def test_get_task_fields(self):
400  getTaskManagerFields = GetTaskManagerFields(self.taskId)
401  getTaskManagerFieldsEvent = self.eventBuilder.build(EVENT_TYPES.GET_TASK_FIELDS, getTaskManagerFields)
402  self.pollerManager_mock.poll.return_value = [self.tasksManager.SERVER]
403  self.serverConnection_mock.recv.return_value = getTaskManagerFieldsEvent
404 
405  self.tasksManager.poll()
406  self.serverConnection_mock.send.assert_called_once_with(ANY)
407 
408 

◆ test_new_task_tasksDataManager_response()

def tests.test_dtm_TasksManager.TestTasksManager.test_new_task_tasksDataManager_response (   self)

Definition at line 139 of file test_dtm_TasksManager.py.

139  def test_new_task_tasksDataManager_response(self):
140 
141  newTaskEvent = self.eventBuilder.build(EVENT_TYPES.SCHEDULE_TASK, self.newTask)
142  self.tasksManager.onNewTask(newTaskEvent)
143 
144  generalResponseEvent = self.eventBuilder.build(EVENT_TYPES.NEW_TASK_RESPONSE, self.generalResponse)
145  generalResponseEvent.uid = newTaskEvent.uid
146  self.pollerManager_mock.poll.return_value = [self.tasksManager.SERVER]
147  self.serverConnection_mock.recv.return_value = generalResponseEvent
148 
149  match_event = Matcher(matchShedulerNewEvents, newTaskEvent)
150  taskLog = self.tasksManager.createTaskLog(self.newTask)
151  self.tasksManager.poll()
152 
153  self.assertEqual(len(self.tasksManager.tasksQueue), 1, "task isnot in tasksQueue")
154  self.assertEqual(len(self.tasksManager.pendingTasks), 1, "")
155  self.assertTrue(self.check_task_in_dbi(taskLog, False), "insertion in dbi is broken")
156  self.scheduler_mock.send.assert_called_once_with(match_event)
157 
158 
Here is the call graph for this function:

◆ test_new_task_tasksDataManager_response_err()

def tests.test_dtm_TasksManager.TestTasksManager.test_new_task_tasksDataManager_response_err (   self)

Definition at line 176 of file test_dtm_TasksManager.py.

176  def test_new_task_tasksDataManager_response_err(self):
177 
178  newTaskEvent = self.eventBuilder.build(EVENT_TYPES.NEW_TASK, self.newTask)
179  newTaskEvent.connect_name = "server"
180  self.tasksManager.onNewTask(newTaskEvent)
181 
182  self.generalResponse.errorCode = self.generalResponse.ERROR_OK + 1
183  generalResponseEvent = self.eventBuilder.build(EVENT_TYPES.NEW_TASK_RESPONSE, self.generalResponse)
184  generalResponseEvent.uid = newTaskEvent.uid
185  self.pollerManager_mock.poll.return_value = [self.tasksManager.SERVER]
186  self.serverConnection_mock.recv.return_value = generalResponseEvent
187 
188  self.tasksManager.addConnection("server", self.serverConnection_mock)
189 
190  self.tasksManager.poll()
191 
192  self.assertEqual(len(self.tasksManager.tasksQueue), 0, "task is in tasksQueue")
193  self.assertEqual(len(self.tasksManager.pendingTasks), 0, "")
194  self.serverConnection_mock.send.assert_called_once_with(ANY) #send response
195  self.assertEqual(self.scheduler_mock.send.call_count, 0, "")
196 
197 
198 

◆ test_update_task()

def tests.test_dtm_TasksManager.TestTasksManager.test_update_task (   self)

Definition at line 280 of file test_dtm_TasksManager.py.

280  def test_update_task(self):
281  #add task
282  self.tasksManager.tasksQueue[self.newTask.id] = self.newTask
283 
284  updateTaskEvent = self.eventBuilder.build(EVENT_TYPES.UPDATE_TASK, self.updateTask)
285  self.pollerManager_mock.poll.return_value = [self.tasksManager.SERVER]
286  self.serverConnection_mock.recv.return_value = updateTaskEvent
287 
288  match_event = Matcher(matchShedulerNewEvents, updateTaskEvent)
289  self.tasksManager.poll()
290 
291  self.assertEqual(len(self.tasksManager.tasksQueue), 1, "task isnt in tasksQueue")
292  self.tasksDataManager_mock.send.assert_called_once_with(updateTaskEvent)
293  self.scheduler_mock.send.assert_called_once_with(match_event)
294  self.serverConnection_mock.send.assert_called_once_with(ANY)
295 
296 

◆ test_update_task_field()

def tests.test_dtm_TasksManager.TestTasksManager.test_update_task_field (   self)

Definition at line 447 of file test_dtm_TasksManager.py.

447  def test_update_task_field(self):
448  #add task
449  newTaskEvent = self.eventBuilder.build(EVENT_TYPES.NEW_TASK, self.newTask)
450  self.tasksManager.addNewTaskData(newTaskEvent)
451 
452  self.check_task_in_dbi(self.newTask, False)
453 
454  updateTaskFields = UpdateTaskFields(self.taskId)
455  updateTaskFields.fields["state"] = EEResponseData.TASK_STATE_TERMINATED
456 
457  updateTaskEvent = self.eventBuilder.build(EVENT_TYPES.UPDATE_TASK_FIELDS, updateTaskFields)
458  self.pollerManager_mock.poll.return_value = [self.tasksManager.SERVER]
459  self.serverConnection_mock.recv.return_value = updateTaskEvent
460 
461  self.tasksManager.poll()
462 
463  self.assertEqual(len(self.tasksManager.tasksQueue), 0, "task is in tasksQueue")
464  self.assertEquals(self.tasksDataManager_mock.send.call_count, 2, "")
465 
466  lookBackTaskLogScheme = TaskBackLogScheme(self.newTask)
467  resBackTaskLogScheme = self.tasksManager.dbi.fetch(lookBackTaskLogScheme, "id=%s" % self.taskId)[0]
468  assert resBackTaskLogScheme is None
469 
470  lookTaskLogScheme = TaskLogScheme(self.newTask)
471  resTaskLogScheme = self.tasksManager.dbi.fetch(lookTaskLogScheme, "id=%s" % self.taskId)[0]
472  assert resTaskLogScheme is not None
473 
Here is the call graph for this function:

◆ test_update_task_field1()

def tests.test_dtm_TasksManager.TestTasksManager.test_update_task_field1 (   self)

Definition at line 409 of file test_dtm_TasksManager.py.

409  def test_update_task_field1(self):
410  updateTaskFields = UpdateTaskFields(self.taskId)
411  updateTaskFieldsEvent = self.eventBuilder.build(EVENT_TYPES.UPDATE_TASK_FIELDS, updateTaskFields)
412  self.pollerManager_mock.poll.return_value = [self.tasksManager.SERVER]
413  self.serverConnection_mock.recv.return_value = updateTaskFieldsEvent
414 
415  self.tasksManager.poll()
416 
417 

◆ test_update_task_not_procssing_task()

def tests.test_dtm_TasksManager.TestTasksManager.test_update_task_not_procssing_task (   self)

Definition at line 297 of file test_dtm_TasksManager.py.

297  def test_update_task_not_procssing_task(self):
298  updateTaskEvent = self.eventBuilder.build(EVENT_TYPES.UPDATE_TASK, self.updateTask)
299  self.pollerManager_mock.poll.return_value = [self.tasksManager.SERVER]
300  self.serverConnection_mock.recv.return_value = updateTaskEvent
301 
302  self.tasksManager.poll()
303 
304  self.assertEqual(self.tasksDataManager_mock.send.call_count, 0, "event send to tasksDataManager")
305  self.assertEqual(self.scheduler_mock.send.call_count, 0, "event send to scheduler")
306  self.serverConnection_mock.send.assert_called_once_with(ANY)
307 
308 

Member Data Documentation

◆ adminServerFake

tests.test_dtm_TasksManager.TestTasksManager.adminServerFake

Definition at line 78 of file test_dtm_TasksManager.py.

◆ config

tests.test_dtm_TasksManager.TestTasksManager.config

Definition at line 65 of file test_dtm_TasksManager.py.

◆ connectionBuilderLight_mock

tests.test_dtm_TasksManager.TestTasksManager.connectionBuilderLight_mock

Definition at line 81 of file test_dtm_TasksManager.py.

◆ connections_mock

tests.test_dtm_TasksManager.TestTasksManager.connections_mock

Definition at line 87 of file test_dtm_TasksManager.py.

◆ dbi_mock

tests.test_dtm_TasksManager.TestTasksManager.dbi_mock

Definition at line 83 of file test_dtm_TasksManager.py.

◆ deleteTasks

tests.test_dtm_TasksManager.TestTasksManager.deleteTasks

Definition at line 98 of file test_dtm_TasksManager.py.

◆ eventBuilder

tests.test_dtm_TasksManager.TestTasksManager.eventBuilder

Definition at line 92 of file test_dtm_TasksManager.py.

◆ fetchTasksResultsFromCache

tests.test_dtm_TasksManager.TestTasksManager.fetchTasksResultsFromCache

Definition at line 99 of file test_dtm_TasksManager.py.

◆ generalResponse

tests.test_dtm_TasksManager.TestTasksManager.generalResponse

Definition at line 96 of file test_dtm_TasksManager.py.

◆ newTask

tests.test_dtm_TasksManager.TestTasksManager.newTask

Definition at line 95 of file test_dtm_TasksManager.py.

◆ pollerManager_mock

tests.test_dtm_TasksManager.TestTasksManager.pollerManager_mock

Definition at line 80 of file test_dtm_TasksManager.py.

◆ scheduler_mock

tests.test_dtm_TasksManager.TestTasksManager.scheduler_mock

Definition at line 86 of file test_dtm_TasksManager.py.

◆ serverConnection_mock

tests.test_dtm_TasksManager.TestTasksManager.serverConnection_mock

Definition at line 84 of file test_dtm_TasksManager.py.

◆ taskId

tests.test_dtm_TasksManager.TestTasksManager.taskId

Definition at line 94 of file test_dtm_TasksManager.py.

◆ tasksDataManager_mock

tests.test_dtm_TasksManager.TestTasksManager.tasksDataManager_mock

Definition at line 85 of file test_dtm_TasksManager.py.

◆ tasksManager

tests.test_dtm_TasksManager.TestTasksManager.tasksManager

Definition at line 91 of file test_dtm_TasksManager.py.

◆ updateTask

tests.test_dtm_TasksManager.TestTasksManager.updateTask

Definition at line 97 of file test_dtm_TasksManager.py.

The documentation for this class was generated from the following file: