Update TestStepSpec

* Implementing tests.