Added missed test case.
Split EnsuresContextAction into separate classes to simplify reuse.

Related Work Items: #2