<reponame>shivamMg/influxable
import arrow
import pytest
from datetime import datetime
from decimal import Decimal as D
from influxable import attributes, exceptions


class TestBaseAttribute:
    def test_clean_success(self):
        base_attr = attributes.BaseAttribute()
        base_attr._value = 5
        base_attr.clean(5)
        value = base_attr.get_internal_value()
        assert value == 5

    def test_clean_with_none_value_success(self):
        base_attr = attributes.BaseAttribute()
        base_attr.clean(None)
        value = base_attr.get_internal_value()
        assert value is None

    def test_clean_with_default_value_success(self):
        base_attr = attributes.BaseAttribute(default=10)
        base_attr.clean(None)
        value = base_attr.get_internal_value()
        assert value == 10

    def test_clone_success(self):
        base_attr = attributes.BaseAttribute()
        cloned_attr = base_attr.clone()
        assert base_attr != cloned_attr

    def test_clone_with_value_success(self):
        base_attr = attributes.BaseAttribute()
        base_attr._value = 5
        cloned_attr = base_attr.clone()
        value = base_attr.get_internal_value()
        cloned_value = cloned_attr.get_internal_value()
        assert base_attr != cloned_attr
        assert value == cloned_value

    def test_get_internal_value_success(self):
        base_attr = attributes.BaseAttribute()
        base_attr._value = 5
        assert base_attr._value == 5
        assert base_attr.value == 5
        assert base_attr.get_internal_value() == 5

    def test_get_prep_value_success(self):
        base_attr = attributes.BaseAttribute()
        base_attr._value = 5
        assert base_attr.get_prep_value() == base_attr.to_influx(5)

    def test_get_name_success(self):
        base_attr = attributes.BaseAttribute(name='my_field')
        assert base_attr.attribute_name == 'my_field'
        assert base_attr.name == 'my_field'

    def test_reset_success(self):
        base_attr = attributes.BaseAttribute()
        base_attr._value = 5
        base_attr.reset()
        assert base_attr.get_internal_value() is None

    def test_set_internal_value_success(self):
        base_attr = attributes.BaseAttribute()
        base_attr.set_internal_value(5)
        assert base_attr.get_internal_value() == 5
        base_attr.set_internal_value(True)
        assert base_attr.get_internal_value() is True
        base_attr.set_internal_value('hello')
        assert base_attr.get_internal_value() == 'hello'
        base_attr.set_internal_value(None)
        assert base_attr.get_internal_value() is None

    def test_validate_is_nullable_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            base_attr = attributes.BaseAttribute(is_nullable=False)
            base_attr.set_internal_value(None)

    def test_validate_is_nullable_success(self):
        base_attr = attributes.BaseAttribute(is_nullable=False)
        base_attr.set_internal_value(5)


class TestIntegerFieldAttribute:
    def test_to_python_success(self):
        attr = attributes.IntegerFieldAttribute()
        assert attr.to_python(5) == 5

    def test_to_influx_success(self):
        attr = attributes.IntegerFieldAttribute()
        assert attr.to_influx(5) == '5i'

    def test_validate_min_value_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            base_attr = attributes.IntegerFieldAttribute(min_value=5)
            base_attr.set_internal_value(4)

    def test_validate_min_value_success(self):
        base_attr = attributes.IntegerFieldAttribute(min_value=5)
        base_attr.set_internal_value(6)

    def test_validate_max_value_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            base_attr = attributes.IntegerFieldAttribute(max_value=5)
            base_attr.set_internal_value(6)

    def test_validate_max_value_success(self):
        base_attr = attributes.IntegerFieldAttribute(max_value=5)
        base_attr.set_internal_value(4)

    def test_validate_invalid_min_value_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            attributes.IntegerFieldAttribute(min_value='ok')

    def test_validate_invalid_max_value_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            attributes.IntegerFieldAttribute(max_value='ok')


class TestFloatFieldAttribute:
    def test_clean_success(self):
        attr = attributes.FloatFieldAttribute()
        attr.set_internal_value(5.2345)
        assert attr.get_internal_value() == D(5.2345)

    def test_clean_with_max_nb_decimals_success(self):
        attr = attributes.FloatFieldAttribute(max_nb_decimals=2)
        attr.clean(5.2345)
        assert attr.get_internal_value() == D(5.23).quantize(D('.01'))

    def test_to_python_success(self):
        attr = attributes.FloatFieldAttribute()
        assert attr.to_python(5.2504) == 5.2504

    def test_to_influx_success(self):
        attr = attributes.FloatFieldAttribute()
        assert attr.to_influx(5.2504) == '5.2504'

    def test_validate_invalid_type_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            attributes.FloatFieldAttribute(max_nb_decimals='ok')

    def test_validate_negative_max_nb_decimals_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            attributes.FloatFieldAttribute(max_nb_decimals=-5)


class TestStringFieldAttribute:
    def test_to_python_success(self):
        attr = attributes.StringFieldAttribute()
        assert attr.to_python(5) == '5'

    def test_to_influx_success(self):
        attr = attributes.StringFieldAttribute()
        assert attr.to_influx('5') == "\'5\'"

    def test_validate_invalid_choices_type_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            attributes.StringFieldAttribute(choices=5)

    def test_validate_invalid_choices_item_type_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            attributes.StringFieldAttribute(choices=[5, 2, 3])

    def test_validate_bad_choice_type_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            attr = attributes.StringFieldAttribute(choices=['first', 'last'])
            attr.set_internal_value('ok')

    def test_validate_invalid_max_length_type_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            attributes.StringFieldAttribute(max_length='test')

    def test_validate_invalid_negative_max_length_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            attributes.StringFieldAttribute(max_length=-7)

    def test_validate_bad_max_length_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            attr = attributes.StringFieldAttribute(max_length=5)
            attr.set_internal_value('test_value')


class TestBooleanFieldAttribute:
    def test_to_python_success(self):
        attr = attributes.BooleanFieldAttribute()
        assert attr.to_python(True) is True

    def test_to_influx_success(self):
        attr = attributes.BooleanFieldAttribute()
        assert attr.to_influx(True) == "true"


class TestTimestampFieldAttribute:
    def test_to_python_success(self):
        attr = attributes.TimestampFieldAttribute()
        attr.set_internal_value(1570209691)
        assert attr.to_python(1570209691) == D('1570209691000000000')

    def test_to_influx_success(self):
        attr = attributes.TimestampFieldAttribute()
        attr.set_internal_value(1570209691)
        assert attr.to_influx(1570209691000000000) == '1570209691000000000'

    def test_clean_success(self):
        attr = attributes.TimestampFieldAttribute()
        attr.set_internal_value(1570209691)
        assert attr.to_python(1570209691) == D('1570209691000000000')

    def test_clean_with_auto_now_success(self):
        attr = attributes.TimestampFieldAttribute(auto_now=True)
        attr.set_internal_value(None)
        assert attr.get_internal_value() is not None

    def test_clean_with_none_success(self):
        attr = attributes.TimestampFieldAttribute(auto_now=False)
        attr.set_internal_value(None)
        assert attr.get_internal_value() is None

    def test_convert_to_nanoseconds_success(self):
        attr = attributes.TimestampFieldAttribute()
        attr.set_internal_value(1570209691)
        assert attr.convert_to_nanoseconds(1570209691) == D('1570209691000000000')

    def test_convert_to_precision_success(self):
        attr = attributes.TimestampFieldAttribute()
        attr.set_internal_value(1570209691)
        assert attr.convert_to_precision(1570209691, 'ms') == D('1570209691000')

    def test_validate_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            attributes.TimestampFieldAttribute(precision='k')


class TestDateTimeFieldAttribute:
    def test_to_python_success(self):
        attr = attributes.DateTimeFieldAttribute()
        attr.set_internal_value('2019-10-01 10:11:05')
        dt = arrow.get('2019-10-01 10:11:05', attr.str_format).datetime
        assert attr.to_python('2019-10-01 10:11:05') == dt

    def test_to_python_with_datetime_success(self):
        attr = attributes.DateTimeFieldAttribute()
        dt = arrow.get('2019-10-01 10:11:05', attr.str_format).datetime
        attr.set_internal_value(dt)
        assert attr.to_python(dt) == dt

    def test_to_influx_success(self):
        attr = attributes.DateTimeFieldAttribute()
        attr.set_internal_value('2019-10-04 21:20:34')
        dt = attr.get_internal_value()
        assert attr.to_influx(dt) == '1570224034000000000'

    def test_get_internal_value_success(self):
        attr = attributes.DateTimeFieldAttribute()
        attr.set_internal_value('2019-10-01 10:11:05')
        assert attr.get_internal_value() == '2019-10-01 10:11:05'

    def test_clean_success(self):
        attr = attributes.DateTimeFieldAttribute()
        attr.set_internal_value('2019-10-04 21:20:34')
        dt = arrow.get('2019-10-04 21:20:34', attr.str_format).datetime
        assert attr.to_python('2019-10-04 21:20:34') == dt

    def test_clean_with_auto_now_success(self):
        attr = attributes.DateTimeFieldAttribute(auto_now=True)
        attr.set_internal_value(None)
        assert attr.get_internal_value() is not None
        assert isinstance(attr._value, datetime)

    def test_clean_with_none_success(self):
        attr = attributes.DateTimeFieldAttribute(auto_now=False)
        attr.set_internal_value(None)
        assert attr.get_internal_value() is None

import arrow
import pytest
from datetime import datetime
from decimal import Decimal as D
from influxable import attributes, exceptions


class TestBaseAttribute:
    def test_clean_success(self):
        base_attr = attributes.BaseAttribute()
        base_attr._value = 5
        base_attr.clean(5)
        value = base_attr.get_internal_value()
        assert value == 5

    def test_clean_with_none_value_success(self):
        base_attr = attributes.BaseAttribute()
        base_attr.clean(None)
        value = base_attr.get_internal_value()
        assert value is None

    def test_clean_with_default_value_success(self):
        base_attr = attributes.BaseAttribute(default=10)
        base_attr.clean(None)
        value = base_attr.get_internal_value()
        assert value == 10

    def test_clone_success(self):
        base_attr = attributes.BaseAttribute()
        cloned_attr = base_attr.clone()
        assert base_attr != cloned_attr

    def test_clone_with_value_success(self):
        base_attr = attributes.BaseAttribute()
        base_attr._value = 5
        cloned_attr = base_attr.clone()
        value = base_attr.get_internal_value()
        cloned_value = cloned_attr.get_internal_value()
        assert base_attr != cloned_attr
        assert value == cloned_value

    def test_get_internal_value_success(self):
        base_attr = attributes.BaseAttribute()
        base_attr._value = 5
        assert base_attr._value == 5
        assert base_attr.value == 5
        assert base_attr.get_internal_value() == 5

    def test_get_prep_value_success(self):
        base_attr = attributes.BaseAttribute()
        base_attr._value = 5
        assert base_attr.get_prep_value() == base_attr.to_influx(5)

    def test_get_name_success(self):
        base_attr = attributes.BaseAttribute(name='my_field')
        assert base_attr.attribute_name == 'my_field'
        assert base_attr.name == 'my_field'

    def test_reset_success(self):
        base_attr = attributes.BaseAttribute()
        base_attr._value = 5
        base_attr.reset()
        assert base_attr.get_internal_value() is None

    def test_set_internal_value_success(self):
        base_attr = attributes.BaseAttribute()
        base_attr.set_internal_value(5)
        assert base_attr.get_internal_value() == 5
        base_attr.set_internal_value(True)
        assert base_attr.get_internal_value() is True
        base_attr.set_internal_value('hello')
        assert base_attr.get_internal_value() == 'hello'
        base_attr.set_internal_value(None)
        assert base_attr.get_internal_value() is None

    def test_validate_is_nullable_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            base_attr = attributes.BaseAttribute(is_nullable=False)
            base_attr.set_internal_value(None)

    def test_validate_is_nullable_success(self):
        base_attr = attributes.BaseAttribute(is_nullable=False)
        base_attr.set_internal_value(5)


class TestIntegerFieldAttribute:
    def test_to_python_success(self):
        attr = attributes.IntegerFieldAttribute()
        assert attr.to_python(5) == 5

    def test_to_influx_success(self):
        attr = attributes.IntegerFieldAttribute()
        assert attr.to_influx(5) == '5i'

    def test_validate_min_value_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            base_attr = attributes.IntegerFieldAttribute(min_value=5)
            base_attr.set_internal_value(4)

    def test_validate_min_value_success(self):
        base_attr = attributes.IntegerFieldAttribute(min_value=5)
        base_attr.set_internal_value(6)

    def test_validate_max_value_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            base_attr = attributes.IntegerFieldAttribute(max_value=5)
            base_attr.set_internal_value(6)

    def test_validate_max_value_success(self):
        base_attr = attributes.IntegerFieldAttribute(max_value=5)
        base_attr.set_internal_value(4)

    def test_validate_invalid_min_value_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            attributes.IntegerFieldAttribute(min_value='ok')

    def test_validate_invalid_max_value_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            attributes.IntegerFieldAttribute(max_value='ok')


class TestFloatFieldAttribute:
    def test_clean_success(self):
        attr = attributes.FloatFieldAttribute()
        attr.set_internal_value(5.2345)
        assert attr.get_internal_value() == D(5.2345)

    def test_clean_with_max_nb_decimals_success(self):
        attr = attributes.FloatFieldAttribute(max_nb_decimals=2)
        attr.clean(5.2345)
        assert attr.get_internal_value() == D(5.23).quantize(D('.01'))

    def test_to_python_success(self):
        attr = attributes.FloatFieldAttribute()
        assert attr.to_python(5.2504) == 5.2504

    def test_to_influx_success(self):
        attr = attributes.FloatFieldAttribute()
        assert attr.to_influx(5.2504) == '5.2504'

    def test_validate_invalid_type_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            attributes.FloatFieldAttribute(max_nb_decimals='ok')

    def test_validate_negative_max_nb_decimals_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            attributes.FloatFieldAttribute(max_nb_decimals=-5)


class TestStringFieldAttribute:
    def test_to_python_success(self):
        attr = attributes.StringFieldAttribute()
        assert attr.to_python(5) == '5'

    def test_to_influx_success(self):
        attr = attributes.StringFieldAttribute()
        assert attr.to_influx('5') == "\'5\'"

    def test_validate_invalid_choices_type_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            attributes.StringFieldAttribute(choices=5)

    def test_validate_invalid_choices_item_type_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            attributes.StringFieldAttribute(choices=[5, 2, 3])

    def test_validate_bad_choice_type_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            attr = attributes.StringFieldAttribute(choices=['first', 'last'])
            attr.set_internal_value('ok')

    def test_validate_invalid_max_length_type_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            attributes.StringFieldAttribute(max_length='test')

    def test_validate_invalid_negative_max_length_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            attributes.StringFieldAttribute(max_length=-7)

    def test_validate_bad_max_length_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            attr = attributes.StringFieldAttribute(max_length=5)
            attr.set_internal_value('test_value')


class TestBooleanFieldAttribute:
    def test_to_python_success(self):
        attr = attributes.BooleanFieldAttribute()
        assert attr.to_python(True) is True

    def test_to_influx_success(self):
        attr = attributes.BooleanFieldAttribute()
        assert attr.to_influx(True) == "true"


class TestTimestampFieldAttribute:
    def test_to_python_success(self):
        attr = attributes.TimestampFieldAttribute()
        attr.set_internal_value(1570209691)
        assert attr.to_python(1570209691) == D('1570209691000000000')

    def test_to_influx_success(self):
        attr = attributes.TimestampFieldAttribute()
        attr.set_internal_value(1570209691)
        assert attr.to_influx(1570209691000000000) == '1570209691000000000'

    def test_clean_success(self):
        attr = attributes.TimestampFieldAttribute()
        attr.set_internal_value(1570209691)
        assert attr.to_python(1570209691) == D('1570209691000000000')

    def test_clean_with_auto_now_success(self):
        attr = attributes.TimestampFieldAttribute(auto_now=True)
        attr.set_internal_value(None)
        assert attr.get_internal_value() is not None

    def test_clean_with_none_success(self):
        attr = attributes.TimestampFieldAttribute(auto_now=False)
        attr.set_internal_value(None)
        assert attr.get_internal_value() is None

    def test_convert_to_nanoseconds_success(self):
        attr = attributes.TimestampFieldAttribute()
        attr.set_internal_value(1570209691)
        assert attr.convert_to_nanoseconds(1570209691) == D('1570209691000000000')

    def test_convert_to_precision_success(self):
        attr = attributes.TimestampFieldAttribute()
        attr.set_internal_value(1570209691)
        assert attr.convert_to_precision(1570209691, 'ms') == D('1570209691000')

    def test_validate_fail(self):
        with pytest.raises(exceptions.InfluxDBAttributeValueError):
            attributes.TimestampFieldAttribute(precision='k')


class TestDateTimeFieldAttribute:
    def test_to_python_success(self):
        attr = attributes.DateTimeFieldAttribute()
        attr.set_internal_value('2019-10-01 10:11:05')
        dt = arrow.get('2019-10-01 10:11:05', attr.str_format).datetime
        assert attr.to_python('2019-10-01 10:11:05') == dt

    def test_to_python_with_datetime_success(self):
        attr = attributes.DateTimeFieldAttribute()
        dt = arrow.get('2019-10-01 10:11:05', attr.str_format).datetime
        attr.set_internal_value(dt)
        assert attr.to_python(dt) == dt

    def test_to_influx_success(self):
        attr = attributes.DateTimeFieldAttribute()
        attr.set_internal_value('2019-10-04 21:20:34')
        dt = attr.get_internal_value()
        assert attr.to_influx(dt) == '1570224034000000000'

    def test_get_internal_value_success(self):
        attr = attributes.DateTimeFieldAttribute()
        attr.set_internal_value('2019-10-01 10:11:05')
        assert attr.get_internal_value() == '2019-10-01 10:11:05'

    def test_clean_success(self):
        attr = attributes.DateTimeFieldAttribute()
        attr.set_internal_value('2019-10-04 21:20:34')
        dt = arrow.get('2019-10-04 21:20:34', attr.str_format).datetime
        assert attr.to_python('2019-10-04 21:20:34') == dt

    def test_clean_with_auto_now_success(self):
        attr = attributes.DateTimeFieldAttribute(auto_now=True)
        attr.set_internal_value(None)
        assert attr.get_internal_value() is not None
        assert isinstance(attr._value, datetime)

    def test_clean_with_none_success(self):
        attr = attributes.DateTimeFieldAttribute(auto_now=False)
        attr.set_internal_value(None)
        assert attr.get_internal_value() is None