register_space: use sync::Mutex instead of std::sync::Mutex

Then we don't need to unwrap

BUG=chromium:831850
TEST=cargo test
CQ-DEPEND=CL:1506828
Change-Id: I4200ea6351d61df1974e5e4c8583e783b21ea0eb
Reviewed-on: https://chromium-review.googlesource.com/1509514
Commit-Ready: Zach Reizner <zachr@chromium.org>
Tested-by: Jingkui Wang <jkwang@google.com>
Reviewed-by: Jingkui Wang <jkwang@google.com>

1 files changed, 7 insertions, 6 deletions

diff --git a/devices/src/register_space/register.rs b/devices/src/register_space/register.rs
index 3b113f9..44beaf5 100644
--- a/devices/src/register_space/register.rs
+++ b/devices/src/register_space/register.rs
@@ -6,7 +6,8 @@ use std;
 use std::boxed::Box;
 use std::cmp::{max, min, Ord, Ordering, PartialEq, PartialOrd};
 use std::mem::size_of;
-use std::sync::{Arc, Mutex, MutexGuard};
+use std::sync::{Arc, MutexGuard};
+use sync::Mutex;
 
 use data_model::DataInit;
 
@@ -216,7 +217,7 @@ impl<T: RegisterValue> Register<T> {
     }
 
     fn lock(&self) -> MutexGuard<RegisterInner<T>> {
-        self.inner.lock().expect("fail to lock register")
+        self.inner.lock()
     }
 }
 
@@ -562,16 +563,16 @@ mod tests {
 
         let s2 = state.clone();
         r.set_write_cb(move |val: u8| {
-            *s2.lock().unwrap() = val as u8;
+            *s2.lock() = val as u8;
             val
         });
         let data: [u8; 4] = [0, 0, 0, 0xff];
         r.write_bar(0, &data);
-        assert_eq!(*state.lock().unwrap(), 0xf);
+        assert_eq!(*state.lock(), 0xf);
         r.set_value(0xab);
-        assert_eq!(*state.lock().unwrap(), 0xf);
+        assert_eq!(*state.lock(), 0xf);
         let data: [u8; 1] = [0xfc];
         r.write_bar(3, &data);
-        assert_eq!(*state.lock().unwrap(), 0xc);
+        assert_eq!(*state.lock(), 0xc);
     }
 }