Add a unit test showing augmentation handling