# Licensed to the Apache Software Foundation (ASF) under one or more

# contributor license agreements. See the NOTICE file distributed with

# this work for additional information regarding copyright ownership.

# The ASF licenses this file to You 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.

from pyignite.api import (

sql_fields, sql_fields_cursor_get_page,

cache_get_or_create, sql, sql_cursor_get_page,

cache_get_configuration,

)

from pyignite.datatypes.prop_codes import *

from pyignite.utils import entity_id, unwrap_binary

initial_data = [

('John', 'Doe', 5),

('Jane', 'Roe', 4),

('Joe', 'Bloggs', 4),

('Richard', 'Public', 3),

('Negidius', 'Numerius', 3),

]

create_query = '''CREATE TABLE Student (

id INT(11) PRIMARY KEY,

first_name CHAR(24),

last_name CHAR(32),

grade INT(11))'''

insert_query = '''INSERT INTO Student(id, first_name, last_name, grade)

VALUES (?, ?, ?, ?)'''

select_query = 'SELECT id, first_name, last_name, grade FROM Student'

drop_query = 'DROP TABLE Student IF EXISTS'

page_size = 4

def test_sql(client):

# cleanup

client.sql(drop_query)

result = sql_fields(

client,

'PUBLIC',

create_query,

page_size,

include_field_names=True

)

assert result.status == 0, result.message

for i, data_line in enumerate(initial_data, start=1):

fname, lname, grade = data_line

result = sql_fields(

client,

'PUBLIC',

insert_query,

page_size,

query_args=[i, fname, lname, grade],

include_field_names=True

)

assert result.status == 0, result.message

result = cache_get_configuration(client, 'SQL_PUBLIC_STUDENT')

assert result.status == 0, result.message

binary_type_name = result.value[PROP_QUERY_ENTITIES][0]['value_type_name']

result = sql(

client,

'SQL_PUBLIC_STUDENT',

binary_type_name,

'TRUE',

page_size

)

assert result.status == 0, result.message

assert len(result.value['data']) == page_size

assert result.value['more'] is True

for wrapped_object in result.value['data'].values():

data = unwrap_binary(client, wrapped_object)

assert data.type_id == entity_id(binary_type_name)

cursor = result.value['cursor']

while result.value['more']:

result = sql_cursor_get_page(client, cursor)

assert result.status == 0, result.message

for wrapped_object in result.value['data'].values():

data = unwrap_binary(client, wrapped_object)

assert data.type_id == entity_id(binary_type_name)

# repeat cleanup

result = sql_fields(client, 'PUBLIC', drop_query, page_size)

assert result.status == 0

def test_sql_fields(client):

# cleanup

client.sql(drop_query)

result = sql_fields(

client,

'PUBLIC',

create_query,

page_size,

include_field_names=True

)

assert result.status == 0, result.message

for i, data_line in enumerate(initial_data, start=1):

fname, lname, grade = data_line

result = sql_fields(

client,

'PUBLIC',

insert_query,

page_size,

query_args=[i, fname, lname, grade],

include_field_names=True

)

assert result.status == 0, result.message

result = sql_fields(

client,

'PUBLIC',

select_query,

page_size,

include_field_names=True

)

assert result.status == 0

assert len(result.value['data']) == page_size

assert result.value['more'] is True

cursor = result.value['cursor']

result = sql_fields_cursor_get_page(client, cursor, field_count=4)

assert result.status == 0

assert len(result.value['data']) == len(initial_data) - page_size

assert result.value['more'] is False

# repeat cleanup

result = sql_fields(client, 'PUBLIC', drop_query, page_size)

assert result.status == 0