# BSD 3-Clause License; see https://github.com/scikit-hep/awkward/blob/main/LICENSE

from __future__ import annotations

import numpy as np
import pytest

import awkward as ak


def test_array():
    record = ak.contents.RecordArray([ak.contents.NumpyArray(np.arange(10))], ["x"])
    array = ak.Array(record)

    with pytest.raises(AttributeError):
        array.x = 10

    with pytest.raises(AttributeError):
        array.not_an_existing_attribute = 10

    array._not_an_existing_attribute = 10
    assert array._not_an_existing_attribute == 10


def test_record():
    record = ak.contents.RecordArray([ak.contents.NumpyArray(np.arange(10))], ["x"])
    array = ak.Array(record)
    record = array[0]

    with pytest.raises(AttributeError):
        record.x = 10

    with pytest.raises(AttributeError):
        record.not_an_existing_attribute = 10

    record._not_an_existing_attribute = 10
    assert record._not_an_existing_attribute == 10


def test_bad_behavior_array():
    class BadBehaviorBase:
        FIELD_STRING = "I am not a list of fields!"

        @property
        def fields(self):
            return self.__class__.FIELD_STRING

        @fields.setter
        def fields(self, value):
            self.__class__.FIELD_STRING = value

    class BadBehaviorArray(BadBehaviorBase, ak.Array):
        pass

    class BadBehaviorRecord(BadBehaviorBase, ak.Record):
        pass

    behavior = {("*", "bad"): BadBehaviorArray, "bad": BadBehaviorRecord}

    record = ak.contents.RecordArray([ak.contents.NumpyArray(np.arange(10))], ["x"])
    array = ak.Array(record, with_name="bad", behavior=behavior)
    assert isinstance(array, BadBehaviorArray)

    assert array.fields == BadBehaviorArray.FIELD_STRING
    array.fields = "yo ho ho and a bottle of rum"
    assert array.fields == "yo ho ho and a bottle of rum"


def test_bad_behavior_record():
    class BadBehaviorBase:
        FIELD_STRING = "I am not a list of fields!"

        @property
        def fields(self):
            return self.__class__.FIELD_STRING

        @fields.setter
        def fields(self, value):
            self.__class__.FIELD_STRING = value

    class BadBehaviorArray(BadBehaviorBase, ak.Array):
        pass

    class BadBehaviorRecord(BadBehaviorBase, ak.Record):
        pass

    behavior = {("*", "bad"): BadBehaviorArray, "bad": BadBehaviorRecord}
    record = ak.contents.RecordArray([ak.contents.NumpyArray(np.arange(10))], ["x"])
    array = ak.Array(record, with_name="bad", behavior=behavior)
    record = array[0]
    assert isinstance(record, BadBehaviorRecord)

    assert record.fields == BadBehaviorRecord.FIELD_STRING
    record.fields = "yo ho ho and a bottle of rum"
    assert record.fields == "yo ho ho and a bottle of rum"