Source code
Revision control
Copy as Markdown
Other Tools
use arraystring::{error::Error, prelude::*, utils::is_char_boundary, utils::is_inside_boundary};
use std::panic::{catch_unwind, AssertUnwindSafe, RefUnwindSafe};
use std::{fmt::Debug, iter::FromIterator};
fn unwind<R, F>(func: F) -> Result<R, ()>
where
F: FnOnce() -> R,
{
catch_unwind(AssertUnwindSafe(func)).map_err(|_| ())
}
static STRINGS: [&'static str; 8] = [
"🤔🤔🤔🤔🤔🤔🤔",
"ABCDEFGHIJKLMNOPQRSASHUDAHSDIUASH ",
"iejueueheuheuheu 0",
"",
"1",
"ab",
" ",
" 899saH(8hadhaiuhsidnkandu",
];
#[test]
fn try_from_str() {
assert(String::from, MaxString::try_from_str);
}
#[test]
fn from_str_truncate() {
assert(String::from, MaxString::from_str_truncate);
}
#[test]
fn from_str_unchecked() {
assert(String::from, |s| unsafe {
MaxString::from_str_unchecked(s)
});
}
#[test]
fn try_from_chars() {
assert(
|s| String::from_iter(s.chars()),
|s| MaxString::try_from_chars(s.chars()),
);
}
#[test]
fn from_chars() {
assert(
|s| String::from_iter(s.chars()),
|s| MaxString::from_chars(s.chars()),
);
}
#[test]
fn from_chars_unchecked() {
assert(
|s| String::from_iter(s.chars()),
|s| unsafe { MaxString::from_chars_unchecked(s.chars()) },
);
}
#[test]
fn try_from_iter() {
assert(
|s| String::from_iter(vec![s]),
|s| MaxString::try_from_iterator(vec![s]),
);
}
#[test]
fn from_iter() {
assert(
|s| String::from_iter(vec![s]),
|s| MaxString::from_iterator(vec![s]),
);
}
#[test]
fn from_iter_unchecked() {
assert(
|s| String::from_iter(vec![s]),
|s| unsafe { MaxString::from_iterator_unchecked(vec![s]) },
);
}
#[test]
fn try_from_utf8() {
assert(
|s| String::from_utf8(s.as_bytes().to_vec()),
|s| MaxString::try_from_utf8(s.as_bytes()),
);
}
#[test]
fn from_utf8() {
assert(
|s| String::from_utf8(s.as_bytes().to_vec()),
|s| MaxString::from_utf8(s.as_bytes()),
);
}
#[inline]
fn invalidate_utf8(buf: &mut [u8]) -> &mut [u8] {
if buf.len() >= 4 {
buf[0] = 0;
buf[1] = 159;
buf[2] = 146;
buf[3] = 150;
}
buf
}
#[test]
fn try_from_utf8_invalid() {
unsafe {
assert(
|s| String::from_utf8(invalidate_utf8(s.to_owned().as_bytes_mut()).to_vec()),
|s| MaxString::try_from_utf8(invalidate_utf8(s.to_owned().as_bytes_mut())),
);
}
}
#[test]
fn from_utf8_invalid() {
unsafe {
assert(
|s| String::from_utf8(invalidate_utf8(s.to_owned().as_bytes_mut()).to_vec()),
|s| MaxString::from_utf8(invalidate_utf8(s.to_owned().as_bytes_mut())),
);
}
}
#[test]
fn try_from_utf16() {
let utf16 = |s: &str| s.encode_utf16().collect::<Vec<_>>();
assert(
|s| String::from_utf16(&utf16(s)),
|s| MaxString::try_from_utf16(&utf16(s)),
);
}
#[test]
fn from_utf16() {
let utf16 = |s: &str| s.encode_utf16().collect::<Vec<_>>();
assert(
|s| String::from_utf16(&utf16(s)),
|s| MaxString::from_utf16(&utf16(s)),
);
}
#[test]
fn from_utf16_lossy() {
let utf16 = |s: &str| s.encode_utf16().collect::<Vec<_>>();
assert(
|s| String::from_utf16(&utf16(s)),
|s| MaxString::from_utf16(&utf16(s)),
);
}
fn invalidate_utf16(buf: &mut [u16]) -> &mut [u16] {
if buf.len() >= 7 {
buf[0] = 0xD834;
buf[1] = 0xDD1E;
buf[2] = 0x006D;
buf[3] = 0x0075;
buf[4] = 0xD800;
buf[5] = 0x0069;
buf[6] = 0x0063;
}
buf
}
#[test]
fn try_from_utf16_invalid() {
let utf16 = |s: &str| s.encode_utf16().collect::<Vec<_>>();
assert(
|s| String::from_utf16(invalidate_utf16(&mut utf16(s))),
|s| MaxString::try_from_utf16(invalidate_utf16(&mut utf16(s))),
);
}
#[test]
fn from_utf16_invalid() {
let utf16 = |s: &str| s.encode_utf16().collect::<Vec<_>>();
assert(
|s| String::from_utf16(invalidate_utf16(&mut utf16(s))),
|s| MaxString::from_utf16(invalidate_utf16(&mut utf16(s))),
);
}
#[test]
fn from_utf16_lossy_invalid() {
let utf16 = |s: &str| s.encode_utf16().collect::<Vec<_>>();
assert(
|s| String::from_utf16(invalidate_utf16(&mut utf16(s))),
|s| MaxString::from_utf16(invalidate_utf16(&mut utf16(s))),
);
}
#[test]
fn from_utf8_unchecked() {
unsafe {
assert(
|s| String::from_utf8_unchecked(s.as_bytes().to_vec()),
|s| MaxString::from_utf8_unchecked(s.as_bytes()),
);
}
}
#[test]
fn try_push_str() {
assert(
|s| {
let mut st = String::from(s);
st.push_str(s);
st
},
|s| {
let mut ms = MaxString::try_from_str(s).unwrap();
ms.try_push_str(s).map(|()| ms)
},
);
}
#[test]
fn push_str() {
assert(
|s| {
let mut st = String::from(s);
st.push_str(s);
st
},
|s| {
let mut ms = MaxString::try_from_str(s).unwrap();
ms.push_str(s);
ms
},
);
}
#[test]
fn add_str() {
assert(
|s| String::from(s) + s,
|s| MaxString::try_from_str(s).unwrap() + s,
);
}
#[test]
fn push_str_unchecked() {
assert(
|s| {
let mut st = String::from(s);
st.push_str(s);
st
},
|s| {
let mut ms = MaxString::try_from_str(s).unwrap();
unsafe { ms.push_str_unchecked(s) };
ms
},
);
}
#[test]
fn push() {
assert(
|s| {
let mut s = String::from(s);
s.push('🤔');
s
},
|s| {
let mut ms = MaxString::try_from_str(s).unwrap();
ms.try_push('🤔').map(|()| ms)
},
);
}
#[test]
fn push_unchecked() {
assert(
|s| {
let mut s = String::from(s);
s.push('🤔');
s
},
|s| {
let mut ms = MaxString::try_from_str(s).unwrap();
unsafe { ms.push_unchecked('🤔') };
ms
},
);
}
#[test]
fn truncate() {
assert(
|s| {
unwind(move || {
let mut s = String::from(s);
s.truncate(2);
s
})
},
|s| {
let mut ms = MaxString::try_from_str(s).unwrap();
ms.truncate(2).map(|()| ms)
},
);
}
#[test]
fn pop() {
assert(
|s| {
let mut s = String::from(s);
let old = s.pop();
(s, old)
},
|s| {
let mut ms = MaxString::try_from_str(s).unwrap();
let old = ms.pop();
(ms, old)
},
);
}
#[test]
fn trim() {
assert(
|s| String::from(s).trim().to_owned(),
|s| {
let mut ms = MaxString::try_from_str(s).unwrap();
ms.trim();
ms
},
);
}
#[test]
fn remove() {
assert(
|s| {
unwind(move || {
let mut s = String::from(s);
let removed = s.remove(2);
(removed, s)
})
},
|s| {
let mut ms = MaxString::try_from_str(s).unwrap();
ms.remove(2).map(|r| (r, ms))
},
);
}
#[test]
fn retain() {
assert(
|s| {
let mut s = String::from(s);
s.retain(|c| c == 'a');
s
},
|s| {
let mut ms = MaxString::try_from_str(s).unwrap();
ms.retain(|c| c == 'a');
ms
},
);
}
#[test]
fn try_insert() {
assert(
|s| {
unwind(move || {
let mut s = String::from(s);
s.insert(2, 'a');
s
})
},
|s| {
let mut ms = MaxString::try_from_str(s).unwrap();
ms.try_insert(2, 'a').map(|()| ms)
},
);
}
#[test]
fn insert_unchecked() {
assert(
|s| {
unwind(move || {
let mut s = String::from(s);
s.insert(2, 'a');
s
})
},
|s| {
let mut ms = MaxString::try_from_str(s).unwrap();
is_inside_boundary(2u8, ms.len())
.map_err(Error::from)
.and_then(|()| Ok(is_char_boundary(&ms, 2)?))
.map(|()| unsafe { ms.insert_unchecked(2, 'a') })
.map(|()| ms)
},
);
}
#[test]
fn try_insert_str() {
assert(
|s| {
unwind(move || {
let mut st = String::from(s);
st.insert_str(2, s);
st
})
},
|s| {
let mut ms = MaxString::try_from_str(s).unwrap();
ms.try_insert_str(2, s).map(|()| ms)
},
);
}
#[test]
fn insert_str() {
assert(
|s| {
unwind(move || {
let mut st = String::from(s);
st.insert_str(2, s);
(st, ())
})
},
|s| {
let mut ms = MaxString::try_from_str(s).unwrap();
let res = ms.insert_str(2, s);
res.map(|()| (ms, ()))
},
);
}
#[test]
fn insert_str_unchecked() {
assert(
|s| {
unwind(move || {
let mut st = String::from(s);
st.insert_str(2, s);
st
})
},
|s| {
let mut ms = MaxString::try_from_str(s).unwrap();
is_inside_boundary(2u8, ms.len())
.map_err(Error::from)
.and_then(|()| Ok(is_char_boundary(&ms, 2)?))
.map(|()| unsafe { ms.insert_str_unchecked(2, s) })
.map(|()| ms)
},
);
}
#[test]
fn clear() {
assert(
|s| {
let mut st = String::from(s);
st.clear();
st
},
|s| {
let mut ms = MaxString::try_from_str(s).unwrap();
ms.clear();
ms
},
);
}
#[test]
fn split_off() {
assert(
|s| {
unwind(move || {
let mut st = String::from(s);
let split = st.split_off(2);
(st, split)
})
},
|s| {
let mut ms = MaxString::try_from_str(s).unwrap();
ms.split_off(2).map(|s| (ms, s))
},
);
}
#[test]
fn drain() {
assert(
|s| {
unwind(move || {
let mut st = String::from(s);
let drained: String = st.drain(..2).collect();
(st, drained)
})
},
|s| {
let mut ms = MaxString::try_from_str(s).unwrap();
let drained = ms.drain(..2).map(|d| d.collect::<String>());
drained.map(|d| (ms, d))
},
);
}
#[test]
fn replace_range() {
assert(
|s| {
unwind(move || {
let mut st = String::from(s);
st.replace_range(..2, s);
(st, ())
})
},
|s| {
let mut ms = MaxString::try_from_str(s).unwrap();
ms.replace_range(..2, s).map(|()| (ms, ()))
},
);
}
#[test]
fn len() {
assert(
|s| {
let st = String::from(s);
st.len().to_string()
},
|s| {
let ms = MaxString::try_from_str(s).unwrap();
ms.len().to_string()
},
);
}
#[test]
fn is_empty() {
assert(
|s| {
let st = String::from(s);
st.is_empty().to_string()
},
|s| {
let ms = MaxString::try_from_str(s).unwrap();
ms.is_empty().to_string()
},
);
}
#[test]
fn new() {
assert_eq!(String::new().as_str(), MaxString::new().as_str());
}
// Internal hackery to make the function `assert` possible
trait Normalize<EQ: PartialEq> {
fn normalize(&self) -> EQ;
}
impl Normalize<Result<String, ()>> for () {
fn normalize(&self) -> Result<String, ()> {
Ok("".to_owned())
}
}
impl Normalize<Result<String, ()>> for MaxString {
fn normalize(&self) -> Result<String, ()> {
Ok(self.as_str().to_owned())
}
}
impl Normalize<Result<String, ()>> for String {
fn normalize(&self) -> Result<String, ()> {
Ok(self.as_str().to_owned())
}
}
impl<'a> Normalize<Result<String, ()>> for &'a str {
fn normalize(&self) -> Result<String, ()> {
Ok(self.to_string())
}
}
impl Normalize<Result<String, ()>> for char {
fn normalize(&self) -> Result<String, ()> {
Ok(self.to_string())
}
}
impl<N: Normalize<Result<String, ()>>> Normalize<Result<String, ()>> for Option<N> {
fn normalize(&self) -> Result<String, ()> {
self.as_ref().ok_or(()).and_then(|n| n.normalize())
}
}
impl<E, K: PartialEq, N: Normalize<Result<K, ()>>> Normalize<Result<K, ()>> for Result<N, E> {
fn normalize(&self) -> Result<K, ()> {
self.as_ref().map_err(|_| ()).and_then(|n| n.normalize())
}
}
impl<K: PartialEq, L: PartialEq, M: Normalize<K>, N: Normalize<L>> Normalize<Result<(K, L), ()>>
for (M, N)
{
fn normalize(&self) -> Result<(K, L), ()> {
Ok((self.0.normalize(), self.1.normalize()))
}
}
impl<J, K, L, M, N, O> Normalize<Result<(J, K, L), ()>> for (M, N, O)
where
J: PartialEq,
K: PartialEq,
L: PartialEq,
M: Normalize<J>,
N: Normalize<K>,
O: Normalize<L>,
{
fn normalize(&self) -> Result<(J, K, L), ()> {
Ok((self.0.normalize(), self.1.normalize(), self.2.normalize()))
}
}
fn assert<Q, F, G, T, U>(f: F, g: G)
where
Q: PartialEq + Debug,
T: Normalize<Q>,
U: Normalize<Q>,
F: Fn(&'static str) -> T + RefUnwindSafe,
G: Fn(&'static str) -> U + RefUnwindSafe,
{
let _ = env_logger::try_init();
for string in STRINGS.into_iter() {
let f = f(string).normalize();
let g = g(string).normalize();
assert_eq!(f, g);
}
}