/****************************************************************************

**

** Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies).

** All rights reserved.

** Contact: Nokia Corporation (qt-info@nokia.com)

**

** This file is part of the test suite of the Qt Toolkit.

**

** $QT_BEGIN_LICENSE:LGPL$

** No Commercial Usage

** This file contains pre-release code and may not be distributed.

** You may use this file in accordance with the terms and conditions

** contained in the Technology Preview License Agreement accompanying

** this package.

**

** GNU Lesser General Public License Usage

** Alternatively, this file may be used under the terms of the GNU Lesser

** General Public License version 2.1 as published by the Free Software

** Foundation and appearing in the file LICENSE.LGPL included in the

** packaging of this file.  Please review the following information to

** ensure the GNU Lesser General Public License version 2.1 requirements

** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.

**

** In addition, as a special exception, Nokia gives you certain additional

** rights.  These rights are described in the Nokia Qt LGPL Exception

** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.

**

** If you have questions regarding the use of this file, please contact

** Nokia at qt-info@nokia.com.

**

**

**

**

**

**

**

**

** $QT_END_LICENSE$

**

****************************************************************************/

#include <QtTest/QtTest>

#include <q3sqlcursor.h>

#include <qsqlfield.h>

#include <qsqldriver.h>

#include "../qsqldatabase/tst_databases.h"

//TESTED_FILES=

QT_FORWARD_DECLARE_CLASS(QSqlDatabase)

class tst_Q3SqlCursor : public QObject

{

Q_OBJECT

public:

    tst_Q3SqlCursor();

    virtual ~tst_Q3SqlCursor();

public slots:

    void initTestCase();

    void cleanupTestCase();

    void init();

    void cleanup();

private slots:

    void copyConstructor_data() { generic_data(); }

    void copyConstructor();

    void value_data() { generic_data(); }

    void value();

    void primaryIndex_data() { generic_data(); }

    void primaryIndex();

    void insert_data() { generic_data(); }

    void insert();

    void select_data() { generic_data(); }

    void select();

    void setFilter_data() { generic_data(); }

    void setFilter();

    void setName_data() { generic_data(); }

    void setName();

    // problem specific tests

    void unicode_data() { generic_data(); }

    void unicode();

    void precision_data() { generic_data(); }

    void precision();

    void insertORA_data() { generic_data("QOCI"); }

    void insertORA();

    void batchInsert_data() { generic_data(); }

    void batchInsert();

    void insertSpecial_data() { generic_data(); }

    void insertSpecial();

    void updateNoPK_data() { generic_data(); }

    void updateNoPK();

    void insertFieldNameContainsWS_data() { generic_data(); }

    void insertFieldNameContainsWS(); // For task 117996

private:

    void generic_data(const QString &engine=QString());

    void createTestTables( QSqlDatabase db );

    void dropTestTables( QSqlDatabase db );

    void populateTestTables( QSqlDatabase db );

    tst_Databases dbs;

};

tst_Q3SqlCursor::tst_Q3SqlCursor()

{

}

tst_Q3SqlCursor::~tst_Q3SqlCursor()

{

}

void tst_Q3SqlCursor::generic_data(const QString &engine)

{

    if ( dbs.fillTestTable(engine) == 0 ) {

        if(engine.isEmpty())

        else

           QSKIP( (QString("No database drivers of type %1 are available in this Qt configuration").arg(engine)).toLocal8Bit(), SkipAll );

    }

}

void tst_Q3SqlCursor::createTestTables( QSqlDatabase db )

{

    if ( !db.isValid() )

    QSqlQuery q( db );

    if (tst_Databases::isSqlServer(db)) {

        QVERIFY_SQL(q, exec("SET ANSI_DEFAULTS ON"));

        QVERIFY_SQL(q, exec("SET IMPLICIT_TRANSACTIONS OFF"));

    }

    else if(tst_Databases::isPostgreSQL(db))

        QVERIFY_SQL( q, exec("set client_min_messages='warning'"));

    // please never ever change this table; otherwise fix all tests ;)

    if ( tst_Databases::isMSAccess( db ) ) {

    } else {

    }

    if ( tst_Databases::isSqlServer( db ) ) {

    } else {

    }

    if (tst_Databases::isMSAccess(db)) {

    } else {

    }

}

void tst_Q3SqlCursor::dropTestTables( QSqlDatabase db )

{

    if ( !db.isValid() )

    QStringList tableNames;

    tst_Databases::safeDropTables( db, tableNames );

}

void tst_Q3SqlCursor::populateTestTables( QSqlDatabase db )

{

    if (!db.isValid())

    QSqlQuery q( db );

    q.addBindValue(QVariantList() << 0 << 1 << 2 << 3);

    q.addBindValue(QVariantList() << "VarChar0" << "VarChar1" << "VarChar2" << "VarChar3");

    q.addBindValue(QVariantList() << "Char0" << "Char1" << "Char2" << "Char3");

    q.addBindValue(QVariantList() << 1.1 << 2.2 << 3.3 << 4.4);

    QVERIFY_SQL(q, execBatch());

}

void tst_Q3SqlCursor::initTestCase()

{

    dbs.open();

    for ( QStringList::ConstIterator it = dbs.dbNames.begin(); it != dbs.dbNames.end(); ++it ) {

    }

}

void tst_Q3SqlCursor::cleanupTestCase()

{

    for ( QStringList::ConstIterator it = dbs.dbNames.begin(); it != dbs.dbNames.end(); ++it ) {

    }

    dbs.close();

}

void tst_Q3SqlCursor::init()

{

}

void tst_Q3SqlCursor::cleanup()

{

    QFETCH( QString, dbName );

    QSqlDatabase db = QSqlDatabase::database( dbName );

    CHECK_DATABASE( db );

    if ( QTest::currentTestFailed() ) {

    }

}

void tst_Q3SqlCursor::copyConstructor()

{

    QFETCH( QString, dbName );

    QSqlDatabase db = QSqlDatabase::database( dbName );

    CHECK_DATABASE( db );

    Q3SqlCursor cur2;

    {

    }

    QSqlRecord* rec = cur2.primeUpdate();

    Q_ASSERT( rec );

    QCOMPARE( (int)rec->count(), 4 );

    int i = 0;

    while ( cur2.next() ) {

    }

}

void tst_Q3SqlCursor::value()

{

    QFETCH( QString, dbName );

    QSqlDatabase db = QSqlDatabase::database( dbName );

    CHECK_DATABASE( db );

    QVERIFY_SQL(cur, select( cur.index( QString("id") ) ));

    int i = 0;

    while ( cur.next() ) {

    }

}

void tst_Q3SqlCursor::primaryIndex()

{

    QFETCH( QString, dbName );

    QSqlDatabase db = QSqlDatabase::database( dbName );

    CHECK_DATABASE( db );

    QSqlIndex index = cur.primaryIndex();

    if ( tst_Databases::isMSAccess( db ) ) {

    } else {

    }

    QVERIFY(!index.isDescending(0));

    QVERIFY(!index.isDescending(1));

}

void tst_Q3SqlCursor::insert()

{

    QFETCH( QString, dbName );

    QSqlDatabase db = QSqlDatabase::database( dbName );

    CHECK_DATABASE( db );

    QSqlRecord* irec = cur.primeInsert();

    QVERIFY( irec != 0 );

    // check that primeInsert returns a valid QSqlRecord

    QCOMPARE( (int)irec->count(), 4 );

    if ( ( irec->field( 0 ).type() != QVariant::Int ) &&

         ( irec->field( 0 ).type() != QVariant::Double ) ) {

    }

    QCOMPARE( QVariant::typeToName( irec->field( 1 ).type() ), QVariant::typeToName( QVariant::String ) );

    QCOMPARE( QVariant::typeToName( irec->field( 2 ).type() ), QVariant::typeToName( QVariant::String ) );

    QVERIFY((QVariant::typeToName(irec->field(3).type()) == QVariant::typeToName(QVariant::Double)) ||

            (QVariant::typeToName(irec->field(3).type()) == QVariant::typeToName(QVariant::String)));

    QCOMPARE( irec->field( 0 ).name().upper(), QString( "ID" ) );

    QCOMPARE( irec->field( 1 ).name().upper(), QString( "T_VARCHAR" ) );

    QCOMPARE( irec->field( 2 ).name().upper(), QString( "T_CHAR" ) );

    QCOMPARE( irec->field( 3 ).name().upper(), QString( "T_NUMERIC" ) );

    irec->setValue( "id", 400 );

    irec->setValue( "t_varchar", "SomeVarChar" );

    irec->setValue( "t_char", "SomeChar" );

    irec->setValue( "t_numeric", 400.400 );

    // restore old test-tables

    populateTestTables( db );

}

void tst_Q3SqlCursor::insertSpecial()

{

    QFETCH( QString, dbName );

    QSqlDatabase db = QSqlDatabase::database( dbName );

    CHECK_DATABASE( db );

    QSqlRecord* irec = cur.primeInsert();

    QVERIFY( irec != 0 );

    QStringList strings;

    strings << "StringWith'ATick" << "StringWith\"Doublequote" << "StringWith\\Backslash" << "StringWith~Tilde";

    strings << "StringWith%Percent" << "StringWith_Underscore" << "StringWith[SquareBracket" << "StringWith{Brace";

    strings << "StringWith''DoubleTick" << "StringWith\\Lot\\of\\Backslash" << "StringWith\"lot\"of\"quotes\"";

    strings << "'StartsAndEndsWithTick'" << "\"StartsAndEndsWithQuote\"";

    strings << "StringWith\nCR" << "StringWith\n\rCRLF";

    int i = 800;

    // INSERT the strings

    QStringList::Iterator it;

    for ( it = strings.begin(); it != strings.end(); ++it ) {

    }

    QVERIFY( cur.select( "id >= 800 and id < 900" ) );

    int i2 = 800;

    while( cur.next() ) {

    }

    QCOMPARE( i, i2 );

    populateTestTables( db );

}

void tst_Q3SqlCursor::batchInsert()

{

    QFETCH( QString, dbName );

    QSqlDatabase db = QSqlDatabase::database( dbName );

    CHECK_DATABASE( db );

    QSqlQuery q( db );

    int i = 0;

    for ( ; i < 100; ++i ) {

        }

    }

    for ( ; i < 200; ++i ) {

        QSqlRecord* irec = cur.primeInsert();

        Q_ASSERT( irec );

        irec->setValue( "id", i );

        irec->setValue( "t_varchar", "blah" );

        irec->setValue( "t_char", "blah" );

        irec->setValue( "t_numeric", 1.1 );

        }

    }

    i = 0;

    while ( q.next() ) {

    }

    QCOMPARE( i, 200 );

    populateTestTables( db );

}

static QString dumpUtf8( const QString& str )

{

    QString res;

    for ( int i = 0; i < (int)str.length(); ++i ) {

    }

    return res;

}

void tst_Q3SqlCursor::insertORA()

{

    QFETCH( QString, dbName );

    QSqlDatabase db = QSqlDatabase::database( dbName );

    CHECK_DATABASE( db );

    if (tst_Databases::getOraVersion(db) < 9)

    /****** CHARSET TEST ******/

    QSqlQuery q( db );

    static const QString val1( "blah1" );

    QSqlRecord* irec = cur.primeInsert();

    irec->setValue( "id", 1 );

    irec->setValue( "t_char", val1 );

    QVERIFY( cur.insert() );

    QVERIFY_SQL(cur, select());

    QVERIFY( cur.next() );

    if ( cur.value( "t_char" ).toString() != val1 )

    static const unsigned short utf8arr[] = { 0xd792,0xd79c,0xd792,0xd79c,0xd799,0x00 };

    static const QString utf8str = QString::fromUcs2( utf8arr );

    irec = cur.primeInsert();

    irec->setValue( "id", 2 );

    irec->setValue( "t_char", utf8str );

    QVERIFY( cur.insert() );

    QVERIFY_SQL(cur, select( "id=2" ));

    QVERIFY( cur.next() );

    // until qtest knows non-fatal errors we use qDebug instead

    if ( cur.value( "t_char" ).toString() != utf8str )

        qDebug( QString( "Wrong value for t_char: expected '%1', got '%2'" ).arg( dumpUtf8 ( utf8str ) ).arg(

                dumpUtf8( cur.value( "t_char" ).toString() ) ) );

    /****** NCHARSET TEST ********/

    irec = cur2.primeInsert();

    irec->setValue( "id", 1 );

    irec->setValue( "t_nchar", val1 );

    QVERIFY( cur2.insert() );

    QVERIFY_SQL(cur2, select());

    QVERIFY( cur2.next() );

    if ( cur2.value( "t_nchar" ).toString() != val1 )

        qDebug( QString( "Wrong value for t_nchar: expected '%1', got '%2'" ).arg( val1 ).arg(

                cur2.value( "t_nchar" ).toString() ) );

    irec = cur2.primeInsert();

    irec->setValue( "id", 2 );

    irec->setValue( "t_nchar", utf8str );

    QVERIFY( cur2.insert() );

    QVERIFY_SQL(cur2, select( "id=2" ));

    QVERIFY( cur2.next() );