This feature is WPF-specific and is only available in a WPF Test. (Window Capture is also available - it is used just like Browser Capture.)
As of version 2015 R1 you can find the custom steps in the Step Builder under General section.
Be sure to enter the applicable WindowCaption so Test Studio acts on the correct window. The WindowState options are: