/*
 * Copyright (C) 2005 Mike McCormack for CodeWeavers
 *
 * A test program for MSI records
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
 */

#include <libmsi.h>
#include <string.h>
#include <fcntl.h>
#include <unistd.h>

#ifdef _WIN32
#include <windows.h>
#endif

#include "test.h"

static const char *msifile = "winetest-record.msi";

static bool create_temp_file (char *name)
{
    unsigned char buffer[26], i;
    int fd;

    strcpy (name, "msitext-XXXXXX.tmp");
    fd = g_mkstemp (name);
    g_return_val_if_fail (fd != -1, FALSE);

    for (i = 0; i < 26; i++)
        buffer[i] = i + 'a';
    write (fd, buffer, sizeof (buffer));
    close (fd);

    return TRUE;
}

static void test_msirecord (void)
{
    GInputStream *in, *in2;
    unsigned r, sz;
    int i;
    LibmsiRecord *h;
    char buf[10];
    const char str[] = "hello";
    char filename[100];

    /* check behaviour with an invalid record */
    r = libmsi_record_get_field_count (0);
    ok (r == 0, "field count for invalid record not -1\n");
    r = libmsi_record_is_null (0, 0);
    ok (r == true, "invalid handle not considered to be null...\n");
    r = libmsi_record_get_int (0, 0);
    ok (r == LIBMSI_NULL_INT, "got integer from invalid record\n");
    r = libmsi_record_set_int (0, 0, 0);
    ok (!r, "libmsi_record_set_int returned wrong error\n");
    r = libmsi_record_set_int (0, -1, 0);
    ok (!r, "libmsi_record_set_int returned wrong error\n");
    r = libmsi_record_clear (0);
    ok (!r, "libmsi_record_clear returned wrong error\n");


    /* check behaviour of a record with 0 elements */
    h = libmsi_record_new (0);
    ok (h!=0, "couldn't create record with zero elements\n");
    r = libmsi_record_get_field_count (h);
    ok (r == 0, "field count should be zero\n");
    r = libmsi_record_is_null (h, 0);
    ok (r, "new record wasn't null\n");
    r = libmsi_record_is_null (h, 1);
    ok (r, "out of range record wasn't null\n");
    r = libmsi_record_is_null (h, -1);
    ok (r, "out of range record wasn't null\n");
    check_record_string (h, 0, "");
    /* same record, but add an integer to it */
    r = libmsi_record_set_int (h, 0, 0);
    ok (r, "Failed to set integer at 0 to 0\n");
    r = libmsi_record_is_null (h, 0);
    ok (r == false, "new record is null after setting an integer\n");
    r = libmsi_record_set_int (h, 0, 1);
    ok (r, "Failed to set integer at 0 to 1\n");
    r = libmsi_record_set_int (h, 1, 1);
    ok (!r, "set integer at 1\n");
    r = libmsi_record_set_int (h, -1, 0);
    ok (!r, "set integer at -1\n");
    r = libmsi_record_is_null (h, 0);
    ok (r == false, "new record is null after setting an integer\n");
    r = libmsi_record_get_int (h, 0);
    ok (r, "failed to get integer\n");

    /* same record, but add a null or empty string to it */
    r = libmsi_record_set_string (h, 0, NULL);
    ok (r, "Failed to set null string at 0\n");
    r = libmsi_record_is_null (h, 0);
    ok (r == true, "null string not null field\n");
    check_record_string (h, 0, "");
    r = libmsi_record_set_string (h, 0, "");
    ok (r, "Failed to set empty string at 0\n");
    r = libmsi_record_is_null (h, 0);
    ok (r == true, "null string not null field\n");
    check_record_string (h, 0, "");

    /* same record, but add a string to it */
    r = libmsi_record_set_string (h, 0, str);
    ok (r, "Failed to set string at 0\n");
    r = libmsi_record_get_int (h, 0);
    ok (r == LIBMSI_NULL_INT, "should get invalid integer\n");
    check_record_string (h, 0, str);

    /* same record, check we can wipe all the data */
    r = libmsi_record_clear (h);
    ok (r, "Failed to clear record\n");
    r = libmsi_record_clear (h);
    ok (r, "Failed to clear record again\n");
    r = libmsi_record_is_null (h, 0);
    ok (r, "cleared record wasn't null\n");

    /* same record, try converting strings to integers */
    i = libmsi_record_set_string (h, 0, "42");
    ok (i, "Failed to set string at 0\n");
    i = libmsi_record_get_int (h, 0);
    ok (i == 42, "should get invalid integer\n");
    i = libmsi_record_set_string (h, 0, "-42");
    ok (i, "Failed to set string at 0\n");
    i = libmsi_record_get_int (h, 0);
    ok (i == -42, "should get invalid integer\n");
    i = libmsi_record_set_string (h, 0, " 42");
    ok (i, "Failed to set string at 0\n");
    i = libmsi_record_get_int (h, 0);
    ok (i == LIBMSI_NULL_INT, "should get invalid integer\n");
    i = libmsi_record_set_string (h, 0, "42 ");
    ok (i, "Failed to set string at 0\n");
    i = libmsi_record_get_int (h, 0);
    ok (i == LIBMSI_NULL_INT, "should get invalid integer\n");
    i = libmsi_record_set_string (h, 0, "42.0");
    ok (i, "Failed to set string at 0\n");
    i = libmsi_record_get_int (h, 0);
    ok (i == LIBMSI_NULL_INT, "should get invalid integer\n");
    i = libmsi_record_set_string (h, 0, "0x42");
    ok (i, "Failed to set string at 0\n");
    i = libmsi_record_get_int (h, 0);
    ok (i == LIBMSI_NULL_INT, "should get invalid integer\n");
    i = libmsi_record_set_string (h, 0, "1000000000000000");
    ok (i, "Failed to set string at 0\n");
    i = libmsi_record_get_int (h, 0);
    ok (i == -1530494976, "should get truncated integer\n");
    i = libmsi_record_set_string (h, 0, "2147483647");
    ok (i, "Failed to set string at 0\n");
    i = libmsi_record_get_int (h, 0);
    ok (i == 2147483647, "should get maxint\n");
    i = libmsi_record_set_string (h, 0, "-2147483647");
    ok (i, "Failed to set string at 0\n");
    i = libmsi_record_get_int (h, 0);
    ok (i == -2147483647, "should get -maxint-1\n");
    i = libmsi_record_set_string (h, 0, "4294967297");
    ok (i, "Failed to set string at 0\n");
    i = libmsi_record_get_int (h, 0);
    ok (i == 1, "should get one\n");
    i = libmsi_record_set_string (h, 0, "foo");
    ok (i, "Failed to set string at 0\n");
    i = libmsi_record_get_int (h, 0);
    ok (i == LIBMSI_NULL_INT, "should get zero\n");
    i = libmsi_record_set_string (h, 0, "");
    ok (i, "Failed to set string at 0\n");
    i = libmsi_record_get_int (h, 0);
    ok (i == LIBMSI_NULL_INT, "should get zero\n");
    i = libmsi_record_set_string (h, 0, "+1");
    ok (i, "Failed to set string at 0\n");
    i = libmsi_record_get_int (h, 0);
    ok (i == LIBMSI_NULL_INT, "should get zero\n");

    /* same record, try converting integers to strings */
    r = libmsi_record_set_int (h, 0, 32);
    ok (r, "Failed to set integer at 0 to 32\n");
    check_record_string (h, 0, "32");
    r = libmsi_record_set_int (h, 0, -32);
    ok (r, "Failed to set integer at 0 to 32\n");
    check_record_string (h, 0, "-32");

    /* same record, now try streams */
    r = libmsi_record_load_stream (h, 0, NULL);
    ok (!r, "set NULL stream\n");
    sz = sizeof buf;
    in = libmsi_record_get_stream (h, 0);
    ok (!in, "read non-stream type\n");

    /* same record, now close it */
    g_object_unref (h);

    /* now try streams in a new record - need to create a file to play with */
    g_assert (create_temp_file (filename));

    /* streams can't be inserted in field 0 for some reason */
    h = libmsi_record_new (2);
    ok (h, "couldn't create a two field record\n");
    r = libmsi_record_load_stream (h, 0, filename);
    ok (!r, "added stream to field 0\n");
    r = libmsi_record_load_stream (h, 1, filename);
    ok (r, "failed to add stream to record\n");
    unlink (filename); /* Windows 98 doesn't like this at all, so don't check return. */

    in = libmsi_record_get_stream (h, 1);
    ok (in, "failed to get stream\n");
    in2 = libmsi_record_get_stream (h, 1);
    ok (in2, "failed to get stream\n");
    sz = g_input_stream_read (in, buf, sizeof(buf), NULL, NULL);
    ok (sz == 10, "failed to read stream\n");
    ok (!strncmp (buf, "abcdefghij", 10), "read the wrong thing\n");
    sz = g_input_stream_read (in2, buf, sizeof(buf), NULL, NULL);
    ok (sz == 10, "failed to read stream\n");
    ok (!strncmp (buf, "abcdefghij", 10), "read the wrong thing\n");
    sz = g_input_stream_read (in, buf, sizeof(buf), NULL, NULL);
    ok (sz == 10, "failed to read stream\n");
    ok (!strncmp (buf, "klmnopqrst", 10), "read the wrong thing\n");
    sz = g_input_stream_read (in, buf, sizeof(buf), NULL, NULL);
    ok (sz == 6, "failed to read stream\n");
    ok (!strncmp (buf, "uvwxyz", 6), "read the wrong thing\n");
    g_object_unref (in);

    sz = g_input_stream_read (in2, buf, sizeof(buf), NULL, NULL);
    ok (sz == 10, "failed to read stream\n");
    ok (!strncmp (buf, "klmnopqrst", 10), "read the wrong thing\n");
    sz = g_seekable_tell (G_SEEKABLE (in2));
    ok (sz == 20, "failed to get current position\n");
    r = g_seekable_seek (G_SEEKABLE (in2), 0, G_SEEK_END, NULL, NULL);
    ok (r, "failed to seek\n");
    sz = g_seekable_tell (G_SEEKABLE (in2));
    ok (sz == 26, "failed to get current position\n");
    r = g_seekable_seek (G_SEEKABLE (in2), 0, G_SEEK_SET, NULL, NULL);
    ok (r, "failed to seek\n");
    sz = g_seekable_tell (G_SEEKABLE (in2));
    ok (sz == 0, "failed to get current position\n");
    sz = g_input_stream_read (in2, buf, sizeof(buf), NULL, NULL);
    ok (sz == 10, "failed to read stream\n");
    ok (!strncmp (buf, "abcdefghij", 10), "read the wrong thing\n");
    g_object_unref (in2);

    in = g_memory_input_stream_new_from_data ("12345", 5, NULL);
    r = libmsi_record_set_stream (h, 1, in, 5, NULL, NULL);
    ok (r, "failed to set stream to record\n");
    g_object_unref(in);
    in = libmsi_record_get_stream (h, 1);
    ok (in, "failed to get stream\n");
    sz = g_input_stream_read (in, buf, sizeof(buf), NULL, NULL);
    ok (sz == 5, "failed to read stream\n");
    ok (!strncmp (buf, "12345", 5), "read the wrong thing\n");
    g_object_unref(in);

    /* now close the stream record */
    g_object_unref (h);
    unlink (filename); /* Delete it for sure, when everything else is closed. */
}

static void test_MsiRecordGetString (void)
{
    LibmsiRecord *rec;
    char buf[100];
    unsigned sz;
    unsigned r;

    rec = libmsi_record_new (2);
    ok (rec != 0, "Expected a valid handle\n");
    check_record_string(rec, 1, "");
    g_object_unref (rec);

    rec = libmsi_record_new (1);
    ok (rec != 0, "Expected a valid handle\n");

    r = libmsi_record_set_int (rec, 1, 5);
    ok (r, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r);
    check_record_string (rec, 1, "5");

    r = libmsi_record_set_int (rec, 1, -5);
    ok (r, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r);
    check_record_string (rec, 1, "-5");

    g_object_unref (rec);
}

static void test_MsiRecordGetInteger (void)
{
    LibmsiRecord *rec;
    int val;
    unsigned r;

    rec = libmsi_record_new (1);
    ok (rec != 0, "Expected a valid handle\n");

    r = libmsi_record_set_string (rec, 1, "5");
    ok (r, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r);

    val = libmsi_record_get_int (rec, 1);
    ok (val == 5, "Expected 5, got %d\n", val);

    r = libmsi_record_set_string (rec, 1, "-5");
    ok (r, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r);

    val = libmsi_record_get_int (rec, 1);
    ok (val == -5, "Expected -5, got %d\n", val);

    r = libmsi_record_set_string (rec, 1, "5apple");
    ok (r, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r);

    val = libmsi_record_get_int (rec, 1);
    ok (val == LIBMSI_NULL_INT, "Expected LIBMSI_NULL_INT, got %d\n", val);

    g_object_unref (rec);
}

static void test_fieldzero (void)
{
    GError *error = NULL;
    LibmsiDatabase *hdb;
    LibmsiQuery *hview;
    LibmsiRecord *rec;
    char buf[100];
    const char *query;
    unsigned sz;
    unsigned r;

    rec = libmsi_record_new (1);
    ok (rec != 0, "Expected a valid handle\n");

    r = libmsi_record_get_int (rec, 0);
    ok (r == LIBMSI_NULL_INT, "Expected LIBMSI_NULL_INT, got %d\n", r);

    check_record_string (rec, 0, "");

    r = libmsi_record_is_null (rec, 0);
    ok (r == true, "Expected true, got %d\n", r);

    r = libmsi_record_get_int (rec, 1);
    ok (r == LIBMSI_NULL_INT, "Expected LIBMSI_NULL_INT, got %d\n", r);

    r = libmsi_record_set_int (rec, 1, 42);
    ok (r, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r);

    r = libmsi_record_get_int (rec, 0);
    ok (r == LIBMSI_NULL_INT, "Expected LIBMSI_NULL_INT, got %d\n", r);

    check_record_string (rec, 0, "");

    r = libmsi_record_is_null (rec, 0);
    ok (r == true, "Expected true, got %d\n", r);

    r = libmsi_record_get_int (rec, 1);
    ok (r == 42, "Expected 42, got %d\n", r);

    r = libmsi_record_set_string (rec, 1, "bologna");
    ok (r, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r);

    r = libmsi_record_get_int (rec, 0);
    ok (r == LIBMSI_NULL_INT, "Expected LIBMSI_NULL_INT, got %d\n", r);

    check_record_string (rec, 0, "");

    r = libmsi_record_is_null (rec, 0);
    ok (r == true, "Expected true, got %d\n", r);

    check_record_string (rec, 1, "bologna");

    g_object_unref (rec);

    hdb = libmsi_database_new(msifile, LIBMSI_DB_FLAGS_CREATE, NULL, NULL);
    ok (hdb, "libmsi_database_open failed\n");

    query = "CREATE TABLE `drone` ( "
           "`id` INT, `name` CHAR (32), `number` CHAR (32) "
           "PRIMARY KEY `id`)";
    hview = libmsi_query_new (hdb, query, NULL);
    ok (hview, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r);
    r = libmsi_query_execute (hview, 0, NULL);
    ok (r, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r);
    r = libmsi_query_close (hview, NULL);
    ok (r, "libmsi_query_close failed\n");
    g_object_unref (hview);

    query = "INSERT INTO `drone` ( `id`, `name`, `number` )"
           "VALUES ('1', 'Abe', '8675309')";
    hview = libmsi_query_new (hdb, query, NULL);
    ok (hview, "libmsi_query failed\n");
    r = libmsi_query_execute (hview, 0, NULL);
    ok (r, "libmsi_query_execute failed\n");
    r = libmsi_query_close (hview, NULL);
    ok (r, "libmsi_query_close failed\n");
    g_object_unref (hview);

    rec = libmsi_database_get_primary_keys (hdb, "drone", &error);
    ok (rec, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r);

    r = libmsi_record_get_int (rec, 0);
    ok (r == LIBMSI_NULL_INT, "Expected LIBMSI_NULL_INT, got %d\n", r);

    check_record_string (rec, 0, "drone");

    r = libmsi_record_is_null (rec, 0);
    ok (r == false, "Expected false, got %d\n", r);

    g_object_unref (rec);

    rec = libmsi_database_get_primary_keys (hdb, "nosuchtable", &error);
    g_error_matches(error, LIBMSI_RESULT_ERROR, LIBMSI_RESULT_INVALID_TABLE);

    query = "SELECT * FROM `drone` WHERE `id` = 1";
    hview = libmsi_query_new (hdb, query, NULL);
    ok (hview, "Expected LIBMSI_RESULT_SUCCESS\n");
    r = libmsi_query_execute (hview, 0, NULL);
    ok (r, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r);
    rec = libmsi_query_fetch (hview, NULL);
    ok (r, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r);

    r = libmsi_record_get_int (rec, 0);
    ok (r == LIBMSI_NULL_INT, "Expected NULL value, got %d\n", r);
    r = libmsi_record_is_null (rec, 0);
    ok (r == true, "Expected true, got %d\n", r);

    g_object_unref (hview);
    g_object_unref (rec);
    g_object_unref (hdb);
    unlink (msifile);
}

int main ()
{
#if !GLIB_CHECK_VERSION(2,35,1)
    g_type_init ();
#endif

    test_msirecord ();
    test_MsiRecordGetString ();
    test_MsiRecordGetInteger ();
    test_fieldzero ();
}