Gitiles
Code Review Sign In
gerrit-public.fairphone.software / platform / external / autotest / 076a655d23cfd631b73bb668de46a236c1c16497 / . / frontend / afe / models_test.py
blob: a4eabfd59dd81b81672828bca1662f54662a9dbe [file] [log] [blame]
showarded2afea2009-07-07 20:54:07 +0000[diff] [blame]1#!/usr/bin/python
2
Prathmesh Prabhu076a6552017-02-27 15:25:00 -0800[diff] [blame^]3import datetime
showarded2afea2009-07-07 20:54:07 +0000[diff] [blame]4import unittest
5import common
6from autotest_lib.frontend import setup_django_environment
7from autotest_lib.frontend.afe import frontend_test_utils
Prathmesh Prabhu076a6552017-02-27 15:25:00 -0800[diff] [blame^]8from autotest_lib.frontend.afe import models, model_attributes, model_logic
9from autotest_lib.client.common_lib import global_config
10from autotest_lib.client.common_lib import control_data
showarded2afea2009-07-07 20:54:07 +0000[diff] [blame]11
showarda5288b42009-07-28 20:06:08 +0000[diff] [blame]12
13class AclGroupTest(unittest.TestCase,
14 frontend_test_utils.FrontendTestMixin):
15 def setUp(self):
16 self._frontend_common_setup()
17
18
19 def tearDown(self):
20 self._frontend_common_teardown()
21
22
23 def _check_acls(self, host, acl_name_list):
24 actual_acl_names = [acl_group.name for acl_group
25 in host.aclgroup_set.all()]
26 self.assertEquals(set(actual_acl_names), set(acl_name_list))
27
28
29 def test_on_host_membership_change(self):
30 host1, host2 = self.hosts[1:3]
31 everyone_acl = models.AclGroup.objects.get(name='Everyone')
32
33 host1.aclgroup_set.clear()
34 self._check_acls(host1, [])
35 host2.aclgroup_set.add(everyone_acl)
36 self._check_acls(host2, ['Everyone', 'my_acl'])
37
38 models.AclGroup.on_host_membership_change()
39
40 self._check_acls(host1, ['Everyone'])
41 self._check_acls(host2, ['my_acl'])
42
43
showardafd97de2009-10-01 18:45:09 +0000[diff] [blame]44class HostTest(unittest.TestCase,
45 frontend_test_utils.FrontendTestMixin):
46 def setUp(self):
47 self._frontend_common_setup()
48
49
50 def tearDown(self):
51 self._frontend_common_teardown()
52
53
54 def test_add_host_previous_one_time_host(self):
55 # ensure that when adding a host which was previously used as a one-time
56 # host, the status isn't reset, since this can interfere with the
57 # scheduler.
58 host = models.Host.create_one_time_host('othost')
59 self.assertEquals(host.invalid, True)
60 self.assertEquals(host.status, models.Host.Status.READY)
61
62 host.status = models.Host.Status.RUNNING
63 host.save()
64
65 host2 = models.Host.add_object(hostname='othost')
66 self.assertEquals(host2.id, host.id)
67 self.assertEquals(host2.status, models.Host.Status.RUNNING)
68
69
Dan Shib5b8b4f2016-11-02 14:04:02 -0700[diff] [blame]70 def test_check_board_labels_allowed(self):
Shuqian Zhao40e182b2016-10-11 11:55:11 -0700[diff] [blame]71 host = models.Host.create_one_time_host('othost')
72 # First check with host with no board label.
Dan Shib5b8b4f2016-11-02 14:04:02 -0700[diff] [blame]73 self.assertEqual(host.check_board_labels_allowed([host]), None)
Shuqian Zhao40e182b2016-10-11 11:55:11 -0700[diff] [blame]74
75 # Second check with host with board label
76 label = models.Label.add_object(name='board:test')
77 label.host_set.add(host)
78 self.assertRaises(model_logic.ValidationError,
Dan Shib5b8b4f2016-11-02 14:04:02 -0700[diff] [blame]79 host.check_board_labels_allowed, [host],
80 ['board:new_board'])
Shuqian Zhao40e182b2016-10-11 11:55:11 -0700[diff] [blame]81
82
showarded2afea2009-07-07 20:54:07 +0000[diff] [blame]83class SpecialTaskUnittest(unittest.TestCase,
84 frontend_test_utils.FrontendTestMixin):
85 def setUp(self):
86 self._frontend_common_setup()
87
88
89 def tearDown(self):
90 self._frontend_common_teardown()
91
92
showardc0ac3a72009-07-08 21:14:45 +0000[diff] [blame]93 def _create_task(self):
94 return models.SpecialTask.objects.create(
jamesren76fcf192010-04-21 20:39:50 +0000[diff] [blame]95 host=self.hosts[0], task=models.SpecialTask.Task.VERIFY,
96 requested_by=models.User.current_user())
showarded2afea2009-07-07 20:54:07 +0000[diff] [blame]97
showardc0ac3a72009-07-08 21:14:45 +0000[diff] [blame]98
99 def test_execution_path(self):
100 task = self._create_task()
showarded2afea2009-07-07 20:54:07 +0000[diff] [blame]101 self.assertEquals(task.execution_path(), 'hosts/host1/1-verify')
102
103
showardc0ac3a72009-07-08 21:14:45 +0000[diff] [blame]104 def test_status(self):
105 task = self._create_task()
106 self.assertEquals(task.status, 'Queued')
107
108 task.update_object(is_active=True)
109 self.assertEquals(task.status, 'Running')
110
showarde60e44e2009-11-13 20:45:38 +0000[diff] [blame]111 task.update_object(is_active=False, is_complete=True, success=True)
showardc0ac3a72009-07-08 21:14:45 +0000[diff] [blame]112 self.assertEquals(task.status, 'Completed')
113
showarde60e44e2009-11-13 20:45:38 +0000[diff] [blame]114 task.update_object(success=False)
115 self.assertEquals(task.status, 'Failed')
116
showardc0ac3a72009-07-08 21:14:45 +0000[diff] [blame]117
showard97446882009-07-20 22:37:28 +0000[diff] [blame]118 def test_activate(self):
119 task = self._create_task()
120 task.activate()
121 self.assertTrue(task.is_active)
122 self.assertFalse(task.is_complete)
123
124
125 def test_finish(self):
126 task = self._create_task()
127 task.activate()
showarde60e44e2009-11-13 20:45:38 +0000[diff] [blame]128 task.finish(True)
showard97446882009-07-20 22:37:28 +0000[diff] [blame]129 self.assertFalse(task.is_active)
130 self.assertTrue(task.is_complete)
showarde60e44e2009-11-13 20:45:38 +0000[diff] [blame]131 self.assertTrue(task.success)
showard97446882009-07-20 22:37:28 +0000[diff] [blame]132
133
showard9bb960b2009-11-19 01:02:11 +0000[diff] [blame]134 def test_requested_by_from_queue_entry(self):
135 job = self._create_job(hosts=[0])
136 task = models.SpecialTask.objects.create(
137 host=self.hosts[0], task=models.SpecialTask.Task.VERIFY,
138 queue_entry=job.hostqueueentry_set.all()[0])
showardfd8b89f2010-01-20 19:06:30 +0000[diff] [blame]139 self.assertEquals(task.requested_by.login, 'autotest_system')
showard9bb960b2009-11-19 01:02:11 +0000[diff] [blame]140
141
showardd1195652009-12-08 22:21:02 +0000[diff] [blame]142class HostQueueEntryUnittest(unittest.TestCase,
143 frontend_test_utils.FrontendTestMixin):
144 def setUp(self):
145 self._frontend_common_setup()
146
147
148 def tearDown(self):
149 self._frontend_common_teardown()
150
151
152 def test_execution_path(self):
153 entry = self._create_job(hosts=[1]).hostqueueentry_set.all()[0]
154 entry.execution_subdir = 'subdir'
155 entry.save()
156
showardfd8b89f2010-01-20 19:06:30 +0000[diff] [blame]157 self.assertEquals(entry.execution_path(), '1-autotest_system/subdir')
showardd1195652009-12-08 22:21:02 +0000[diff] [blame]158
159
jamesrene3656232010-03-02 00:00:30 +0000[diff] [blame]160class ModelWithInvalidTest(unittest.TestCase,
161 frontend_test_utils.FrontendTestMixin):
162 def setUp(self):
163 self._frontend_common_setup()
164
165
166 def tearDown(self):
167 self._frontend_common_teardown()
168
169
170 def test_model_with_invalid_delete(self):
171 self.assertFalse(self.hosts[0].invalid)
172 self.hosts[0].delete()
173 self.assertTrue(self.hosts[0].invalid)
174 self.assertTrue(models.Host.objects.get(id=self.hosts[0].id))
175
176
177 def test_model_with_invalid_delete_queryset(self):
178 for host in self.hosts:
179 self.assertFalse(host.invalid)
180
181 hosts = models.Host.objects.all()
182 hosts.delete()
183 self.assertEqual(hosts.count(), len(self.hosts))
184
185 for host in hosts:
186 self.assertTrue(host.invalid)
187
188
189 def test_cloned_queryset_delete(self):
190 """
191 Make sure that a cloned queryset maintains the custom delete()
192 """
193 to_delete = ('host1', 'host2')
194
195 for host in self.hosts:
196 self.assertFalse(host.invalid)
197
198 hosts = models.Host.objects.all().filter(hostname__in=to_delete)
199 hosts.delete()
200 all_hosts = models.Host.objects.all()
201 self.assertEqual(all_hosts.count(), len(self.hosts))
202
203 for host in all_hosts:
204 if host.hostname in to_delete:
205 self.assertTrue(
206 host.invalid,
207 '%s.invalid expected to be True' % host.hostname)
208 else:
209 self.assertFalse(
210 host.invalid,
211 '%s.invalid expected to be False' % host.hostname)
212
213
214 def test_normal_delete(self):
215 job = self._create_job(hosts=[1])
216 self.assertEqual(1, models.Job.objects.all().count())
217
218 job.delete()
219 self.assertEqual(0, models.Job.objects.all().count())
220
221
222 def test_normal_delete_queryset(self):
223 self._create_job(hosts=[1])
224 self._create_job(hosts=[2])
225
226 self.assertEqual(2, models.Job.objects.all().count())
227
228 models.Job.objects.all().delete()
229 self.assertEqual(0, models.Job.objects.all().count())
230
231
Shuqian Zhao54a5b672016-05-11 22:12:17 +0000[diff] [blame]232class KernelTest(unittest.TestCase, frontend_test_utils.FrontendTestMixin):
233 def setUp(self):
234 self._frontend_common_setup()
235
236
237 def tearDown(self):
238 self._frontend_common_teardown()
239
240
241 def test_create_kernels_none(self):
242 self.assertEqual(None, models.Kernel.create_kernels(None))
243
244
245 def test_create_kernels(self):
246 self.god.stub_function(models.Kernel, '_create')
247
248 num_kernels = 3
249 kernel_list = [object() for _ in range(num_kernels)]
250 result = [object() for _ in range(num_kernels)]
251
252 for kernel, response in zip(kernel_list, result):
253 models.Kernel._create.expect_call(kernel).and_return(response)
254 self.assertEqual(result, models.Kernel.create_kernels(kernel_list))
255 self.god.check_playback()
256
257
258 def test_create(self):
259 kernel = models.Kernel._create({'version': 'version'})
260 self.assertEqual(kernel.version, 'version')
261 self.assertEqual(kernel.cmdline, '')
262 self.assertEqual(kernel, models.Kernel._create({'version': 'version'}))
263
264
jamesren4a41e012010-07-16 22:33:48 +0000[diff] [blame]265class ParameterizedJobTest(unittest.TestCase,
266 frontend_test_utils.FrontendTestMixin):
267 def setUp(self):
268 self._frontend_common_setup()
269
270
271 def tearDown(self):
272 self._frontend_common_teardown()
273
274
275 def test_job(self):
276 global_config.global_config.override_config_value(
277 'AUTOTEST_WEB', 'parameterized_jobs', 'True')
278
279 test = models.Test.objects.create(
280 name='name', author='author', test_class='class',
281 test_category='category',
Aviv Keshet3dd8beb2013-05-13 17:36:04 -0700[diff] [blame]282 test_type=control_data.CONTROL_TYPE.SERVER, path='path')
jamesren4a41e012010-07-16 22:33:48 +0000[diff] [blame]283 parameterized_job = models.ParameterizedJob.objects.create(test=test)
284 job = self._create_job(hosts=[1], control_file=None,
285 parameterized_job=parameterized_job)
286
287 self.assertEqual(job, parameterized_job.job())
288
289
290class JobTest(unittest.TestCase, frontend_test_utils.FrontendTestMixin):
291 def setUp(self):
292 self._frontend_common_setup()
293
294
295 def tearDown(self):
296 self._frontend_common_teardown()
297
298
299 def test_check_parameterized_jobs_no_args(self):
300 self.assertRaises(Exception, models.Job.check_parameterized_job,
301 control_file=None, parameterized_job=None)
302
303
304 def test_check_parameterized_jobs_both_args(self):
305 self.assertRaises(Exception, models.Job.check_parameterized_job,
306 control_file=object(), parameterized_job=object())
307
308
309 def test_check_parameterized_jobs_disabled(self):
310 self.assertRaises(Exception, models.Job.check_parameterized_job,
311 control_file=None, parameterized_job=object())
312
313
314 def test_check_parameterized_jobs_enabled(self):
315 global_config.global_config.override_config_value(
316 'AUTOTEST_WEB', 'parameterized_jobs', 'True')
317 self.assertRaises(Exception, models.Job.check_parameterized_job,
318 control_file=object(), parameterized_job=None)
319
320
Jakob Juelich3bb7c802014-09-02 16:31:11 -0700[diff] [blame]321class SerializationTest(unittest.TestCase,
322 frontend_test_utils.FrontendTestMixin):
323 def setUp(self):
Jakob Juelichf88fa932014-09-03 17:58:04 -0700[diff] [blame]324 self._frontend_common_setup(fill_data=False)
Jakob Juelich3bb7c802014-09-02 16:31:11 -0700[diff] [blame]325
326
327 def tearDown(self):
328 self._frontend_common_teardown()
329
330
Jakob Juelichf88fa932014-09-03 17:58:04 -0700[diff] [blame]331 def _get_example_response(self):
332 return {'hosts': [{'aclgroup_set': [{'description': '',
333 'id': 1,
334 'name': 'Everyone',
335 'users': [{
336 'access_level': 100,
337 'id': 1,
338 'login': 'autotest_system',
339 'reboot_after': 0,
340 'reboot_before': 1,
341 'show_experimental': False}]}],
342 'dirty': True,
343 'hostattribute_set': [],
344 'hostname': '100.107.2.163',
345 'id': 2,
346 'invalid': False,
347 'labels': [{'id': 7,
348 'invalid': False,
349 'kernel_config': '',
350 'name': 'power:battery',
351 'only_if_needed': False,
352 'platform': False},
353 {'id': 9,
354 'invalid': False,
355 'kernel_config': '',
356 'name': 'hw_video_acc_h264',
357 'only_if_needed': False,
358 'platform': False},
359 {'id': 10,
360 'invalid': False,
361 'kernel_config': '',
362 'name': 'hw_video_acc_enc_h264',
363 'only_if_needed': False,
364 'platform': False},
365 {'id': 11,
366 'invalid': False,
367 'kernel_config': '',
368 'name': 'webcam',
369 'only_if_needed': False,
370 'platform': False},
371 {'id': 12,
372 'invalid': False,
373 'kernel_config': '',
374 'name': 'touchpad',
375 'only_if_needed': False,
376 'platform': False},
377 {'id': 13,
378 'invalid': False,
379 'kernel_config': '',
380 'name': 'spring',
381 'only_if_needed': False,
382 'platform': False},
383 {'id': 14,
384 'invalid': False,
385 'kernel_config': '',
386 'name': 'board:daisy',
387 'only_if_needed': False,
388 'platform': True},
389 {'id': 15,
390 'invalid': False,
391 'kernel_config': '',
392 'name': 'board_freq_mem:daisy_1.7GHz',
393 'only_if_needed': False,
394 'platform': False},
395 {'id': 16,
396 'invalid': False,
397 'kernel_config': '',
398 'name': 'bluetooth',
399 'only_if_needed': False,
400 'platform': False},
401 {'id': 17,
402 'invalid': False,
403 'kernel_config': '',
404 'name': 'gpu_family:mali',
405 'only_if_needed': False,
406 'platform': False},
Jakob Juelichf88fa932014-09-03 17:58:04 -0700[diff] [blame]407 {'id': 19,
408 'invalid': False,
409 'kernel_config': '',
410 'name': 'ec:cros',
411 'only_if_needed': False,
412 'platform': False},
413 {'id': 20,
414 'invalid': False,
415 'kernel_config': '',
416 'name': 'storage:mmc',
417 'only_if_needed': False,
418 'platform': False},
419 {'id': 21,
420 'invalid': False,
421 'kernel_config': '',
422 'name': 'hw_video_acc_vp8',
423 'only_if_needed': False,
424 'platform': False},
425 {'id': 22,
426 'invalid': False,
427 'kernel_config': '',
428 'name': 'video_glitch_detection',
429 'only_if_needed': False,
430 'platform': False},
431 {'id': 23,
432 'invalid': False,
433 'kernel_config': '',
434 'name': 'pool:suites',
435 'only_if_needed': False,
436 'platform': False},
437 {'id': 25,
438 'invalid': False,
439 'kernel_config': '',
440 'name': 'daisy-board-name',
441 'only_if_needed': False,
442 'platform': False}],
443 'leased': False,
Matthew Sartori68186332015-04-27 17:19:53 -0700[diff] [blame]444 'lock_reason': '',
Jakob Juelichf88fa932014-09-03 17:58:04 -0700[diff] [blame]445 'lock_time': None,
446 'locked': False,
447 'protection': 0,
448 'shard': {'hostname': '1', 'id': 1},
Prathmesh Prabhu076a6552017-02-27 15:25:00 -0800[diff] [blame^]449 'status': 'Ready',
450 'synch_id': None}],
Jakob Juelichf88fa932014-09-03 17:58:04 -0700[diff] [blame]451 'jobs': [{'control_file': 'some control file\n\n\n',
452 'control_type': 2,
453 'created_on': '2014-09-04T13:09:35',
454 'dependency_labels': [{'id': 14,
455 'invalid': False,
456 'kernel_config': '',
457 'name': 'board:daisy',
458 'only_if_needed': False,
459 'platform': True},
460 {'id': 23,
461 'invalid': False,
462 'kernel_config': '',
463 'name': 'pool:suites',
464 'only_if_needed': False,
465 'platform': False},
466 {'id': 25,
467 'invalid': False,
468 'kernel_config': '',
469 'name': 'daisy-board-name',
470 'only_if_needed': False,
471 'platform': False}],
472 'email_list': '',
473 'hostqueueentry_set': [{'aborted': False,
474 'active': False,
475 'complete': False,
476 'deleted': False,
477 'execution_subdir': '',
478 'finished_on': None,
479 'id': 5,
480 'meta_host': {
481 'id': 14,
482 'invalid': False,
483 'kernel_config': '',
484 'name': 'board:daisy',
485 'only_if_needed': False,
486 'platform': True},
Prashanth Balasubramanian8c98ac12014-12-23 11:26:44 -0800[diff] [blame]487 'host_id': None,
Jakob Juelichf88fa932014-09-03 17:58:04 -0700[diff] [blame]488 'started_on': None,
489 'status': 'Queued'}],
490 'id': 5,
491 'jobkeyval_set': [{'id': 10,
Fang Dengff361592015-02-02 15:27:34 -0800[diff] [blame]492 'job_id': 5,
Jakob Juelichf88fa932014-09-03 17:58:04 -0700[diff] [blame]493 'key': 'suite',
494 'value': 'dummy'},
495 {'id': 11,
Fang Dengff361592015-02-02 15:27:34 -0800[diff] [blame]496 'job_id': 5,
Jakob Juelichf88fa932014-09-03 17:58:04 -0700[diff] [blame]497 'key': 'build',
498 'value': 'daisy-release'},
499 {'id': 12,
Fang Dengff361592015-02-02 15:27:34 -0800[diff] [blame]500 'job_id': 5,
Jakob Juelichf88fa932014-09-03 17:58:04 -0700[diff] [blame]501 'key': 'experimental',
502 'value': 'False'}],
503 'max_runtime_hrs': 72,
504 'max_runtime_mins': 1440,
505 'name': 'daisy-experimental',
506 'owner': 'autotest',
507 'parse_failed_repair': True,
508 'priority': 40,
509 'reboot_after': 0,
510 'reboot_before': 1,
511 'run_reset': True,
512 'run_verify': False,
513 'shard': {'hostname': '1', 'id': 1},
514 'synch_count': 1,
515 'test_retry': 0,
516 'timeout': 24,
Dan Shic9e17142015-02-19 11:50:55 -0800[diff] [blame]517 'timeout_mins': 1440,
518 'require_ssp': None},
Jakob Juelichf88fa932014-09-03 17:58:04 -0700[diff] [blame]519 {'control_file': 'some control file\n\n\n',
520 'control_type': 2,
521 'created_on': '2014-09-04T13:09:35',
522 'dependency_labels': [{'id': 14,
523 'invalid': False,
524 'kernel_config': '',
525 'name': 'board:daisy',
526 'only_if_needed': False,
527 'platform': True},
528 {'id': 23,
529 'invalid': False,
530 'kernel_config': '',
531 'name': 'pool:suites',
532 'only_if_needed': False,
533 'platform': False},
534 {'id': 25,
535 'invalid': False,
536 'kernel_config': '',
537 'name': 'daisy-board-name',
538 'only_if_needed': False,
539 'platform': False}],
540 'email_list': '',
541 'hostqueueentry_set': [{'aborted': False,
542 'active': False,
543 'complete': False,
544 'deleted': False,
545 'execution_subdir': '',
546 'finished_on': None,
547 'id': 7,
548 'meta_host': {
549 'id': 14,
550 'invalid': False,
551 'kernel_config': '',
552 'name': 'board:daisy',
553 'only_if_needed': False,
554 'platform': True},
Prashanth Balasubramanian8c98ac12014-12-23 11:26:44 -0800[diff] [blame]555 'host_id': None,
Jakob Juelichf88fa932014-09-03 17:58:04 -0700[diff] [blame]556 'started_on': None,
557 'status': 'Queued'}],
558 'id': 7,
559 'jobkeyval_set': [{'id': 16,
Fang Dengff361592015-02-02 15:27:34 -0800[diff] [blame]560 'job_id': 7,
Jakob Juelichf88fa932014-09-03 17:58:04 -0700[diff] [blame]561 'key': 'suite',
562 'value': 'dummy'},
563 {'id': 17,
Fang Dengff361592015-02-02 15:27:34 -0800[diff] [blame]564 'job_id': 7,
Jakob Juelichf88fa932014-09-03 17:58:04 -0700[diff] [blame]565 'key': 'build',
566 'value': 'daisy-release'},
567 {'id': 18,
Fang Dengff361592015-02-02 15:27:34 -0800[diff] [blame]568 'job_id': 7,
Jakob Juelichf88fa932014-09-03 17:58:04 -0700[diff] [blame]569 'key': 'experimental',
570 'value': 'False'}],
571 'max_runtime_hrs': 72,
572 'max_runtime_mins': 1440,
573 'name': 'daisy-experimental',
574 'owner': 'autotest',
575 'parse_failed_repair': True,
576 'priority': 40,
577 'reboot_after': 0,
578 'reboot_before': 1,
579 'run_reset': True,
580 'run_verify': False,
581 'shard': {'hostname': '1', 'id': 1},
582 'synch_count': 1,
583 'test_retry': 0,
584 'timeout': 24,
Dan Shic9e17142015-02-19 11:50:55 -0800[diff] [blame]585 'timeout_mins': 1440,
586 'require_ssp': None}]}
Jakob Juelich3bb7c802014-09-02 16:31:11 -0700[diff] [blame]587
Jakob Juelich3bb7c802014-09-02 16:31:11 -0700[diff] [blame]588
Jakob Juelichf88fa932014-09-03 17:58:04 -0700[diff] [blame]589 def test_response(self):
590 heartbeat_response = self._get_example_response()
591 hosts_serialized = heartbeat_response['hosts']
592 jobs_serialized = heartbeat_response['jobs']
Jakob Juelich3bb7c802014-09-02 16:31:11 -0700[diff] [blame]593
Jakob Juelichf88fa932014-09-03 17:58:04 -0700[diff] [blame]594 # Persisting is automatically done inside deserialize
595 hosts = [models.Host.deserialize(host) for host in hosts_serialized]
596 jobs = [models.Job.deserialize(job) for job in jobs_serialized]
Jakob Juelich3bb7c802014-09-02 16:31:11 -0700[diff] [blame]597
Jakob Juelichf88fa932014-09-03 17:58:04 -0700[diff] [blame]598 generated_heartbeat_response = {
599 'hosts': [host.serialize() for host in hosts],
600 'jobs': [job.serialize() for job in jobs]
601 }
Fang Dengff361592015-02-02 15:27:34 -0800[diff] [blame]602 example_response = self._get_example_response()
603 # For attribute-like objects, we don't care about its id.
604 for r in [generated_heartbeat_response, example_response]:
605 for job in r['jobs']:
606 for keyval in job['jobkeyval_set']:
607 keyval.pop('id')
608 for host in r['hosts']:
609 for attribute in host['hostattribute_set']:
610 keyval.pop('id')
611 self.assertEqual(generated_heartbeat_response, example_response)
Jakob Juelich3bb7c802014-09-02 16:31:11 -0700[diff] [blame]612
613
Jakob Juelich116ff0f2014-09-17 18:25:16 -0700[diff] [blame]614 def test_update(self):
615 job = self._create_job(hosts=[1])
616 serialized = job.serialize(include_dependencies=False)
617 serialized['owner'] = 'some_other_owner'
618
619 job.update_from_serialized(serialized)
620 self.assertEqual(job.owner, 'some_other_owner')
621
622 serialized = job.serialize()
623 self.assertRaises(
624 ValueError,
625 job.update_from_serialized, serialized)
626
627
Jakob Juelichf865d332014-09-29 10:47:49 -0700[diff] [blame]628 def test_sync_aborted(self):
629 job = self._create_job(hosts=[1])
630 serialized = job.serialize()
631
632 serialized['hostqueueentry_set'][0]['aborted'] = True
633 serialized['hostqueueentry_set'][0]['status'] = 'Running'
634
635 models.Job.deserialize(serialized)
636
637 job = models.Job.objects.get(pk=job.id)
638 self.assertTrue(job.hostqueueentry_set.all()[0].aborted)
639 self.assertEqual(job.hostqueueentry_set.all()[0].status, 'Queued')
Jakob Juelich116ff0f2014-09-17 18:25:16 -0700[diff] [blame]640
641
showarded2afea2009-07-07 20:54:07 +0000[diff] [blame]642if __name__ == '__main__':
643 unittest.main()