SecretStore: fixed test (#7878)

This commit is contained in:
Svyatoslav Nikolsky 2018-02-14 12:49:56 +03:00 committed by Marek Kotewicz
parent 5584739b83
commit ebb92947a3
2 changed files with 5 additions and 5 deletions

View File

@ -191,9 +191,9 @@ impl<T> SessionImpl<T> where T: SessionTransport {
self.data.lock().continue_with = Some(action);
}
/// Get continue action.
pub fn continue_action(&self) -> Option<ContinueAction> {
self.data.lock().continue_with.clone()
/// Take continue action.
pub fn take_continue_action(&self) -> Option<ContinueAction> {
self.data.lock().continue_with.take()
}
/// Wait for session completion.

View File

@ -471,7 +471,7 @@ impl ClusterCore {
if is_master_node && session.is_finished() {
data.sessions.negotiation_sessions.remove(&session.id());
match session.wait() {
Ok((version, master)) => match session.continue_action() {
Ok((version, master)) => match session.take_continue_action() {
Some(ContinueAction::Decrypt(session, is_shadow_decryption)) => {
let initialization_error = if data.self_key_pair.public() == &master {
session.initialize(version, is_shadow_decryption, false)
@ -498,7 +498,7 @@ impl ClusterCore {
},
None => (),
},
Err(error) => match session.continue_action() {
Err(error) => match session.take_continue_action() {
Some(ContinueAction::Decrypt(session, _)) => {
data.sessions.decryption_sessions.remove(&session.id());
session.on_session_error(&meta.self_node_id, error);