Пример #1
0
                              self.client.GetJobReference, invalid)

    def testRaiseError(self):
        # Confirm we handle arbitrary errors gracefully.
        try:
            bigquery_client.BigqueryClient.RaiseError({})
        except bigquery_client.BigqueryError as _:
            pass

    def testJsonToInsertEntry(self):
        result = [
            bigquery_client.JsonToInsertEntry(None, '{"a":1}'),
            bigquery_client.JsonToInsertEntry('key', '{"b":2}'),
        ]
        self.assertEquals([None, 'key'], [x[0] for x in result])
        self.assertEquals(1, result[0][1]['a'])
        self.assertEquals(2, result[1][1]['b'])

        self.assertRaisesRegexp(bigquery_client.BigqueryClientError,
                                r'Could not parse',
                                bigquery_client.JsonToInsertEntry, None,
                                '_junk_')
        self.assertRaisesRegexp(bigquery_client.BigqueryClientError,
                                r'not a JSON object',
                                bigquery_client.JsonToInsertEntry, None,
                                '[1, 2]')


if __name__ == '__main__':
    googletest.main()
class CsvFormatterTest(TableFormatterTest):

  def setUp(self):
    self.format_class = table_formatter.CsvFormatter
    super(CsvFormatterTest, self).setUp()

  def testStr(self):
    table_repr = '\n'.join((
        'foo,longer header',
        'a,3',
        'abc,123'))
    self.assertEquals(table_repr, str(self.formatter))


class NullFormatterTest(TableFormatterTest):

  def setUp(self):
    self.format_class = table_formatter.NullFormatter
    super(NullFormatterTest, self).setUp()

  def testStr(self):
    self.assertEquals('', str(self.formatter))

  def testUnicodeRow(self):
    self.assertEquals('', unicode(self.formatter))


if __name__ == '__main__':
  googletest.main()