extern crate env_logger;
#[macro_use]
extern crate log;
extern crate bmemcached;
use std::sync::Arc;
use std::thread;
use bmemcached::errors::{Error, ErrorKind};
use bmemcached::{MemcachedClient, Status};
#[test]
fn multiple_threads() {
let _ = env_logger::try_init();
let mut threads = vec![];
let client = Arc::new(MemcachedClient::new(vec!["127.0.0.1:11211"], 5).unwrap());
for i in 0..4 {
let client = client.clone();
debug!("Starting thread {}", i);
threads.push(thread::spawn(move || {
debug!("Started {}", i);
let data = format!("data_n{}", i);
client.set(&data, &data, 100).unwrap();
let val: String = client.get(&data).unwrap();
client.delete(&data).unwrap();
debug!("Finished {}", i);
val
}));
}
for (i, thread) in threads.into_iter().enumerate() {
let result = thread.join();
assert_eq!(result.unwrap(), format!("data_n{}", i));
}
}
#[test]
fn get_set_delete() {
let _ = env_logger::try_init();
let client = MemcachedClient::new(vec!["127.0.0.1:11211"], 1).unwrap();
let key = "Hello Get, Set, Delete Client";
let value = "World";
client.set(key, value, 1000).unwrap();
let rv: String = client.get(key).unwrap();
assert_eq!(rv, value);
client.delete(key).unwrap();
}
#[test]
fn get_set_u8() {
let _ = env_logger::try_init();
let client = MemcachedClient::new(vec!["127.0.0.1:11211"], 1).unwrap();
let key = "Hello u8";
let value = 1 as u8;
client.set(key, value, 1000).unwrap();
let rv: u8 = client.get(key).unwrap();
assert_eq!(rv, value);
client.delete(key).unwrap();
}
#[test]
fn get_set_u16() {
let _ = env_logger::try_init();
let client = MemcachedClient::new(vec!["127.0.0.1:11211"], 1).unwrap();
let key = "Hello u16";
let value = 1 as u16;
client.set(key, value, 1000).unwrap();
let rv: u16 = client.get(key).unwrap();
assert_eq!(rv, value);
client.delete(key).unwrap();
}
#[test]
fn get_set_u32() {
let _ = env_logger::try_init();
let client = MemcachedClient::new(vec!["127.0.0.1:11211"], 1).unwrap();
let key = "Hello u32";
let value = 1 as u32;
client.set(key, value, 1000).unwrap();
let rv: u32 = client.get(key).unwrap();
assert_eq!(rv, value);
client.delete(key).unwrap();
}
#[test]
fn get_set_u64() {
let _ = env_logger::try_init();
let client = MemcachedClient::new(vec!["127.0.0.1:11211"], 1).unwrap();
let key = "Hello u64";
let value = 1 as u64;
client.set(key, value, 1000).unwrap();
let rv: u64 = client.get(key).unwrap();
assert_eq!(rv, value);
client.delete(key).unwrap();
}
#[test]
fn add() {
let _ = env_logger::try_init();
let client = MemcachedClient::new(vec!["127.0.0.1:11211"], 1).unwrap();
let key = "Hello Add Client";
let value = "World";
client.add(key, value, 1000).unwrap();
let rv: String = client.get(key).unwrap();
assert_eq!(rv, value);
match client.add(key, value, 1000) {
Err(Error(ErrorKind::Status(Status::KeyExists), _)) => (),
e => panic!("Wrong status returned {:?}", e),
}
client.delete(key).unwrap();
}
#[test]
fn replace() {
let _ = env_logger::try_init();
let client = MemcachedClient::new(vec!["127.0.0.1:11211"], 1).unwrap();
let key = "Hello Replace Client";
let value = "World";
client.add(key, value, 1000).unwrap();
let rv: String = client.get(key).unwrap();
assert_eq!(rv, value);
client.replace(key, "New value", 100).unwrap();
let rv: String = client.get(key).unwrap();
assert_eq!(rv, "New value");
client.delete(key).unwrap();
}
#[test]
fn increment() {
let _ = env_logger::try_init();
let client = MemcachedClient::new(vec!["127.0.0.1:11211"], 1).unwrap();
let key = "Hello Increment Client";
assert_eq!(client.increment(key, 1, 0, 1000).unwrap(), 0);
assert_eq!(client.increment(key, 1, 1, 1000).unwrap(), 1);
client.delete(key).unwrap();
}
#[test]
fn decrement() {
let _ = env_logger::try_init();
let client = MemcachedClient::new(vec!["127.0.0.1:11211"], 1).unwrap();
let key = "Hello Decrement Client";
assert_eq!(client.decrement(key, 1, 10, 1000).unwrap(), 10);
assert_eq!(client.decrement(key, 1, 1, 1000).unwrap(), 9);
client.delete(key).unwrap();
}