def test_context_extablished(self): mock_context = MagicMock(return_value=MagicMock()) mock_context.return_value.established = True mock_context.return_value.step = MagicMock(return_value='out_token') with patch.object(salt.utils.vmware.gssapi, 'InitContext', mock_context): mock_context.established = True mock_context.step = MagicMock(return_value=None) with self.assertRaises(excs.CommandExecutionError) as excinfo: salt.utils.vmware.get_gssapi_token('principal', 'host', 'domain') self.assertEqual(mock_context.step.called, 0) self.assertIn('Context established, but didn\'t receive token', excinfo.exception.strerror)