1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96

use serde::{Deserialize, Serialize};
#[cfg(feature = "sqlx")]
use sqlx::{FromRow, Type};
use time::OffsetDateTime;
use uuid::Uuid;

use crate::{CreateStepRequest, ExecutionResult};

#[derive(Debug, Deserialize, Serialize)]
#[cfg_attr(feature = "sqlx", derive(FromRow))]
pub struct Step {
    pub execution_uuid: Uuid,
    pub state: StepState,
    pub result: StepResult,
    #[serde(with = "time::serde::rfc3339")]
    pub created_at: OffsetDateTime,
}

impl Default for Step {
    fn default() -> Self {
        Self {
            execution_uuid: Default::default(),
            state: StepState::Setup,
            result: StepResult::Skipped,
            created_at: OffsetDateTime::now_utc(),
        }
    }
}

impl Step {
    pub fn execution_result(&self) -> Option<ExecutionResult> {
        match (&self.state, &self.result) {
            (_, StepResult::Failed) => Some(ExecutionResult::Failed),
            (StepState::Teardown, StepResult::Passed | StepResult::Skipped) => {
                Some(ExecutionResult::Passed)
            }
            (StepState::Setup | StepState::Run, StepResult::Passed | StepResult::Skipped) => None,
        }
    }
}

#[cfg_attr(feature = "sqlx", derive(Type))]
#[cfg_attr(
    feature = "sqlx",
    sqlx(type_name = "step_state", rename_all = "snake_case")
)]
#[derive(Debug, Deserialize, Serialize)]
#[serde(rename_all = "SCREAMING_SNAKE_CASE")]
pub enum StepState {
    Setup,
    Run,
    Teardown,
}

#[cfg_attr(feature = "sqlx", derive(Type))]
#[cfg_attr(
    feature = "sqlx",
    sqlx(type_name = "step_result", rename_all = "snake_case")
)]
#[derive(Debug, Deserialize, Serialize)]
#[serde(rename_all = "SCREAMING_SNAKE_CASE")]
pub enum StepResult {
    Passed,
    Failed,
    Skipped,
}

pub enum ExhaustiveSteps {
    SetupPassed,
    SetupFailed,
    SetupSkipped,
    RunPassed,
    RunFailed,
    RunSkipped,
    TeardownPassed,
    TeardownFailed,
    TeardownSkipped,
}

impl From<ExhaustiveSteps> for CreateStepRequest {
    fn from(exhaustive_step: ExhaustiveSteps) -> Self {
        let (state, result) = match exhaustive_step {
            ExhaustiveSteps::SetupPassed => (StepState::Setup, StepResult::Passed),
            ExhaustiveSteps::SetupFailed => (StepState::Setup, StepResult::Failed),
            ExhaustiveSteps::SetupSkipped => (StepState::Setup, StepResult::Skipped),
            ExhaustiveSteps::RunPassed => (StepState::Run, StepResult::Passed),
            ExhaustiveSteps::RunFailed => (StepState::Run, StepResult::Failed),
            ExhaustiveSteps::RunSkipped => (StepState::Run, StepResult::Skipped),
            ExhaustiveSteps::TeardownPassed => (StepState::Teardown, StepResult::Passed),
            ExhaustiveSteps::TeardownFailed => (StepState::Teardown, StepResult::Failed),
            ExhaustiveSteps::TeardownSkipped => (StepState::Teardown, StepResult::Skipped),
        };

        Self { state, result }
    }
}