mirror of
https://github.com/clockworklabs/SpacetimeDB.git
synced 2026-05-14 11:48:28 -04:00
Zeke Foppa
117 lines
3.4 KiB
Python
117 lines
3.4 KiB
Python
from .. import Smoketest
|
|
import string
|
|
import functools
|
|
|
|
|
|
ints = "u8", "u16", "u32", "u64", "u128", "i8", "i16", "i32", "i64", "i128"
|
|
|
|
class IntTests:
|
|
make_func = lambda int_ty: lambda self: self.do_test_autoinc(int_ty)
|
|
for int_ty in ints:
|
|
locals()[f"test_autoinc_{int_ty}"] = make_func(int_ty)
|
|
del int_ty, make_func
|
|
|
|
|
|
|
|
autoinc1_template = string.Template("""
|
|
#[spacetimedb::table(name = person_$KEY_TY)]
|
|
pub struct Person_$KEY_TY {
|
|
#[auto_inc]
|
|
key_col: $KEY_TY,
|
|
name: String,
|
|
}
|
|
|
|
#[spacetimedb::reducer]
|
|
pub fn add_$KEY_TY(ctx: &ReducerContext, name: String, expected_value: $KEY_TY) {
|
|
let value = ctx.db.person_$KEY_TY().insert(Person_$KEY_TY { key_col: 0, name });
|
|
assert_eq!(value.key_col, expected_value);
|
|
}
|
|
|
|
#[spacetimedb::reducer]
|
|
pub fn say_hello_$KEY_TY(ctx: &ReducerContext) {
|
|
for person in ctx.db.person_$KEY_TY().iter() {
|
|
log::info!("Hello, {}:{}!", person.key_col, person.name);
|
|
}
|
|
log::info!("Hello, World!");
|
|
}
|
|
""")
|
|
|
|
|
|
|
|
class AutoincBasic(IntTests, Smoketest):
|
|
"This tests the auto_inc functionality"
|
|
|
|
MODULE_CODE = f"""
|
|
#![allow(non_camel_case_types)]
|
|
use spacetimedb::{{log, ReducerContext, Table}};
|
|
{"".join(autoinc1_template.substitute(KEY_TY=int_ty) for int_ty in ints)}
|
|
"""
|
|
|
|
def do_test_autoinc(self, int_ty):
|
|
self.call(f"add_{int_ty}", "Robert", 1)
|
|
self.call(f"add_{int_ty}", "Julie", 2)
|
|
self.call(f"add_{int_ty}", "Samantha", 3)
|
|
self.call(f"say_hello_{int_ty}")
|
|
logs = self.logs(4)
|
|
self.assertIn("Hello, 3:Samantha!", logs)
|
|
self.assertIn("Hello, 2:Julie!", logs)
|
|
self.assertIn("Hello, 1:Robert!", logs)
|
|
self.assertIn("Hello, World!", logs)
|
|
|
|
|
|
|
|
autoinc2_template = string.Template("""
|
|
#[spacetimedb::table(name = person_$KEY_TY)]
|
|
pub struct Person_$KEY_TY {
|
|
#[auto_inc]
|
|
#[unique]
|
|
key_col: $KEY_TY,
|
|
#[unique]
|
|
name: String,
|
|
}
|
|
|
|
#[spacetimedb::reducer]
|
|
pub fn add_new_$KEY_TY(ctx: &ReducerContext, name: String) -> Result<(), Box<dyn Error>> {
|
|
let value = ctx.db.person_$KEY_TY().try_insert(Person_$KEY_TY { key_col: 0, name })?;
|
|
log::info!("Assigned Value: {} -> {}", value.key_col, value.name);
|
|
Ok(())
|
|
}
|
|
|
|
#[spacetimedb::reducer]
|
|
pub fn update_$KEY_TY(ctx: &ReducerContext, name: String, new_id: $KEY_TY) {
|
|
ctx.db.person_$KEY_TY().name().delete(&name);
|
|
let _value = ctx.db.person_$KEY_TY().insert(Person_$KEY_TY { key_col: new_id, name });
|
|
}
|
|
|
|
#[spacetimedb::reducer]
|
|
pub fn say_hello_$KEY_TY(ctx: &ReducerContext) {
|
|
for person in ctx.db.person_$KEY_TY().iter() {
|
|
log::info!("Hello, {}:{}!", person.key_col, person.name);
|
|
}
|
|
log::info!("Hello, World!");
|
|
}
|
|
""")
|
|
|
|
|
|
class AutoincUnique(IntTests, Smoketest):
|
|
"""This tests unique constraints being violated during autoinc insertion"""
|
|
|
|
MODULE_CODE = f"""
|
|
#![allow(non_camel_case_types)]
|
|
use std::error::Error;
|
|
use spacetimedb::{{log, ReducerContext, Table}};
|
|
{"".join(autoinc2_template.substitute(KEY_TY=int_ty) for int_ty in ints)}
|
|
"""
|
|
|
|
def do_test_autoinc(self, int_ty):
|
|
self.call(f"update_{int_ty}", "Robert", 2)
|
|
self.call(f"add_new_{int_ty}", "Success")
|
|
with self.assertRaises(Exception):
|
|
self.call(f"add_new_{int_ty}", "Failure")
|
|
|
|
self.call(f"say_hello_{int_ty}")
|
|
logs = self.logs(4)
|
|
self.assertIn("Hello, 2:Robert!", logs)
|
|
self.assertIn("Hello, 1:Success!", logs)
|
|
self.assertIn("Hello, World!", logs)
|