use pretty_assertions::assert_eq;

#[test]
fn vertex_entries() {
    // Check vertex entry points and builtin attribute handling.
    let actual = wgsl_to_wgpu::create_shader_module(
        include_str!("wgsl/vertex_entries.wgsl"),
        "shader.wgsl",
        wgsl_to_wgpu::WriteOptions {
            rustfmt: true,
            ..Default::default()
        },
    )
    .unwrap();

    assert_eq!(include_str!("output/vertex_entries.rs"), actual);
}

#[test]
fn shader_stage_collection() {
    // Check the visibility: wgpu::ShaderStages::COMPUTE
    let actual = wgsl_to_wgpu::create_shader_module(
        include_str!("wgsl/shader_stage_collection.wgsl"),
        "shader.wgsl",
        wgsl_to_wgpu::WriteOptions {
            rustfmt: true,
            derive_encase_host_shareable: true,
            ..Default::default()
        },
    )
    .unwrap();

    assert_eq!(include_str!("output/shader_stage_collection.rs"), actual);
}