Skip to content

Commit

Permalink
tasks: add more dependency tests
Browse files Browse the repository at this point in the history
  • Loading branch information
@sandydoo
sandydoo committed Nov 23, 2024
1 parent 4726277 commit bc2b55d
Showing 1 changed file with 103 additions and 22 deletions.
125 changes: 103 additions & 22 deletions devenv-tasks/src/lib.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -1145,15 +1145,9 @@ mod test {

#[tokio::test]
async fn test_before_tasks() -> Result<(), Error> {
let script1 = create_script(
"#!/bin/sh\necho 'Task 1 is running' && sleep 0.5 && echo 'Task 1 completed'",
)?;
let script2 = create_script(
"#!/bin/sh\necho 'Task 2 is running' && sleep 0.5 && echo 'Task 2 completed'",
)?;
let script3 = create_script(
"#!/bin/sh\necho 'Task 3 is running' && sleep 0.5 && echo 'Task 3 completed'",
)?;
let script1 = create_basic_script("1")?;
let script2 = create_basic_script("2")?;
let script3 = create_basic_script("3")?;

let tasks = Tasks::new(
Config::try_from(json!({
Expand All @@ -1162,7 +1156,7 @@ mod test {
{
"name": "myapp:task_1",
"command": script1.to_str().unwrap(),
"after": ["myapp:task_3"]
"before": ["myapp:task_2", "myapp:task_3"]
},
{
"name": "myapp:task_2",
Expand All @@ -1188,22 +1182,60 @@ mod test {
(name1, TaskStatus::Completed(TaskCompleted::Success(_, _))),
(name2, TaskStatus::Completed(TaskCompleted::Success(_, _))),
(name3, TaskStatus::Completed(TaskCompleted::Success(_, _)))
] if name1 == "myapp:task_2" && name2 == "myapp:task_3" && name3 == "myapp:task_1"
] if name1 == "myapp:task_1" && name2 == "myapp:task_2" && name3 == "myapp:task_3"
);
Ok(())
}

#[tokio::test]
async fn test_after_tasks() -> Result<(), Error> {
let script1 = create_basic_script("1")?;
let script2 = create_basic_script("2")?;
let script3 = create_basic_script("3")?;

let tasks = Tasks::new(
Config::try_from(json!({
"roots": ["myapp:task_1"],
"tasks": [
{
"name": "myapp:task_1",
"command": script1.to_str().unwrap(),
"after": ["myapp:task_3", "myapp:task_2"]
},
{
"name": "myapp:task_2",
"after": ["myapp:task_3"],
"command": script2.to_str().unwrap()
},
{
"name": "myapp:task_3",
"command": script3.to_str().unwrap()
}
]
}))
.unwrap(),
)
.await?;
tasks.run().await;

let task_statuses = inspect_tasks(&tasks).await;
let task_statuses = task_statuses.as_slice();
assert_matches!(
task_statuses,
[
(name1, TaskStatus::Completed(TaskCompleted::Success(_, _))),
(name2, TaskStatus::Completed(TaskCompleted::Success(_, _))),
(name3, TaskStatus::Completed(TaskCompleted::Success(_, _)))
] if name1 == "myapp:task_3" && name2 == "myapp:task_2" && name3 == "myapp:task_1"
);
Ok(())
}

#[tokio::test]
async fn test_before_and_after_tasks() -> Result<(), Error> {
let script1 = create_script(
"#!/bin/sh\necho 'Task 1 is running' && sleep 0.5 && echo 'Task 1 completed'",
)?;
let script2 = create_script(
"#!/bin/sh\necho 'Task 2 is running' && sleep 0.5 && echo 'Task 2 completed'",
)?;
let script3 = create_script(
"#!/bin/sh\necho 'Task 3 is running' && sleep 0.5 && echo 'Task 3 completed'",
)?;
let script1 = create_basic_script("1")?;
let script2 = create_basic_script("2")?;
let script3 = create_basic_script("3")?;

let tasks = Tasks::new(
Config::try_from(json!({
Expand Down Expand Up @@ -1244,6 +1276,51 @@ mod test {
Ok(())
}

// Test that tasks indirectly linked to the root are picked up and run.
#[tokio::test]
async fn test_transitive_dependencies() -> Result<(), Error> {
let script1 = create_basic_script("1")?;
let script2 = create_basic_script("2")?;
let script3 = create_basic_script("3")?;

let tasks = Tasks::new(
Config::try_from(json!({
"roots": ["myapp:task_3"],
"tasks": [
{
"name": "myapp:task_1",
"command": script1.to_str().unwrap(),
},
{
"name": "myapp:task_2",
"after": ["myapp:task_1"],
"command": script2.to_str().unwrap()
},
{
"name": "myapp:task_3",
"after": ["myapp:task_2"],
"command": script3.to_str().unwrap()
},
]
}))
.unwrap(),
)
.await?;
tasks.run().await;

let task_statuses = inspect_tasks(&tasks).await;
let task_statuses = task_statuses.as_slice();
assert_matches!(
task_statuses,
[
(name1, TaskStatus::Completed(TaskCompleted::Success(_, _))),
(name2, TaskStatus::Completed(TaskCompleted::Success(_, _))),
(name3, TaskStatus::Completed(TaskCompleted::Success(_, _)))
] if name1 == "myapp:task_1" && name2 == "myapp:task_2" && name3 == "myapp:task_3"
);
Ok(())
}

#[tokio::test]
async fn test_dependency_failure() -> Result<(), Error> {
let failing_script = create_script("#!/bin/sh\necho 'Failing task' && exit 1")?;
Expand Down Expand Up @@ -1405,7 +1482,6 @@ fi
Ok(())
}

#[cfg(test)]
async fn inspect_tasks(tasks: &Tasks) -> Vec<(String, TaskStatus)> {
let mut result = Vec::new();
for index in &tasks.tasks_order {
Expand All @@ -1415,7 +1491,6 @@ fi
result
}

#[cfg(test)]
fn create_script(script: &str) -> std::io::Result<tempfile::TempPath> {
let mut temp_file = tempfile::Builder::new()
.prefix("script")
Expand All @@ -1427,4 +1502,10 @@ fi
.set_permissions(fs::Permissions::from_mode(0o755))?;
Ok(temp_file.into_temp_path())
}

fn create_basic_script(tag: &str) -> std::io::Result<tempfile::TempPath> {
create_script(&format!(
"#!/bin/sh\necho 'Task {tag} is running' && sleep 0.1 && echo 'Task {tag} completed'"
))
}
}

0 comments on commit bc2b55d

Please sign in to comment.