Add a few more tests to the buff component