itsscb/heapless
1
0
Fork 0
mirror of https://github.com/rust-embedded/heapless.git synced 2025-10-02 06:50:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Implement equality with CStr

Browse Source
This commit is contained in:
Yota Toyama 2025-04-07 19:03:09 -07:00
parent 08f17940e7
commit 4f1bbe9b11
1 changed files with 39 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
src/c_string.rs
View File

@ -298,6 +298,30 @@ impl<const N: usize, const M: usize> PartialEq<CString<M>> for CString<N> {
}
}
impl<const N: usize> PartialEq<CStr> for CString<N> {
fn eq(&self, rhs: &CStr) -> bool {
self.as_c_str() == rhs
}
}
impl<const N: usize> PartialEq<&CStr> for CString<N> {
fn eq(&self, rhs: &&CStr) -> bool {
self.as_c_str() == *rhs
}
}
impl<const N: usize> PartialEq<CString<N>> for CStr {
fn eq(&self, rhs: &CString<N>) -> bool {
self == rhs.as_c_str()
}
}
impl<const N: usize> PartialEq<CString<N>> for &CStr {
fn eq(&self, rhs: &CString<N>) -> bool {
*self == rhs.as_c_str()
}
}
impl<const N: usize> Eq for CString<N> {}
/// An error to extend [`CString`] with bytes.
@ -474,4 +498,19 @@ mod tests {
!= CString::<4>::from_bytes_with_nul(b"abc\0").unwrap()
);
}
#[test]
fn equal_with_c_str() {
// Empty strings
assert!(CString::<1>::new() == c"");
assert!(c"" == CString::<1>::new());
// Single character
assert!(CString::<2>::from_bytes_with_nul(b"a\0").unwrap() == c"a");
assert!(c"a" == CString::<2>::from_bytes_with_nul(b"a\0").unwrap());
// Multiple characters
assert!(CString::<4>::from_bytes_with_nul(b"abc\0").unwrap() == c"abc");
assert!(c"abc" == CString::<4>::from_bytes_with_nul(b"abc\0").unwrap());
}
}