improve(unit-test): Update and include full example