#!/usr/bin/env python -O
# -*- coding: utf-8 -*-

import datetime
import os
import unittest

from ..base_test import create_connection


class CursorMariaDBTest(unittest.TestCase):

    def setUp(self):
        self.connection = create_connection()

    def tearDown(self):
        self.connection.close()

    def test_insert_parameter(self):
        cursor = self.connection.cursor()
        cursor.execute("CREATE TEMPORARY TABLE test_insert_parameter("
                        "a int not null auto_increment primary key,"
                        "b int, c int, d varchar(20),e date)")
        list_in = []
        for i in range(1, 3001):
            row = (i, i, i, "bar", datetime.date(2019, 1, 1))
            list_in.append(row)
        cursor.executemany("INSERT INTO test_insert_parameter VALUES "
                            "(?,?,?,?,?)", list_in)
        self.assertEqual(len(list_in), cursor.rowcount)
        self.connection.commit()
        cursor.execute("SELECT * FROM test_insert_parameter order by a")
        list_out = cursor.fetchall()
        self.assertEqual(len(list_in), cursor.rowcount)
        self.assertEqual(list_in, list_out)
        cursor.close()

    def test_update_parameter(self):
        cursor = self.connection.cursor()
        cursor.execute("CREATE TEMPORARY TABLE test_update_parameter("
                        "a int not null auto_increment primary key,"
                        "b int, c int, d varchar(20),e date)")
        cursor.execute("set @@autocommit=0")
        list_in = []
        for i in range(1, 3001):
            row = (i, i, i, "bar", datetime.date(2019, 1, 1))
            list_in.append(row)
        cursor.executemany("INSERT INTO test_update_parameter VALUES "
                            "(?,?,?,?,?)", list_in)
        self.assertEqual(len(list_in), cursor.rowcount)
        self.connection.commit()
        cursor.close()
        list_update = []

        cursor = self.connection.cursor()
        cursor.execute("set @@autocommit=0")
        for i in range(1, 3001):
            row = (i + 1, i)
            list_update.append(row)

        cursor.executemany("UPDATE test_update_parameter SET b=? "
                            "WHERE a=?", list_update)
        self.assertEqual(cursor.rowcount, 3000)
        self.connection.commit()
        cursor.close()

    def test_delete_parameter(self):
        cursor = self.connection.cursor()
        cursor.execute("CREATE TEMPORARY TABLE test_delete_parameter("
                        "a int not null auto_increment primary key,"
                        "b int, c int, d varchar(20),e date)")
        cursor.execute("set @@autocommit=0")
        list_in = []
        for i in range(1, 3001):
            row = (i, i, i, "bar", datetime.date(2019, 1, 1))
            list_in.append(row)
        cursor.executemany("INSERT INTO test_delete_parameter VALUES "
                            "(?,?,?,?,?)", list_in)
        self.assertEqual(len(list_in), cursor.rowcount)
        self.connection.commit()
        cursor.close()
        list_delete = []

        cursor = self.connection.cursor()
        cursor.execute("set @@autocommit=0")
        for i in range(1, 3001):
            list_delete.append((i,))

        cursor.executemany("DELETE FROM test_delete_parameter WHERE "
                            "a=?", list_delete)
        self.assertEqual(cursor.rowcount, 3000)
        self.connection.commit()
        cursor.close()


if __name__ == '__main__':
    unittest.main()
