# Licensed under the Apache License: http://www.apache.org/licenses/LICENSE-2.0
# For details: https://github.com/gaogaotiantian/viztracer/blob/master/NOTICE.txt

from viztracer import VizTracer
from viztracer.vizobject import VizObject

from .base_tmpl import BaseTmpl


class Hello(VizObject):
    def __init__(self, tracer):
        super().__init__(tracer, "name", trigger_on_change=False)
        self.a = 1
        self.b = "lol"

    @VizObject.triggerlog
    def change_val(self):
        self.a += 1
        self.b += "a"

    @VizObject.triggerlog(when="both")
    def change_val2(self):
        self.a += 2
        self.b += "b"


class TestVizObject(BaseTmpl):
    def test_basic(self):
        tracer = VizTracer(verbose=0)
        tracer.start()
        a = VizObject(tracer, "my variable")
        a.hello = 1
        a.hello = 2
        tracer.stop()
        entries = tracer.parse()
        self.assertEqual(entries, 3)

    def test_include(self):
        tracer = VizTracer(verbose=0)
        tracer.start()
        a = VizObject(tracer, "my variable", include_attributes=["b", "c"])
        a.hello = 1
        a.b = 2
        a.c = 3
        a.lol = 4
        tracer.stop()
        entries = tracer.parse()
        self.assertEqual(entries, 3)

    def test_exclude(self):
        tracer = VizTracer(verbose=0)
        tracer.start()
        a = VizObject(tracer, "my variable", exclude_attributes=["b", "c"])
        a.hello = 1
        a.b = 2
        a.c = 3
        a.lol = 4
        tracer.stop()
        entries = tracer.parse()
        self.assertEqual(entries, 3)

    def test_trigger_on_change(self):
        tracer = VizTracer(verbose=0)
        tracer.stop()
        tracer.clear()
        tracer.start()
        a = VizObject(tracer, "my variable", trigger_on_change=False)
        a.hello = 1
        a.b = 2
        a.c = 3
        a.lol = 4
        a.log()
        tracer.stop()
        entries = tracer.parse()
        self.assertEqual(entries, 2)

    def test_config(self):
        tracer = VizTracer(verbose=0)
        tracer.start()
        a = VizObject(tracer, "my variable")
        a.config("trigger_on_change", False)
        a.hello = 1
        a.b = 2
        a.c = 3
        a.lol = 4
        a.log()
        tracer.stop()
        entries = tracer.parse()
        self.assertEqual(entries, 2)
        with self.assertRaises(ValueError):
            a.config("invalid", "value")

    def test_decorator(self):
        tracer = VizTracer(verbose=0)
        tracer.start()
        a = Hello(tracer)
        a.config("include_attributes", ["a", "b"])
        a.change_val()
        a.change_val2()
        b = Hello(tracer)
        b.config("include_attributes", ["a", "b"])
        b.change_val()
        b.change_val2()
        tracer.stop()
        entries = tracer.parse()
        self.assertEqual(entries, 10)
        with self.assertRaises(ValueError):
            @VizObject.triggerlog(when="invalid")
            def change_invalid():
                pass
            change_invalid()

    def test_buffer_wrap(self):
        tracer = VizTracer(tracer_entries=10, verbose=0)
        tracer.start()
        a = VizObject(tracer, "my variable")
        for i in range(15):
            a.hello = i
        tracer.stop()
        entries = tracer.parse()
        tracer.save()
        self.assertEqual(entries, 10)

    def test_notracer(self):
        a = VizObject(None, "my variable")
        a.hello = 1
        a.hello = 2
        a = Hello(None)
        a.config("include_attributes", ["a", "b"])
        a.change_val()
        a.change_val2()
        b = Hello(None)
        b.config("include_attributes", ["a", "b"])
        b.change_val()
        b.change_val2()