asonix/cow-tools
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Add to_mut, use std Cow in test types

Browse source
This commit is contained in:
asonix 2023-03-01 20:12:36 -06:00
parent 879ccdfbfb
commit 2476994ea0
1 changed files with 30 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

44
src/lib.rs
View file

@ -26,6 +26,26 @@ where
Self::Owned(t) => std::borrow::Cow::Owned(t),
}
}
pub fn to_mut(&mut self) -> &mut <T as ToOwned>::Owned {
match *self {
Self::Borrowed(t) => {
*self = Self::Owned(t.to_owned());
match self {
Self::Owned(ref mut t) => t,
_ => unreachable!(),
}
}
Self::Ephemeral(t) => {
*self = Self::Owned(t.to_owned());
match self {
Self::Owned(ref mut t) => t,
_ => unreachable!(),
}
}
Self::Owned(ref mut t) => t,
}
}
}
impl<'a, 'b, T> AsRef<T> for TriCow<'a, 'b, T>
@ -95,19 +115,15 @@ mod tests {
fn with_cow<'b>(cow: TriCow<'a, 'b, [u8]>) -> Self;
}
struct A<'a>(TriCow<'a, 'static, [u8]>);
struct A<'a>(std::borrow::Cow<'a, [u8]>);
struct B<'a>(A<'a>);
struct C<'a>(B<'a>, TriCow<'a, 'static, [u8]>);
struct C<'a>(B<'a>, std::borrow::Cow<'a, [u8]>);
struct D<'a>(C<'a>);
struct E<'a>(D<'a>, TriCow<'a, 'static, [u8]>);
struct E<'a>(D<'a>, std::borrow::Cow<'a, [u8]>);
impl<'a> WithCow<'a> for A<'a> {
fn with_cow<'b>(cow: TriCow<'a, 'b, [u8]>) -> Self {
match cow {
TriCow::Borrowed(bytes) => Self(TriCow::Borrowed(bytes)),
TriCow::Ephemeral(bytes) => Self(TriCow::Owned(bytes.to_vec())),
TriCow::Owned(bytes) => Self(TriCow::Owned(bytes)),
}
Self(cow.into_std())
}
}
@ -124,21 +140,21 @@ mod tests {
let (first, second) = bytes.split_at(2);
Self(
B::with_cow(TriCow::Borrowed(first)),
TriCow::Borrowed(second),
std::borrow::Cow::Borrowed(second),
)
}
TriCow::Ephemeral(bytes) => {
let (first, second) = bytes.split_at(2);
Self(
B::with_cow(TriCow::Ephemeral(first)),
TriCow::Owned(Vec::from(second)),
std::borrow::Cow::Owned(Vec::from(second)),
)
}
TriCow::Owned(bytes) => {
let (first, second) = bytes.split_at(2);
Self(
B::with_cow(TriCow::Ephemeral(first)),
TriCow::Owned(Vec::from(second)),
std::borrow::Cow::Owned(Vec::from(second)),
)
}
}
@ -158,21 +174,21 @@ mod tests {
let (first, second) = bytes.split_at(2);
Self(
D::with_cow(TriCow::Borrowed(first)),
TriCow::Borrowed(second),
std::borrow::Cow::Borrowed(second),
)
}
TriCow::Ephemeral(bytes) => {
let (first, second) = bytes.split_at(2);
Self(
D::with_cow(TriCow::Ephemeral(first)),
TriCow::Owned(Vec::from(second)),
std::borrow::Cow::Owned(Vec::from(second)),
)
}
TriCow::Owned(bytes) => {
let (first, second) = bytes.split_at(2);
Self(
D::with_cow(TriCow::Ephemeral(first)),
TriCow::Owned(Vec::from(second)),
std::borrow::Cow::Owned(Vec::from(second)),
)
}
}