sk-ctrl 2.6.0

Kubernetes controller for replaying traces in a simulated cluster
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

use std::env;

use assertables::*;
use tracing_test::traced_test;

use super::*;
use crate::controller::ReconcileContext;
use crate::objects::{
    TRACE_VOLUME_NAME,
    build_driver_job,
    build_local_trace_volume,
};

#[rstest(tokio::test)]
async fn test_build_driver_job_with_extra_args(mut test_sim: Simulation) {
    // this_is_fine.jpg
    unsafe {
        env::set_var(POD_SVC_ACCOUNT_ENV_VAR, TEST_SERVICE_ACCOUNT);
    }

    test_sim.spec.driver.args = Some(vec!["--foo".into(), "bar".into(), "--baz".into()]);
    let (_, client) = make_fake_apiserver();
    let ctx = Arc::new(ReconcileContext::new(&test_sim, client.clone(), SkEventRecorder::mock()));
    let job = build_driver_job(&ctx, &test_sim, "secret", TEST_NAMESPACE).unwrap();

    let job_spec = job.spec.unwrap().template.spec.unwrap();
    let container = job_spec.containers.get(0).unwrap();
    let expected_args = vec![
        "--cert-path",
        "/usr/local/etc/ssl/tls.crt",
        "--key-path",
        "/usr/local/etc/ssl/tls.key",
        "--trace-path",
        "/foo/bar",
        "--sim-name",
        TEST_SIM_NAME.into(),
        "--controller-ns",
        "test-namespace",
        "--foo",
        "bar",
        "--baz",
    ];
    let expected_secrets = vec![corev1::EnvFromSource {
        secret_ref: Some(corev1::SecretEnvSource {
            name: TEST_DRIVER_SECRET_NAME.into(),
            optional: Some(false),
        }),
        ..Default::default()
    }];
    assert_iter_eq!(container.args.as_ref().unwrap(), expected_args);
    assert_iter_eq!(container.env_from.as_ref().unwrap(), expected_secrets);
}

#[rstest]
fn test_build_local_trace_volume_not_local(mut test_sim: Simulation) {
    test_sim.spec.driver.trace_path = "s3://foo/bar".into();
    let res = build_local_trace_volume(&test_sim).unwrap();
    assert_none!(res);
}

#[rstest]
fn test_build_local_trace_volume_skip_volume_mount(mut test_sim: Simulation) {
    test_sim
        .annotations_mut()
        .insert(SKIP_LOCAL_VOLUME_MOUNT_ANNOTATION_KEY.into(), "true".into());
    let res = build_local_trace_volume(&test_sim).unwrap();
    assert_none!(res);
}

#[rstest]
#[traced_test]
fn test_build_local_trace_volume_skip_volume_mount_not_local(mut test_sim: Simulation) {
    test_sim
        .annotations_mut()
        .insert(SKIP_LOCAL_VOLUME_MOUNT_ANNOTATION_KEY.into(), "true".into());
    test_sim.spec.driver.trace_path = "s3://foo/bar".into();
    let res = build_local_trace_volume(&test_sim).unwrap();
    assert_none!(res);
    assert!(logs_contain("ignoring annotation"));
}

#[rstest]
fn test_build_local_trace_volume(test_sim: Simulation) {
    let res = build_local_trace_volume(&test_sim).unwrap();
    let (mount, volume, path) = res.unwrap();

    assert_eq!(mount.name, TRACE_VOLUME_NAME);
    assert_eq!(mount.mount_path, "/foo/bar");
    assert_eq!(volume.name, TRACE_VOLUME_NAME);
    let host_path = volume.host_path.unwrap();
    assert_eq!(host_path.path, "/foo/bar");
    assert_eq!(host_path.type_, Some("File".into()));
    assert_eq!(path, "/foo/bar");
}