# -*- coding: utf-8 -*-

import pytest
from .test_base_class import TestBaseClass
from aerospike import exception as e
from .index_helpers import ensure_dropped_index

import aerospike


class TestListIndex(object):
    @pytest.fixture(autouse=True)
    def setup(self, request, as_connection):
        keys = []
        for i in range(5):
            key = ("test", "demo", i)
            rec = {
                "name": "name%s" % (str(i)),
                "addr": "name%s" % (str(i)),
                "numeric_list": [1, 2, 3, 4],
                "string_list": ["a", "b", "c", "d"],
                "geojson_list": [
                    aerospike.GeoJSON({"type": "Point", "coordinates": [-122.096449, 37.421868]}),
                    aerospike.GeoJSON({"type": "Point", "coordinates": [-122.053321, 37.434212]}),
                ],
                "age": i,
                "no": i,
            }
            as_connection.put(key, rec)
            keys.append(key)

        def teardown():
            """
            Teardown method.
            """
            for key in keys:
                try:
                    as_connection.remove(key)
                except e.RecordNotFound:
                    pass

        request.addfinalizer(teardown)

    def test_pos_listindex_with_correct_parameters(self):
        """
        Invoke index_list_create() with correct arguments
        """
        policy = {}
        retobj = self.as_connection.index_list_create(
            "test", "demo", "string_list", aerospike.INDEX_STRING, "test_string_list_index", policy
        )

        self.as_connection.index_remove("test", "test_string_list_index", policy)
        ensure_dropped_index(self.as_connection, "test", "test_string_list_index")

        assert retobj == 0

    def test_pos_listindex_with_correct_parameters_numeric(self):
        """
        Invoke index_list_create() with correct arguments
        """
        policy = {}
        retobj = self.as_connection.index_list_create(
            "test", "demo", "numeric_list", aerospike.INDEX_NUMERIC, "test_numeric_list_index", policy
        )

        assert retobj == 0
        self.as_connection.index_remove("test", "test_numeric_list_index", policy)
        ensure_dropped_index(self.as_connection, "test", "test_numeric_list_index")

    def test_pos_listindex_with_correct_parameters_set_length_extra(self):
        # Invoke index_list_create() with correct arguments and set length
        # extra
        set_name = "a"
        for _ in range(100):
            set_name = set_name + "a"
        policy = {}
        try:
            self.as_connection.index_list_create(
                "test", set_name, "string_list", aerospike.INDEX_STRING, "test_string_list_index", policy
            )
            assert False
        except e.InvalidRequest as exception:
            assert exception.code == 4
        except Exception as exception:
            assert isinstance(exception, e.InvalidRequest)

    def test_pos_listindex_with_incorrect_bin(self):
        """
        Invoke createindex() with incorrect bin
        """
        policy = {}
        retobj = self.as_connection.index_list_create(
            "test", "demo", "string_list1", aerospike.INDEX_STRING, "test_string_list_index", policy
        )

        assert retobj == 0
        self.as_connection.index_remove("test", "test_string_list_index", policy)
        ensure_dropped_index(self.as_connection, "test", "test_string_list_index")

    def test_pos_create_same_listindex_multiple_times(self):
        """
        Invoke createindex() with multiple times on same bin
        """
        policy = {}
        retobj = self.as_connection.index_list_create(
            "test", "demo", "numeric_list", aerospike.INDEX_NUMERIC, "test_numeric_list_index", policy
        )
        if retobj == 0:
            try:
                self.as_connection.index_list_create(
                    "test", "demo", "numeric_list", aerospike.INDEX_NUMERIC, "test_numeric_list_index", policy
                )
            except e.IndexFoundError:
                assert self.server_version < [6, 1]
            self.as_connection.index_remove("test", "test_numeric_list_index", policy)
            ensure_dropped_index(self.as_connection, "test", "test_numeric_list_index")
        else:
            assert False

    def test_pos_create_same_listindex_multiple_times_different_bin(self):
        """
        Invoke createindex() with multiple times on different bin
        """
        policy = {}
        retobj = self.as_connection.index_list_create(
            "test", "demo", "string_list", aerospike.INDEX_STRING, "test_string_list_index", policy
        )
        if retobj == 0:
            with pytest.raises(e.IndexFoundError):
                retobj = self.as_connection.index_list_create(
                    "test", "demo", "numeric_list", aerospike.INDEX_NUMERIC, "test_string_list_index", policy
                )
                self.as_connection.index_remove("test", "test_string_list_index", policy)
                ensure_dropped_index(self.as_connection, "test", "test_string_list_index")

            self.as_connection.index_remove("test", "test_string_list_index", policy)
            ensure_dropped_index(self.as_connection, "test", "test_string_list_index")
        else:
            assert True is False

    def test_pos_create_different_listindex_multiple_times_same_bin(self):
        """
                    Invoke createindex() with multiple times on same bin with different
        name
        """
        policy = {}
        retobj = self.as_connection.index_list_create(
            "test", "demo", "string_list", aerospike.INDEX_STRING, "test_string_list_index", policy
        )
        if retobj == 0:
            try:
                retobj = self.as_connection.index_list_create(
                    "test", "demo", "string_list", aerospike.INDEX_STRING, "test_string_list_index1", policy
                )
            except e.IndexFoundError:
                assert self.server_version < [6, 1]

            self.as_connection.index_remove("test", "test_string_list_index", policy)
            ensure_dropped_index(self.as_connection, "test", "test_string_list_index")
        else:
            assert True is False

    def test_pos_createlistindex_with_policy(self):
        """
        Invoke createindex() with policy
        """
        policy = {"timeout": 180000}
        retobj = self.as_connection.index_list_create(
            "test", "demo", "num_list_pol", aerospike.INDEX_NUMERIC, "test_numeric_list_index_pol", policy
        )

        assert retobj == 0
        self.as_connection.index_remove("test", "test_numeric_list_index_pol", policy)
        ensure_dropped_index(self.as_connection, "test", "test_numeric_list_index_pol")

    def test_pos_createlistindex_with_policystring(self):
        """
        Invoke createindex() with policy
        """
        policy = {"timeout": 180000}
        retobj = self.as_connection.index_list_create(
            "test", "demo", "string_list", aerospike.INDEX_STRING, "test_string_list_index", policy
        )

        assert retobj == 0
        self.as_connection.index_remove("test", "test_string_list_index", policy)
        ensure_dropped_index(self.as_connection, "test", "test_string_list_index")

    """
    This test case causes a db crash and hence has been commented. Work pending
on the C-client side
    def test_createindex_with_long_index_name(self):
            Invoke createindex() with long index name
        policy = {}
        retobj = self.as_connection.index_list_create( 'test', 'demo',
'age',
'bin2_integer_indexsdadadfasdfasdfeartfqrgahfasdfheudsdfasdfawf312342q3453rf9qwfa\
sdcfasdcalskdcbacfq34915rwcfasdcascnabscbaskjdbcalsjkbcdasc', policy)

        assert retobj == 0L
        self.as_connection.index_remove(policy, 'test',
'bin2_integer_indexsdadadfasdfasdfeartfqrgahfasdfheudsdfasdfawf312342q3453rf9qwfasd\
cfasdcalskdcbacfq34915rwcfasdcascnabscbaskjdbcalsjkbcdasc');
    """

    def test_pos_create_liststringindex_unicode_positive(self):
        """
        Invoke create string index() with correct arguments
        """
        policy = {}
        retobj = self.as_connection.index_list_create(
            "test", "demo", "string_list", aerospike.INDEX_STRING, "uni_name_index", policy
        )

        assert retobj == 0
        self.as_connection.index_remove("test", "uni_name_index", policy)
        ensure_dropped_index(self.as_connection, "test", "uni_name_index")

    def test_pos_create_list_integer_index_unicode(self):
        """
        Invoke createindex() with correct arguments
        """
        policy = {}
        retobj = self.as_connection.index_list_create(
            "test", "demo", "numeric_list", aerospike.INDEX_NUMERIC, "uni_age_index", policy
        )

        assert retobj == 0
        self.as_connection.index_remove("test", "uni_age_index", policy)
        ensure_dropped_index(self.as_connection, "test", "uni_age_index")

    def test_pos_create_list_geojson_index(self):
        """
        Invoke createindex() with correct arguments
        """
        policy = {}
        retobj = self.as_connection.index_list_create(
            "test", "demo", "geojson_list", aerospike.INDEX_GEO2DSPHERE, "geo_index", policy
        )

        assert retobj == 0
        self.as_connection.index_remove("test", "geo_index", policy)
        ensure_dropped_index(self.as_connection, "test", "geo_index")

    # Negative tests
    def test_neg_listindex_with_namespace_is_none(self):
        """
        Invoke createindex() with namespace is None
        """
        policy = {}
        try:
            self.as_connection.index_list_create(
                None, "demo", "string_list", aerospike.INDEX_STRING, "test_string_list_index", policy
            )

        except e.ParamError as exception:
            assert exception.code == -2
            assert exception.msg == "Namespace should be a string"

    def test_neg_listindex_with_set_is_int(self):
        """
        Invoke createindex() with set is int
        """
        policy = {}
        try:
            self.as_connection.index_list_create(
                "test", 1, "string_list", aerospike.INDEX_STRING, "test_string_list_index", policy
            )
            assert False
        except e.ParamError as exception:
            assert exception.code == -2
            assert exception.msg == "Set should be string, unicode or None"
        except Exception as exception:
            assert isinstance(exception, e.ParamError)

    def test_neg_listindex_with_set_is_none(self):
        """
        Invoke createindex() with set is None
        """
        policy = {}
        try:
            self.as_connection.index_list_create(
                "test", None, "string_list", aerospike.INDEX_STRING, "test_string_list_index", policy
            )

        except e.ParamError as exception:
            assert exception.code == -2
            assert exception.msg == "Set should be a string"
        self.as_connection.index_remove("test", "test_string_list_index", policy)
        ensure_dropped_index(self.as_connection, "test", "test_string_list_index")

    def test_neg_listindex_with_bin_is_none(self):
        """
        Invoke createindex() with bin is None
        """
        policy = {}
        try:
            self.as_connection.index_list_create(
                "test", "demo", None, aerospike.INDEX_NUMERIC, "test_numeric_list_index", policy
            )

        except e.ParamError as exception:
            assert exception.code == -2
            assert exception.msg == "Bin should be a string"

    def test_neg_listindex_with_index_is_none(self):
        """
        Invoke createindex() with index is None
        """
        policy = {}
        try:
            self.as_connection.index_list_create("test", "demo", "string_list", aerospike.INDEX_STRING, None, policy)

        except e.ParamError as exception:
            assert exception.code == -2
            assert exception.msg == "Index name should be string or unicode"

    def test_neg_listindex_with_incorrect_namespace(self):
        """
        Invoke createindex() with incorrect namespace
        """
        policy = {}

        try:
            self.as_connection.index_list_create(
                "test1", "demo", "numeric_list", aerospike.INDEX_NUMERIC, "test_numeric_list_index", policy
            )

        except e.NamespaceNotFound as exception:
            assert exception.code == 20

    def test_neg_listindex_with_incorrect_set(self):
        """
        Invoke createindex() with incorrect set
        """
        policy = {}
        retobj = self.as_connection.index_list_create(
            "test", "demo1", "numeric_list", aerospike.INDEX_NUMERIC, "test_numeric_list_index", policy
        )

        assert retobj == 0
        self.as_connection.index_remove("test", "test_numeric_list_index", policy)
        ensure_dropped_index(self.as_connection, "test", "test_numeric_list_index")

    def test_neg_listindex_with_correct_parameters_no_connection(self):
        """
        Invoke index_list_create() with correct arguments no connection
        """
        policy = {}
        config = TestBaseClass.get_connection_config()
        client1 = aerospike.client(config)
        client1.close()

        try:
            client1.index_list_create(
                "test", "demo", "string_list", aerospike.INDEX_STRING, "test_string_list_index", policy
            )

        except e.ClusterError as exception:
            assert exception.code == 11

    def test_neg_listindex_with_no_paramters(self):
        """
        Invoke index_list_create() without any mandatory parameters.
        """
        with pytest.raises(TypeError) as typeError:
            self.as_connection.index_list_create()

        assert "argument 'ns' (pos 1)" in str(typeError.value)
