示例#1
0
    def map_column(self, value):
        """
        Applied to each column of every row returned by `rows`.

        Default behaviour is to escape special characters and identify any self.null_values.
        """
        if value in self.null_values:
            return r'\\N'
        else:
            return default_escape(six.text_type(value))
示例#2
0
文件: postgres.py 项目: 01-/luigi
    def map_column(self, value):
        """
        Applied to each column of every row returned by `rows`.

        Default behaviour is to escape special characters and identify any self.null_values.
        """
        if value in self.null_values:
            return r'\\N'
        else:
            return default_escape(six.text_type(value))
示例#3
0
 def test_handle_failed_job(self, proc, file):
     proc.return_value.returncode = 1
     file.return_value = BytesIO(b"stderr")
     try:
         job = TestSparkSubmitTask()
         job.run()
     except SparkJobError as e:
         self.assertEqual(e.err, "stderr")
         self.assertTrue("STDERR: stderr" in six.text_type(e))
     else:
         self.fail("Should have thrown SparkJobError")
示例#4
0
 def test_handle_failed_job(self, proc, file):
     proc.return_value.returncode = 1
     file.return_value = BytesIO(b'stderr')
     try:
         job = TestExternalProgramTask()
         job.run()
     except ExternalProgramRunError as e:
         self.assertEqual(e.err, 'stderr')
         self.assertIn('STDERR: stderr', six.text_type(e))
     else:
         self.fail('Should have thrown SubprocessRunError')
示例#5
0
 def test_handle_failed_job(self, proc, file):
     proc.return_value.returncode = 1
     file.return_value = BytesIO(b'stderr')
     try:
         job = TestSparkSubmitTask()
         job.run()
     except SparkJobError as e:
         self.assertEqual(e.err, 'stderr')
         self.assertTrue('STDERR: stderr' in six.text_type(e))
     else:
         self.fail("Should have thrown SparkJobError")
示例#6
0
 def test_handle_failed_job(self, proc, file):
     proc.return_value.returncode = 1
     file.return_value = BytesIO(b"stderr")
     try:
         job = TestExternalProgramTask()
         job.run()
     except ExternalProgramRunError as e:
         self.assertEqual(e.err, "stderr")
         self.assertIn("STDERR: stderr", six.text_type(e))
     else:
         self.fail("Should have thrown SubprocessRunError")
示例#7
0
 def test_handle_failed_job(self, proc, file, logger):
     proc.return_value.returncode = 1
     file.return_value = BytesIO(b'stderr')
     try:
         job = TestExternalProgramTask()
         job.run()
     except ExternalProgramRunError as e:
         self.assertEqual(e.err, 'stderr')
         self.assertIn('STDERR: stderr', six.text_type(e))
         self.assertIn(call.info('Program stderr:\nstderr'), logger.mock_calls)
     else:
         self.fail('Should have thrown ExternalProgramRunError')
示例#8
0
 def test_handle_failed_job(self, proc, file, logger):
     proc.return_value.returncode = 1
     file.return_value = BytesIO(b'stderr')
     try:
         job = TestExternalProgramTask()
         job.run()
     except ExternalProgramRunError as e:
         self.assertEqual(e.err, 'stderr')
         self.assertIn('STDERR: stderr', six.text_type(e))
         self.assertIn(call.info('Program stderr:\nstderr'), logger.mock_calls)
     else:
         self.fail('Should have thrown ExternalProgramRunError')