420 lines
11 KiB
Rust
420 lines
11 KiB
Rust
use rhai::{Engine, EvalAltResult, ImmutableString, Scope, INT};
|
|
|
|
#[test]
|
|
fn test_string() -> Result<(), Box<EvalAltResult>> {
|
|
let engine = Engine::new();
|
|
|
|
assert_eq!(
|
|
engine.eval::<String>(r#""Test string: \u2764""#)?,
|
|
"Test string: ❤"
|
|
);
|
|
assert_eq!(
|
|
engine.eval::<String>(r#""Test string: ""\u2764""""#)?,
|
|
r#"Test string: "❤""#
|
|
);
|
|
assert_eq!(
|
|
engine.eval::<String>("\"Test\rstring: \\u2764\"")?,
|
|
"Test\rstring: ❤"
|
|
);
|
|
assert_eq!(
|
|
engine.eval::<String>(" \"Test string: \\u2764\\\n hello, world!\"")?,
|
|
if cfg!(not(feature = "no_position")) {
|
|
"Test string: ❤ hello, world!"
|
|
} else {
|
|
"Test string: ❤ hello, world!"
|
|
}
|
|
);
|
|
assert_eq!(
|
|
engine.eval::<String>(" `Test string: \\u2764\nhello,\\nworld!`")?,
|
|
"Test string: \\u2764\nhello,\\nworld!"
|
|
);
|
|
assert_eq!(
|
|
engine.eval::<String>(r#" `Test string: \\u2764\n``hello``,\\n"world"!`"#)?,
|
|
r#"Test string: \\u2764\n`hello`,\\n"world"!"#
|
|
);
|
|
assert_eq!(
|
|
engine.eval::<String>(" `\nTest string: \\u2764\nhello,\\nworld!`")?,
|
|
"Test string: \\u2764\nhello,\\nworld!"
|
|
);
|
|
assert_eq!(
|
|
engine.eval::<String>(" `\r\nTest string: \\u2764\nhello,\\nworld!`")?,
|
|
"Test string: \\u2764\nhello,\\nworld!"
|
|
);
|
|
assert_eq!(
|
|
engine.eval::<String>(r#""Test string: \x58""#)?,
|
|
"Test string: X"
|
|
);
|
|
assert_eq!(engine.eval::<String>(r#""\"hello\"""#)?, r#""hello""#);
|
|
|
|
assert_eq!(engine.eval::<String>(r#""foo" + "bar""#)?, "foobar");
|
|
|
|
assert!(engine.eval::<bool>(r#"let y = "hello, world!"; "world" in y"#)?);
|
|
assert!(engine.eval::<bool>(r#"let y = "hello, world!"; 'w' in y"#)?);
|
|
assert!(!engine.eval::<bool>(r#"let y = "hello, world!"; "hey" in y"#)?);
|
|
|
|
assert_eq!(engine.eval::<String>(r#""foo" + 123"#)?, "foo123");
|
|
|
|
#[cfg(not(feature = "no_object"))]
|
|
assert_eq!(engine.eval::<String>("to_string(42)")?, "42");
|
|
|
|
#[cfg(not(feature = "no_index"))]
|
|
{
|
|
assert_eq!(engine.eval::<char>(r#"let y = "hello"; y[1]"#)?, 'e');
|
|
assert_eq!(engine.eval::<char>(r#"let y = "hello"; y[-1]"#)?, 'o');
|
|
assert_eq!(engine.eval::<char>(r#"let y = "hello"; y[-4]"#)?, 'e');
|
|
}
|
|
|
|
#[cfg(not(feature = "no_object"))]
|
|
assert_eq!(engine.eval::<INT>(r#"let y = "hello"; y.len"#)?, 5);
|
|
|
|
#[cfg(not(feature = "no_object"))]
|
|
assert_eq!(
|
|
engine.eval::<INT>(r#"let y = "hello"; y.clear(); y.len"#)?,
|
|
0
|
|
);
|
|
|
|
assert_eq!(engine.eval::<INT>(r#"let y = "hello"; len(y)"#)?, 5);
|
|
|
|
#[cfg(not(feature = "no_object"))]
|
|
#[cfg(not(feature = "no_index"))]
|
|
assert_eq!(engine.eval::<char>(r#"let y = "hello"; y[y.len-1]"#)?, 'o');
|
|
|
|
#[cfg(not(feature = "no_float"))]
|
|
assert_eq!(engine.eval::<String>(r#""foo" + 123.4556"#)?, "foo123.4556");
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn test_string_dynamic() -> Result<(), Box<EvalAltResult>> {
|
|
let engine = Engine::new();
|
|
let mut scope = Scope::new();
|
|
scope.push("x", "foo");
|
|
scope.push("y", "foo");
|
|
scope.push("z", "foo");
|
|
|
|
assert!(engine.eval_with_scope::<bool>(&mut scope, r#"x == "foo""#)?);
|
|
assert!(engine.eval_with_scope::<bool>(&mut scope, r#"y == "foo""#)?);
|
|
assert!(engine.eval_with_scope::<bool>(&mut scope, r#"z == "foo""#)?);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn test_string_mut() -> Result<(), Box<EvalAltResult>> {
|
|
let mut engine = Engine::new();
|
|
|
|
engine.register_fn("foo", |s: &str| s.len() as INT);
|
|
engine.register_fn("bar", |s: String| s.len() as INT);
|
|
engine.register_fn("baz", |s: &mut String| s.len());
|
|
|
|
assert_eq!(engine.eval::<char>(r#"pop("hello")"#)?, 'o');
|
|
assert_eq!(engine.eval::<String>(r#"pop("hello", 3)"#)?, "llo");
|
|
assert_eq!(engine.eval::<String>(r#"pop("hello", 10)"#)?, "hello");
|
|
assert_eq!(engine.eval::<String>(r#"pop("hello", -42)"#)?, "");
|
|
|
|
assert_eq!(engine.eval::<INT>(r#"foo("hello")"#)?, 5);
|
|
assert_eq!(engine.eval::<INT>(r#"bar("hello")"#)?, 5);
|
|
assert!(
|
|
matches!(*engine.eval::<INT>(r#"baz("hello")"#).expect_err("should error"),
|
|
EvalAltResult::ErrorFunctionNotFound(f, ..) if f == "baz (&str | ImmutableString | String)"
|
|
)
|
|
);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[cfg(not(feature = "no_object"))]
|
|
#[test]
|
|
fn test_string_substring() -> Result<(), Box<EvalAltResult>> {
|
|
let engine = Engine::new();
|
|
|
|
assert_eq!(
|
|
engine.eval::<String>(r#"let x = "hello! \u2764\u2764\u2764"; x.sub_string(-2, 2)"#)?,
|
|
"❤❤"
|
|
);
|
|
|
|
assert_eq!(
|
|
engine.eval::<String>(
|
|
r#"let x = "\u2764\u2764\u2764 hello! \u2764\u2764\u2764"; x.sub_string(1, 5)"#
|
|
)?,
|
|
"❤❤ he"
|
|
);
|
|
|
|
assert_eq!(
|
|
engine.eval::<String>(
|
|
r#"let x = "\u2764\u2764\u2764 hello! \u2764\u2764\u2764"; x.sub_string(1)"#
|
|
)?,
|
|
"❤❤ hello! ❤❤❤"
|
|
);
|
|
|
|
assert_eq!(
|
|
engine.eval::<String>(
|
|
r#"let x = "\u2764\u2764\u2764 hello! \u2764\u2764\u2764"; x.sub_string(99)"#
|
|
)?,
|
|
""
|
|
);
|
|
|
|
assert_eq!(
|
|
engine.eval::<String>(
|
|
r#"let x = "\u2764\u2764\u2764 hello! \u2764\u2764\u2764"; x.sub_string(1, -1)"#
|
|
)?,
|
|
""
|
|
);
|
|
|
|
assert_eq!(
|
|
engine.eval::<String>(
|
|
r#"let x = "\u2764\u2764\u2764 hello! \u2764\u2764\u2764"; x.sub_string(1, 999)"#
|
|
)?,
|
|
"❤❤ hello! ❤❤❤"
|
|
);
|
|
|
|
assert_eq!(
|
|
engine.eval::<String>(
|
|
r#"let x = "\u2764\u2764\u2764 hello! \u2764\u2764\u2764"; x.crop(1, -1); x"#
|
|
)?,
|
|
""
|
|
);
|
|
|
|
assert_eq!(
|
|
engine.eval::<String>(
|
|
r#"let x = "\u2764\u2764\u2764 hello! \u2764\u2764\u2764"; x.crop(4, 6); x"#
|
|
)?,
|
|
"hello!"
|
|
);
|
|
|
|
assert_eq!(
|
|
engine.eval::<String>(
|
|
r#"let x = "\u2764\u2764\u2764 hello! \u2764\u2764\u2764"; x.crop(1, 999); x"#
|
|
)?,
|
|
"❤❤ hello! ❤❤❤"
|
|
);
|
|
|
|
assert_eq!(
|
|
engine.eval::<String>(
|
|
r#"let x = "\u2764\u2764\u2764 hello! \u2764\u2764\u2764"; x -= 'l'; x"#
|
|
)?,
|
|
"❤❤❤ heo! ❤❤❤"
|
|
);
|
|
|
|
assert_eq!(
|
|
engine.eval::<String>(
|
|
r#"let x = "\u2764\u2764\u2764 hello! \u2764\u2764\u2764"; x -= "\u2764\u2764"; x"#
|
|
)?,
|
|
"❤ hello! ❤"
|
|
);
|
|
assert_eq!(
|
|
engine.eval::<INT>(
|
|
r#"let x = "\u2764\u2764\u2764 hello! \u2764\u2764\u2764"; x.index_of('\u2764')"#
|
|
)?,
|
|
0
|
|
);
|
|
|
|
assert_eq!(
|
|
engine.eval::<INT>(
|
|
r#"let x = "\u2764\u2764\u2764 hello! \u2764\u2764\u2764"; x.index_of('\u2764', 5)"#
|
|
)?,
|
|
11
|
|
);
|
|
|
|
assert_eq!(
|
|
engine.eval::<INT>(
|
|
r#"let x = "\u2764\u2764\u2764 hello! \u2764\u2764\u2764"; x.index_of('\u2764', -6)"#
|
|
)?,
|
|
11
|
|
);
|
|
|
|
assert_eq!(
|
|
engine.eval::<INT>(
|
|
r#"let x = "\u2764\u2764\u2764 hello! \u2764\u2764\u2764"; x.index_of('\u2764', 999)"#
|
|
)?,
|
|
-1
|
|
);
|
|
|
|
assert_eq!(
|
|
engine.eval::<INT>(
|
|
r#"let x = "\u2764\u2764\u2764 hello! \u2764\u2764\u2764"; x.index_of('x')"#
|
|
)?,
|
|
-1
|
|
);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[cfg(not(feature = "no_object"))]
|
|
#[test]
|
|
fn test_string_format() -> Result<(), Box<EvalAltResult>> {
|
|
#[derive(Debug, Clone)]
|
|
struct TestStruct {
|
|
field: i64,
|
|
}
|
|
|
|
let mut engine = Engine::new();
|
|
|
|
engine
|
|
.register_type_with_name::<TestStruct>("TestStruct")
|
|
.register_fn("new_ts", || TestStruct { field: 42 })
|
|
.register_fn("to_string", |ts: TestStruct| format!("TS={}", ts.field))
|
|
.register_fn("to_debug", |ts: TestStruct| {
|
|
format!("!!!TS={}!!!", ts.field)
|
|
});
|
|
|
|
assert_eq!(
|
|
engine.eval::<String>(r#"let x = new_ts(); "foo" + x"#)?,
|
|
"fooTS=42"
|
|
);
|
|
assert_eq!(
|
|
engine.eval::<String>(r#"let x = new_ts(); x + "foo""#)?,
|
|
"TS=42foo"
|
|
);
|
|
#[cfg(not(feature = "no_index"))]
|
|
assert_eq!(
|
|
engine.eval::<String>(r#"let x = [new_ts()]; "foo" + x"#)?,
|
|
"foo[!!!TS=42!!!]"
|
|
);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn test_string_fn() -> Result<(), Box<EvalAltResult>> {
|
|
let mut engine = Engine::new();
|
|
|
|
engine.register_fn("set_to_x", |ch: &mut char| *ch = 'X');
|
|
|
|
#[cfg(not(feature = "no_index"))]
|
|
#[cfg(not(feature = "no_object"))]
|
|
assert_eq!(
|
|
engine.eval::<String>(r#"let x="foo"; x[0].set_to_x(); x"#)?,
|
|
"Xoo"
|
|
);
|
|
#[cfg(not(feature = "no_index"))]
|
|
assert_eq!(
|
|
engine.eval::<String>(r#"let x="foo"; set_to_x(x[0]); x"#)?,
|
|
"foo"
|
|
);
|
|
|
|
engine
|
|
.register_fn("foo1", |s: &str| s.len() as INT)
|
|
.register_fn("foo2", |s: ImmutableString| s.len() as INT)
|
|
.register_fn("foo3", |s: String| s.len() as INT)
|
|
.register_fn("foo4", |s: &mut ImmutableString| s.len() as INT);
|
|
|
|
assert_eq!(engine.eval::<INT>(r#"foo1("hello")"#)?, 5);
|
|
assert_eq!(engine.eval::<INT>(r#"foo2("hello")"#)?, 5);
|
|
assert_eq!(engine.eval::<INT>(r#"foo3("hello")"#)?, 5);
|
|
assert_eq!(engine.eval::<INT>(r#"foo4("hello")"#)?, 5);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[cfg(not(feature = "no_object"))]
|
|
#[cfg(not(feature = "no_index"))]
|
|
#[test]
|
|
fn test_string_split() -> Result<(), Box<EvalAltResult>> {
|
|
let engine = Engine::new();
|
|
|
|
assert_eq!(
|
|
engine.eval::<INT>(
|
|
r#"let x = "\u2764\u2764\u2764 hello! \u2764\u2764\u2764"; x.split(' ').len"#
|
|
)?,
|
|
3
|
|
);
|
|
assert_eq!(
|
|
engine.eval::<INT>(
|
|
r#"let x = "\u2764\u2764\u2764 hello! \u2764\u2764\u2764"; x.split("hello").len"#
|
|
)?,
|
|
2
|
|
);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn test_string_interpolated() -> Result<(), Box<EvalAltResult>> {
|
|
// Make sure strings interpolation works even under raw
|
|
let engine = Engine::new_raw();
|
|
|
|
assert_eq!(engine.eval::<String>("`${}`")?, "");
|
|
|
|
assert_eq!(
|
|
engine.eval::<String>(
|
|
"
|
|
let x = 40;
|
|
`hello ${x+2} worlds!`
|
|
"
|
|
)?,
|
|
"hello 42 worlds!"
|
|
);
|
|
|
|
assert_eq!(
|
|
engine.eval::<String>(
|
|
r#"
|
|
let x = 40;
|
|
"hello ${x+2} worlds!"
|
|
"#
|
|
)?,
|
|
"hello ${x+2} worlds!"
|
|
);
|
|
|
|
assert_eq!(
|
|
engine.eval::<String>(
|
|
"
|
|
const x = 42;
|
|
`hello ${x} worlds!`
|
|
"
|
|
)?,
|
|
"hello 42 worlds!"
|
|
);
|
|
|
|
assert_eq!(engine.eval::<String>("`hello ${}world!`")?, "hello world!");
|
|
|
|
assert_eq!(
|
|
engine.eval::<String>(
|
|
"
|
|
const x = 42;
|
|
`${x} worlds!`
|
|
"
|
|
)?,
|
|
"42 worlds!"
|
|
);
|
|
|
|
assert_eq!(
|
|
engine.eval::<String>(
|
|
"
|
|
const x = 42;
|
|
`hello ${x}`
|
|
"
|
|
)?,
|
|
"hello 42"
|
|
);
|
|
|
|
assert_eq!(
|
|
engine.eval::<String>(
|
|
"
|
|
const x = 20;
|
|
`hello ${let y = x + 1; `${y * 2}`} worlds!`
|
|
"
|
|
)?,
|
|
"hello 42 worlds!"
|
|
);
|
|
|
|
assert_eq!(
|
|
engine.eval::<String>(
|
|
r#"
|
|
let x = 42;
|
|
let y = 123;
|
|
|
|
`
|
|
Undeniable logic:
|
|
1) Hello, ${let w = `${x} world`; if x > 1 { w += "s" } w}!
|
|
2) If ${y} > ${x} then it is ${y > x}!
|
|
`
|
|
"#
|
|
)?,
|
|
"Undeniable logic:\n1) Hello, 42 worlds!\n2) If 123 > 42 then it is true!\n",
|
|
);
|
|
|
|
Ok(())
|
|
}
|