fmt

Signed-off-by: Ana Hobden <[email protected]>

Author: Hoverbear

src/kv.rs

@@ -1,13 +1,13 @@
 // Copyright 2019 TiKV Project Authors. Licensed under Apache-2.0.
 
+#[cfg(feature = "proptest-derive")]
+use proptest::{arbitrary::any_with, collection::size_range};
+#[cfg(feature = "proptest-derive")]
+use proptest_derive::Arbitrary;
 use std::ops::{
     Bound, Deref, DerefMut, Range, RangeFrom, RangeFull, RangeInclusive, RangeTo, RangeToInclusive,
 };
 use std::{fmt, str, u8};
-#[cfg(feature = "proptest-derive")]
-use proptest::{collection::size_range, arbitrary::any_with};
-#[cfg(feature = "proptest-derive")]
-use proptest_derive::Arbitrary;
 
 #[cfg(feature = "proptest-derive")]
 const PROPTEST_KEY_MAX: usize = 1024 * 2; // 2 KB
@@ -67,8 +67,11 @@ impl<'a> fmt::Display for HexRepr<'a> {
 #[derive(Default, Clone, Eq, PartialEq, Ord, PartialOrd, Hash)]
 #[cfg_attr(feature = "proptest-derive", derive(Arbitrary))]
 pub struct Key(
-    #[cfg_attr(feature = "proptest-derive", proptest(strategy = "any_with::<Vec<u8>>((size_range(PROPTEST_KEY_MAX), ()))"))]
-    Vec<u8>
+    #[cfg_attr(
+        feature = "proptest-derive",
+        proptest(strategy = "any_with::<Vec<u8>>((size_range(PROPTEST_KEY_MAX), ()))")
+    )]
+    Vec<u8>,
 );
 
 impl Key {
@@ -185,8 +188,11 @@ impl fmt::Debug for Key {
 #[derive(Default, Clone, Eq, PartialEq, Hash)]
 #[cfg_attr(feature = "proptest-derive", derive(Arbitrary))]
 pub struct Value(
-    #[cfg_attr(feature = "proptest-derive", proptest(strategy = "any_with::<Vec<u8>>((size_range(PROPTEST_VALUE_MAX), ()))"))]
-    Vec<u8>
+    #[cfg_attr(
+        feature = "proptest-derive",
+        proptest(strategy = "any_with::<Vec<u8>>((size_range(PROPTEST_VALUE_MAX), ()))")
+    )]
+    Vec<u8>,
 );
 
 impl Value {

tests/integration/raw.rs

@@ -1,9 +1,13 @@
 // Copyright 2019 TiKV Project Authors. Licensed under Apache-2.0.
 
 use crate::{arb_batch, integration::pd_addr};
-use proptest::{arbitrary::{any, any_with}, collection::size_range, proptest};
 use futures::executor::block_on;
-use tikv_client::{KvPair, raw::Client, Config, Value};
+use proptest::{
+    arbitrary::{any, any_with},
+    collection::size_range,
+    proptest,
+};
+use tikv_client::{raw::Client, Config, KvPair, Value};
 
 proptest! {
     #[test]
@@ -15,12 +19,12 @@ proptest! {
         block_on(
             client.put(pair.key().clone(), pair.value().clone())
         ).unwrap();
-        
+
         let out_value = block_on(
             client.get(pair.key().clone())
         ).unwrap();
         assert_eq!(Some(Value::from(pair.value().clone())), out_value);
-        
+
         block_on(
             client.delete(pair.key().clone())
         ).unwrap();
@@ -38,12 +42,12 @@ proptest! {
         block_on(
             client.batch_put(kvs.clone())
         ).unwrap();
-        
+
         let out_value = block_on(
             client.batch_get(keys.clone())
         ).unwrap();
         assert_eq!(kvs, out_value);
-        
+
         block_on(
             client.batch_delete(keys.clone())
         ).unwrap();
@@ -74,9 +78,9 @@ proptest! {
         assert!(kvs.iter().all(|kv| {
             out_value.iter().find(|out| kv == *out).is_some()
         }));
-        
+
         block_on(
             client.batch_delete(keys.clone())
         ).unwrap();
     }
-}
+}

tests/test.rs

@@ -12,7 +12,10 @@ use proptest::strategy::Strategy;
 const PROPTEST_BATCH_SIZE_MAX: usize = 16;
 
 #[cfg(feature = "proptest")]
-pub fn arb_batch<T: core::fmt::Debug>(single_strategy: impl Strategy<Value = T>, max_batch_size: impl Into<Option<usize>>) -> impl Strategy<Value = Vec<T>> {
+pub fn arb_batch<T: core::fmt::Debug>(
+    single_strategy: impl Strategy<Value = T>,
+    max_batch_size: impl Into<Option<usize>>,
+) -> impl Strategy<Value = Vec<T>> {
     let max_batch_size = max_batch_size.into().unwrap_or(PROPTEST_BATCH_SIZE_MAX);
     proptest::collection::vec(single_strategy, 0..max_batch_size)
-}
+}