Ejemplo n.º 1
0
    def test_init(self):
        td_fdw = self.td_fdw
        self.assertEqual(td_fdw.apikey, 'apikey1234')
        self.assertEqual(td_fdw.database, 'foodb')
        self.assertEqual(td_fdw.table, 'bartbl')
        self.assertEqual(td_fdw.query_engine, 'presto')
        self.assertEqual(td_fdw.columns, ['name', 'age'])

        td_fdw = TreasureDataFdw(
            {
                'apikey': 'apikey1234',
                'database': 'foodb',
                'query': 'SELECT code, COUNT(1) from access_log GROUP BY code',
                'query_engine': 'hive',
            }, ['name', 'age'])
        self.assertEqual(td_fdw.apikey, 'apikey1234')
        self.assertEqual(td_fdw.database, 'foodb')
        self.assertEqual(
            td_fdw.query,
            'SELECT code, COUNT(1) from access_log GROUP BY code')
        self.assertEqual(td_fdw.query_engine, 'hive')
        self.assertEqual(td_fdw.columns, ['name', 'age'])

        td_fdw = TreasureDataFdw({}, [])
        self.assertEqual(td_fdw.apikey, None)
        self.assertEqual(td_fdw.database, None)
        self.assertEqual(td_fdw.query, None)
        self.assertEqual(td_fdw.query_engine, 'presto')
        self.assertEqual(td_fdw.columns, [])
Ejemplo n.º 2
0
 def setUp(self):
     self.td_fdw = TreasureDataFdw(
         {
             'apikey': 'apikey1234',
             'database': 'foodb',
             'table': 'bartbl'
         }, ['name', 'age'])