1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91
use types::CBytes; pub trait Authenticator: Clone { fn get_auth_token(&self) -> CBytes; fn get_cassandra_name(&self) -> Option<&str>; } #[derive(Debug, Clone)] pub struct PasswordAuthenticator<'a> { username: &'a str, password: &'a str, } impl<'a> PasswordAuthenticator<'a> { pub fn new<'b>(username: &'b str, password: &'b str) -> PasswordAuthenticator<'b> { PasswordAuthenticator { username: username, password: password, } } } impl<'a> Authenticator for PasswordAuthenticator<'a> { fn get_auth_token(&self) -> CBytes { let mut token = vec![0]; token.extend_from_slice(self.username.as_bytes()); token.push(0); token.extend_from_slice(self.password.as_bytes()); CBytes::new(token) } fn get_cassandra_name(&self) -> Option<&str> { Some("org.apache.cassandra.auth.PasswordAuthenticator") } } #[derive(Debug, Clone)] pub struct NoneAuthenticator; impl Authenticator for NoneAuthenticator { fn get_auth_token(&self) -> CBytes { CBytes::new(vec![0]) } fn get_cassandra_name(&self) -> Option<&str> { None } } #[cfg(test)] mod tests { use super::*; #[test] fn test_password_authenticator_trait_impl() { let authenticator = PasswordAuthenticator::new("a", "a"); let _ = authenticator_tester(Box::new(authenticator)); } #[test] fn test_password_authenticator_new() { PasswordAuthenticator::new("foo", "bar"); } #[test] fn test_password_authenticator_get_cassandra_name() { let auth = PasswordAuthenticator::new("foo", "bar"); assert_eq!(auth.get_cassandra_name(), Some("org.apache.cassandra.auth.PasswordAuthenticator")); } #[test] fn test_password_authenticator_get_auth_token() { let auth = PasswordAuthenticator::new("foo", "bar"); let mut expected_token = vec![0]; expected_token.extend_from_slice("foo".as_bytes()); expected_token.push(0); expected_token.extend_from_slice("bar".as_bytes()); assert_eq!(auth.get_auth_token().into_plain().unwrap(), expected_token); } #[test] fn test_authenticator_none_get_cassandra_name() { let auth = NoneAuthenticator; assert_eq!(auth.get_cassandra_name(), None); assert_eq!(auth.get_auth_token().into_plain().unwrap(), vec![0]); } fn authenticator_tester<A: Authenticator>(_authenticator: Box<A>) {} }