Disable direct submission at the end of the test for cases where
direct submission is enabled in test body.
Adjust mock csr to restore os context in cases where
context has been replaced in test body.
Related-To: NEO-15017
Signed-off-by: Filip Hazubski <filip.hazubski@intel.com>