use std::borrow::Cow;
use rlua::{Lua, RluaCompat, String, Table};
fn with_str<F>(s: &str, f: F)
where
F: FnOnce(String),
{
Lua::new().context(|lua| {
let string = lua.create_string(s).unwrap();
f(string);
});
}
#[test]
fn compare() {
with_str("teststring", |t| assert_eq!(t, "teststring")); with_str("teststring", |t| assert_eq!(t, b"teststring")); with_str("teststring", |t| assert_eq!(t, b"teststring".to_vec())); with_str("teststring", |t| assert_eq!(t, "teststring".to_string())); with_str("teststring", |t| assert_eq!(t, t)); with_str("teststring", |t| {
assert_eq!(t, Cow::from(b"teststring".as_ref()))
}); with_str("bla", |t| assert_eq!(t, Cow::from(b"bla".to_vec()))); }
#[test]
fn string_views() {
Lua::new().context(|lua| {
lua.load(
r#"
ok = "null bytes are valid utf-8, wh\0 knew?"
err = "but \255 isn't :("
empty = ""
"#,
)
.exec()
.unwrap();
let globals = lua.globals();
let ok: String = globals.get("ok").unwrap();
let err: String = globals.get("err").unwrap();
let empty: String = globals.get("empty").unwrap();
assert_eq!(
ok.to_str().unwrap(),
"null bytes are valid utf-8, wh\0 knew?"
);
assert_eq!(
ok.as_bytes(),
&b"null bytes are valid utf-8, wh\0 knew?"[..]
);
assert!(err.to_str().is_err());
assert_eq!(err.as_bytes(), &b"but \xff isn't :("[..]);
assert_eq!(empty.to_str().unwrap(), "");
assert_eq!(empty.as_bytes_with_nul(), &[0]);
assert_eq!(empty.as_bytes(), &[]);
});
}
#[test]
fn raw_string() {
Lua::new().context(|lua| {
let rs = lua.create_string(&[0, 1, 2, 3, 0, 1, 2, 3]).unwrap();
assert_eq!(rs.as_bytes(), &[0, 1, 2, 3, 0, 1, 2, 3]);
});
}
#[test]
fn string_eq_hash() {
use std::collections::HashSet;
Lua::new().context(|lua| {
lua.load(
r#"
strlist = {
"foo",
"bar",
"foo",
"bar",
"baz"
}
"#,
)
.exec()
.unwrap();
let t: Table = lua.globals().get("strlist").unwrap();
let stringset: HashSet<rlua::String> = t.sequence_values().map(|v| v.unwrap()).collect();
let sorted = lua
.create_table_from((1..).zip(stringset.into_iter()))
.unwrap();
lua.globals().set("filtered", sorted).unwrap();
lua.load(
r#"
table.sort(filtered)
result = table.concat(filtered, ",")
"#,
)
.exec()
.unwrap();
let result: std::string::String = lua.globals().get("result").unwrap();
assert_eq!(result, "bar,baz,foo");
});
}