This feature is WPF-specific and is only available in a WPF Test. Capture Window is also available - it is used just like Capture Browser.
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: