# -*- coding: utf-8 -*-

#   Copyright (c) 2010-2014, MIT Probabilistic Computing Project
#
#   Licensed under the Apache License, Version 2.0 (the "License");
#   you may not use this file except in compliance with the License.
#   You may obtain a copy of the License at
#
#       http://www.apache.org/licenses/LICENSE-2.0
#
#   Unless required by applicable law or agreed to in writing, software
#   distributed under the License is distributed on an "AS IS" BASIS,
#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#   See the License for the specific language governing permissions and
#   limitations under the License.

import StringIO

import bayeslite.bql as bql
import bayeslite.parse as parse

import test_smoke

def bql2sql(string):
    with test_smoke.t1() as bdb:
        phrases = parse.parse_bql_string(string)
        out = StringIO.StringIO()
        bql.compile_bql(bdb, phrases, out)
        return out.getvalue()

def test_select_trivial():
    assert bql2sql('select 0;') == 'select 0;'
    assert bql2sql('select 0 as z;') == 'select 0 as "z";'
    assert bql2sql('select * from t;') == 'select * from "t";'
    assert bql2sql('select t.* from t;') == 'select "t".* from "t";'
    assert bql2sql('select c from t;') == 'select "c" from "t";'
    assert bql2sql('select c as d from t;') == 'select "c" as "d" from "t";'
    assert bql2sql('select t.c as d from t;') == \
        'select "t"."c" as "d" from "t";'
    assert bql2sql('select t.c as d, p as q, x from t;') == \
        'select "t"."c" as "d", "p" as "q", "x" from "t";'
    assert bql2sql('select * from t, u;') == 'select * from "t", "u";'
    assert bql2sql('select * from t as u;') == 'select * from "t" as "u";'
    assert bql2sql('select * where x;') == 'select * where "x";'
    assert bql2sql('select * from t where x;') == \
        'select * from "t" where "x";'
    assert bql2sql('select * group by x;') == 'select * group by "x";'
    assert bql2sql('select * from t where x group by y;') == \
        'select * from "t" where "x" group by "y";'
    assert bql2sql('select * from t where x group by y, z;') == \
        'select * from "t" where "x" group by "y", "z";'
    assert bql2sql('select * order by x;') == 'select * order by "x";'
    assert bql2sql('select * order by x asc;') == 'select * order by "x";'
    assert bql2sql('select * order by x desc;') == \
        'select * order by "x" desc;'
    assert bql2sql('select * order by x, y;') == 'select * order by "x", "y";'
    assert bql2sql('select * order by x desc, y;') == \
        'select * order by "x" desc, "y";'
    assert bql2sql('select * order by x, y asc;') == \
        'select * order by "x", "y";'
    assert bql2sql('select * limit 32;') == 'select * limit 32;'
    assert bql2sql('select * limit 32 offset 16;') == \
        'select * limit 32 offset 16;'
    assert bql2sql('select * limit 16, 32;') == 'select * limit 32 offset 16;'

@@ -794,19 +794,32 @@
 t1() as 
+(
 bdb
+, _table_id)
 :%0A      

Fix bql2sql utility in test_bql.py.