"""Test the functions."""

import unittest
import calibration_db as db
import sqlite3

TEST_HEIGHT = 1.0
TEST_VOLUME = 1000.0


class TestDatabaseMethods(unittest.TestCase):
    """Class for testing database methods."""

    def test_01_setuptable(self):
        """Test the setup_calibration_table function."""
        try:
            db.setup_calibration_table(drop_first=True)
            cursor = db.SQLITE_CONN.cursor()
            cursor.execute("SELECT * FROM cal_data")
            cursor.fetchone()
            self.assertTrue(True, msg="Table is accessible.")
        except sqlite3.OperationalError:
            self.assertTrue(False, msg="Table is not accessible.")

    def test_02_getemptytable(self):
        """Test that an empty table is returned if no entries."""
        self.assertListEqual(db.get_calibration_data(), [], msg="Calibration table is empty.")

    def test_03_insertcalibrationdata(self):
        """Test the insert_calibration_data function."""
        db.insert_calibration_data(TEST_HEIGHT, TEST_VOLUME)
        cursor = db.SQLITE_CONN.cursor()
        cursor.execute("SELECT * FROM cal_data")
        result = cursor.fetchone()
        self.assertEqual(result[1], TEST_HEIGHT, msg="Height fetched OK")
        self.assertEqual(result[2], TEST_VOLUME, msg="Volume fetched OK")

    def test_04_getcaldatasingle(self):
        """Test the get_calibration_data function for a single row."""
        inserted_row = db.insert_calibration_data(TEST_HEIGHT, TEST_VOLUME)
        fetched_row = db.get_calibration_data(single_entry=inserted_row)
        self.assertEqual(fetched_row[1], TEST_HEIGHT, msg="Height fetched OK")
        self.assertEqual(fetched_row[2], TEST_VOLUME, msg="Volume fetched OK")

    def test_05_get_caldataall(self):
        """Test the get_calibration_data function for all rows."""
        db.insert_calibration_data(TEST_HEIGHT, TEST_VOLUME)
        fetched_data = db.get_calibration_data()
        self.assertEqual(len(fetched_data), 3, msg="Should have 3 rows of data.")

    def test_06_deletecaldata(self):
        """Test the delete_calibration_data function."""
        cal_data = db.get_calibration_data()
        start_len = len(cal_data)
        db.delete_calibration_data(cal_data[-1][0])
        end_len = len(db.get_calibration_data())
        self.assertEqual(start_len - 1, end_len, msg="Length of data array is good after delete. {} = {}".format(start_len - 1, end_len))

    def test_07_checkscaledata(self):
        """Test the scaling function."""
        expected = [
            [0.5, 0.5],
            [1.5, 1.5],
            [2.5, 3.0],
            [3.5, 6.0],
            [4.5, 12.0],
            [5.5, 24.0],
            [6.5, 48.0],
            [7.5, 96.0],
            [8.5, 192.0],
            [9.5, 384.0],
            [10.5, 640.0],
        ]

        db.setup_calibration_table(drop_first=True)
        db.insert_calibration_data(1.0, 1.0)
        db.insert_calibration_data(2.0, 2.0)
        db.insert_calibration_data(3.0, 4.0)
        db.insert_calibration_data(4.0, 8.0)
        db.insert_calibration_data(5.0, 16.0)
        db.insert_calibration_data(6.0, 32.0)
        db.insert_calibration_data(7.0, 64.0)
        db.insert_calibration_data(8.0, 128.0)
        db.insert_calibration_data(9.0, 256.0)
        db.insert_calibration_data(10.0, 512.0)

        for i in range(0, len(expected)):
            self.assertEqual(db.get_volume_for_height(expected[i][0]), expected[i][1], msg="Volume equal. {} == {}".format(db.get_volume_for_height(expected[i][0]), expected[i][1]))

    @classmethod
    def tearDownClass(cls):
        """Tear down function."""
        db.setup_calibration_table(drop_first=True)


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