Migrated test towards unit-style.

This test was intended to check the internal behavior of a type and thus is more of a unit test. In this spirit, we've migrated more towards this style by stubbing an internal variable to observe the type's behavior.