Fix tests

Signed-off-by: Yilin Chen <sticnarf@gmail.com>
This commit is contained in:
Yilin Chen 2019-08-12 14:08:49 +08:00
parent 52e9cb2c15
commit a74e1abd1a
No known key found for this signature in database
GPG Key ID: 353E7ED34BF326E0
3 changed files with 12 additions and 16 deletions

View File

@ -38,7 +38,7 @@ features = ["push", "process"]
[dev-dependencies] [dev-dependencies]
clap = "2.32" clap = "2.32"
tempdir = "0.3" tempdir = "0.3"
runtime = "0.3.0-alpha.3" runtime = { version = "0.3.0-alpha.6", default-features = false }
runtime-tokio = "0.3.0-alpha.3" runtime-tokio = "0.3.0-alpha.5"
proptest = "0.9" proptest = "0.9"
proptest-derive = "0.1.0" proptest-derive = "0.1.0"

View File

@ -18,7 +18,7 @@ async fn puts(client: &Client, pairs: impl IntoIterator<Item = impl Into<KvPair>
txn.commit().await.expect("Could not commit transaction"); txn.commit().await.expect("Could not commit transaction");
} }
async fn get(client: &Client, key: Key) -> Value { async fn get(client: &Client, key: Key) -> Option<Value> {
let txn = client.begin().await.expect("Could not begin a transaction"); let txn = client.begin().await.expect("Could not begin a transaction");
txn.get(key).await.expect("Could not get value") txn.get(key).await.expect("Could not get value")
} }

View File

@ -106,8 +106,7 @@ impl Transaction {
/// # let connected_client = connecting_client.await.unwrap(); /// # let connected_client = connecting_client.await.unwrap();
/// let mut txn = connected_client.begin().await.unwrap(); /// let mut txn = connected_client.begin().await.unwrap();
/// let key = "TiKV".to_owned(); /// let key = "TiKV".to_owned();
/// let req = txn.get(key); /// let result: Option<Value> = txn.get(key).await.unwrap();
/// let result: Value = req.await.unwrap();
/// // Finish the transaction... /// // Finish the transaction...
/// txn.commit().await.unwrap(); /// txn.commit().await.unwrap();
/// # }); /// # });
@ -125,15 +124,15 @@ impl Transaction {
/// ///
/// ```rust,no_run /// ```rust,no_run
/// # #![feature(async_await)] /// # #![feature(async_await)]
/// # use tikv_client::{KvPair, Config, transaction::Client}; /// # use tikv_client::{Key, Value, Config, transaction::Client};
/// # use futures::prelude::*; /// # use futures::prelude::*;
/// # use std::collections::HashMap;
/// # futures::executor::block_on(async { /// # futures::executor::block_on(async {
/// # let connecting_client = Client::connect(Config::new(vec!["192.168.0.100", "192.168.0.101"])); /// # let connecting_client = Client::connect(Config::new(vec!["192.168.0.100", "192.168.0.101"]));
/// # let connected_client = connecting_client.await.unwrap(); /// # let connected_client = connecting_client.await.unwrap();
/// let mut txn = connected_client.begin().await.unwrap(); /// let mut txn = connected_client.begin().await.unwrap();
/// let keys = vec!["TiKV".to_owned(), "TiDB".to_owned()]; /// let keys = vec!["TiKV".to_owned(), "TiDB".to_owned()];
/// let req = txn.batch_get(keys); /// let result: HashMap<Key, Value> = txn.batch_get(keys).await.unwrap();
/// let result: Result<HashMap<Key, Value>> = req.await.unwrap();
/// // Finish the transaction... /// // Finish the transaction...
/// txn.commit().await.unwrap(); /// txn.commit().await.unwrap();
/// # }); /// # });
@ -343,8 +342,7 @@ impl Snapshot {
/// # let connected_client = connecting_client.await.unwrap(); /// # let connected_client = connecting_client.await.unwrap();
/// let snapshot = connected_client.snapshot().await.unwrap(); /// let snapshot = connected_client.snapshot().await.unwrap();
/// let key = "TiKV".to_owned(); /// let key = "TiKV".to_owned();
/// let req = snapshot.get(key); /// let result: Option<Value> = snapshot.get(key).await.unwrap();
/// let result: Value = req.await.unwrap();
/// # }); /// # });
/// ``` /// ```
pub async fn get(&self, _key: impl Into<Key>) -> Result<Option<Value>> { pub async fn get(&self, _key: impl Into<Key>) -> Result<Option<Value>> {
@ -356,17 +354,15 @@ impl Snapshot {
/// ///
/// ```rust,no_run /// ```rust,no_run
/// # #![feature(async_await)] /// # #![feature(async_await)]
/// # use tikv_client::{KvPair, Config, TransactionClient}; /// # use tikv_client::{Key, Value, Config, TransactionClient};
/// # use futures::prelude::*; /// # use futures::prelude::*;
/// # use std::collections::HashMap;
/// # futures::executor::block_on(async { /// # futures::executor::block_on(async {
/// # let connecting_client = TransactionClient::connect(Config::new(vec!["192.168.0.100", "192.168.0.101"])); /// # let connecting_client = TransactionClient::connect(Config::new(vec!["192.168.0.100", "192.168.0.101"]));
/// # let connected_client = connecting_client.await.unwrap(); /// # let connected_client = connecting_client.await.unwrap();
/// let mut txn = connected_client.begin().await.unwrap(); /// let snapshot = connected_client.snapshot().await.unwrap();
/// let keys = vec!["TiKV".to_owned(), "TiDB".to_owned()]; /// let keys = vec!["TiKV".to_owned(), "TiDB".to_owned()];
/// let req = txn.batch_get(keys); /// let result: HashMap<Key, Value> = snapshot.batch_get(keys).await.unwrap();
/// let result: Result<HashMap<Key, Value>> = req.await.unwrap();
/// // Finish the transaction...
/// txn.commit().await.unwrap();
/// # }); /// # });
/// ``` /// ```
pub async fn batch_get( pub async fn batch_get(