Пример #1
0
  def test_get_jobs(self):
    tracker = TrackerFile('foo')
    tracker.get_lines = MagicMock()
    tracker.get_lines.return_value = [
      "__VRTrack_JOB_TYPE__ /parent_dir/child_dir/job_1.conf",
      "#admin_approval_required#__VRTrack_JOB_TYPE__ /parent_dir/child_dir/job_2.conf",
      "bad config_line"
    ]

    jobs = tracker.get_jobs()
    self.assertEqual(jobs[0].job_type, '__VRTrack_JOB_TYPE__')
    self.assertEqual(jobs[0].config_file, '/parent_dir/child_dir/job_1.conf')
    self.assertEqual(jobs[0].approval_required, False)
    self.assertIsInstance(jobs[0], PipelineJob)

    self.assertEqual(jobs[1].job_type, '__VRTrack_JOB_TYPE__')
    self.assertEqual(jobs[1].config_file, '/parent_dir/child_dir/job_2.conf')
    self.assertEqual(jobs[1].approval_required, True)
    self.assertIsInstance(jobs[1], PipelineJob)

    self.assertEqual(len(jobs), 2)
Пример #2
0
  def test_get_jobs(self):
    tracker = TrackerFile('foo')
    tracker.get_lines = MagicMock()
    tracker.get_lines.return_value = [
      "__VRTrack_JOB_TYPE__ /parent_dir/child_dir/job_1.conf",
      "#admin_approval_required#__VRTrack_JOB_TYPE__ /parent_dir/child_dir/job_2.conf",
      "bad config_line"
    ]

    jobs = tracker.get_jobs()
    self.assertEqual(jobs[0].job_type, '__VRTrack_JOB_TYPE__')
    self.assertEqual(jobs[0].config_file, '/parent_dir/child_dir/job_1.conf')
    self.assertEqual(jobs[0].approval_required, False)
    self.assertIsInstance(jobs[0], PipelineJob)

    self.assertEqual(jobs[1].job_type, '__VRTrack_JOB_TYPE__')
    self.assertEqual(jobs[1].config_file, '/parent_dir/child_dir/job_2.conf')
    self.assertEqual(jobs[1].approval_required, True)
    self.assertIsInstance(jobs[1], PipelineJob)

    self.assertEqual(len(jobs), 2)
Пример #3
0
    def test_get_lines(self, open_mock):
        file_like_object = StringIO('line 1\nline 2\n')
        open_mock.return_value.__enter__.return_value = file_like_object

        tracker = TrackerFile('foo')
        self.assertEquals(tracker.get_lines(), ['line 1', 'line 2'])
Пример #4
0
  def test_get_lines(self, open_mock):
    file_like_object = StringIO('line 1\nline 2\n')
    open_mock.return_value.__enter__.return_value = file_like_object

    tracker = TrackerFile('foo')
    self.assertEquals(tracker.get_lines(), ['line 1', 'line 2'])