Add missing tests for remaining behaviors