from decorator import Counter


def test_no_args():
    @Counter
    def say_hello():
        print("hello")

    say_hello()
    say_hello()
    say_hello()
    say_hello()

    assert say_hello.count == 4


def test_arg():
    @Counter
    def say_hello(name):
        print(f"hello {name}")

    say_hello("a")
    say_hello("b")
    say_hello("c")
    say_hello("d")

    assert say_hello.count == 4


def test_default_arg():
    @Counter
    def say_hello(name="default"):
        print(f"hello {name}")

    say_hello("a")
    say_hello()
    say_hello("c")
    say_hello()

    assert say_hello.count == 4