feat(users): user wf tests

Author: abcxff

packages/services/monolith/standalone/workflow-worker/Cargo.toml

@@ -17,3 +17,4 @@ job-run.workspace = true
 linode.workspace = true
 pegboard.workspace = true
 rivet-config.workspace = true
+user.workspace = true

packages/services/monolith/standalone/workflow-worker/src/lib.rs

@@ -15,7 +15,8 @@ pub async fn run_from_env(
 		.merge(linode::registry()?)?
 		.merge(ds::registry()?)?
 		.merge(job_run::registry()?)?
-		.merge(pegboard::registry()?)?;
+		.merge(pegboard::registry()?)?
+		.merge(user::registry()?)?;
 
 	let db = db::DatabaseCrdbNats::from_pools(pools.crdb()?, pools.nats()?);
 	let worker = Worker::new(reg.handle(), db);

packages/services/user/src/lib.rs

@@ -1,3 +1,14 @@
+use chirp_workflow::prelude::*;
+
 pub mod ops;
 pub mod types;
 pub mod workflows;
+
+pub fn registry() -> WorkflowResult<Registry> {
+	use workflows::*;
+
+	let mut registry = Registry::new();
+	registry.register_workflow::<user::Workflow>()?;
+
+	Ok(registry)
+}

packages/services/user/src/ops/faker.rs

@@ -0,0 +1,33 @@
+// use chirp_workflow::prelude::*;
+
+// #[derive(Debug, Default)]
+// pub struct Input {}
+
+// #[derive(Debug)]
+// pub struct Output {
+// 	pub user_id: Uuid,
+// }
+
+// #[operation]
+// async fn user(ctx: &OperationCtx, input: &Input) -> GlobalResult<Output>  {
+// 	let user_id = Uuid::new_v4();
+
+// 	// TODO: Move back to faker op after ops can dispatch workflows
+//     let mut creation_sub = ctx
+//         .subscribe::<crate::workflows::user::CreateComplete>(("user_id", user_id))
+//         .await?;
+
+//     ctx.workflow(crate::workflows::user::Input {
+// 		user_id,
+// 		display_name: None,
+// 	})
+// 	.tag("user_id", user_id)
+// 	.dispatch()
+// 	.await?;
+
+//     creation_sub.next().await?;
+
+// 	Ok(Output {
+// 		user_id: user_id,
+// 	})
+// }

packages/services/user/src/ops/mod.rs

@@ -6,5 +6,5 @@ pub mod resolve_display_name;
 pub mod resolve_email;
 pub mod team_list;
 pub mod token_create;
-
+// pub mod faker;
 pub mod identity;

packages/services/user/src/workflows/user/mod.rs

@@ -24,7 +24,7 @@ pub struct Input {
 }
 
 #[workflow]
-pub async fn user(ctx: &mut WorkflowCtx, input: &Input) -> GlobalResult<()> {	
+pub async fn user(ctx: &mut WorkflowCtx, input: &Input) -> GlobalResult<()> {
 	let (display_name, _account_number) = ctx.activity(InsertDbInput {
 		user_id: input.user_id,
 		display_name: input.display_name.clone(),
@@ -273,7 +273,7 @@ async fn insert_db(ctx: &ActivityCtx, input: &InsertDbInput) -> GlobalResult<(St
 	};
 
 	let account_number = gen_account_number();
-	tracing::debug!(%display_name, %account_number, "insert user attempt");
+	tracing::info!(%display_name, %account_number, "insert user attempt");
 
     sql_execute!(
 		[ctx]

packages/services/user/tests/avatar_upload_complete.rs

@@ -1,12 +1,14 @@
 use chirp_workflow::prelude::*;
 use rivet_operation::prelude::proto::backend;
 
+mod common;
+
 const TEST_BODY: &[u8] = b"test file";
 
 #[workflow_test]
 async fn empty(ctx: TestCtx) {
-	let user_res = op!([ctx] faker_user {}).await.unwrap();
-	let user_id = user_res.user_id.unwrap().as_uuid();
+	let user_res = common::make_test_user(&ctx).await.unwrap();
+	let user_id = user_res.user_id;
 
 	// Create the upload
 	let upload_prepare_res = op!([ctx] upload_prepare {

packages/services/user/tests/common/mod.rs

@@ -0,0 +1,32 @@
+use chirp_workflow::prelude::*;
+
+// TODO: Move back to faker op after ops can dispatch workflows
+#[derive(Debug)]
+pub struct FakerUserOutput {
+    pub user_id: Uuid,
+}
+
+pub async fn make_test_user(ctx: &TestCtx) -> GlobalResult<FakerUserOutput> {
+    tracing::info!("attempting to make test user");
+    let user_id = Uuid::new_v4();
+
+    let mut creation_sub = ctx
+        .subscribe::<user::workflows::user::CreateComplete>(("user_id", user_id))
+        .await?;
+
+    ctx.workflow(user::workflows::user::Input {
+		user_id,
+		display_name: None,
+	})
+	.tag("user_id", user_id)
+	.dispatch()
+    .await?;
+
+    creation_sub.next().await?;
+
+    tracing::info!("creation sub received");
+
+    Ok(FakerUserOutput {
+        user_id
+    })
+}

packages/services/user/tests/get.rs

@@ -3,6 +3,8 @@ use proto::backend::{pkg::*};
 use rivet_operation::prelude::proto;
 use rand::Rng;
 
+mod common;
+
 #[workflow_test]
 async fn empty(ctx: TestCtx) {
 	let res = ctx.op(::user::ops::get::Input {

packages/services/user/tests/identity_create.rs

@@ -1,14 +1,12 @@
 use chirp_workflow::prelude::*;
 use rivet_operation::prelude::proto::backend;
 
+mod common;
+
 #[workflow_test]
 async fn email(ctx: TestCtx) {
-	let user_res = op!([ctx] faker_user {
-		..Default::default()
-	})
-	.await
-	.unwrap();
-    let user_id = user_res.user_id.unwrap().as_uuid();
+	let user_res = common::make_test_user(&ctx).await.unwrap();
+	let user_id = user_res.user_id;
 
 	let email = util::faker::email();
 	ctx.op(::user::ops::identity::create::Input {