def test_compile_to_clickhouse_dialect(self):
     meta = MetaData(bind=METRO_ENGINE_CH)
     table = Table('stock_data_tmp', meta, autoload=True, schema='co')
     query = select([table])
     actual = compile_query(query, 'clickhouse')
     expected = 'SELECT date, count, shop_id, plu, created \n' \
                'FROM co.stock_data_tmp'
     self.assertEqual(actual, expected)
    def test_compile_core_query_delete(self):
        c = self.table.c  # column
        query = delete(self.table).where(c.created == self.some_date)

        actual = compile_query(query)
        expected = "DELETE FROM transport.files " \
                   "WHERE transport.files.created = '2100-01-01'"
        self.assertEqual(expected, actual)
    def test_compile_core_query_insert(self):
        c = self.table.c  # column
        query = insert(self.table).values(created=str(self.some_date))

        actual = compile_query(query)
        expected = "INSERT INTO transport.files (created) " \
                   "VALUES ('2100-01-01')"
        self.assertEqual(expected, actual)
    def test_compile_core_query_select(self):
        c = self.table.c  # column
        query = select([c.created]).where(c.created == self.some_date)

        actual = compile_query(query)
        expected = "SELECT transport.files.created \n" \
                   "FROM transport.files \n" \
                   "WHERE transport.files.created = '2100-01-01'"
        self.assertEqual(expected, actual)
    def test_compile_core_query_update(self):
        c = self.table.c  # column
        query = update(self.table).where(c.created == self.some_date).values(
            created=str(self.some_date))

        actual = compile_query(query)
        expected = "UPDATE transport.files " \
                   "SET created='2100-01-01' " \
                   "WHERE transport.files.created = '2100-01-01'"
        self.assertEqual(expected, actual)
 def test_raises_when_incorrect_type_of_query(self):
     with self.assertRaises(QueryCompilerException):
         compile_query('unknown')
 def test_raises_when_unknown_dialect(self):
     with self.assertRaises(QueryCompilerException):
         compile_query(Selectable(), 'unknown')
 def test_compile_orm_query_select(self):
     query = Plu.session.query(Plu.id, Plu.name)
     expected = 'SELECT master_data.products.plu, master_data.products.title \n' \
                'FROM master_data.products'
     actual = compile_query(query)
     self.assertEqual(expected, actual)