Struct DataDrivenGenerator 

pub struct DataDrivenGenerator;

Generates k6 JavaScript code for data-driven testing

Implementations

impl DataDrivenGenerator

pub fn generate_imports(file_type: DataFileType) -> String

Generate k6 imports for data-driven testing

pub fn generate_data_loading(config: &DataDrivenConfig) -> String

Generate k6 code to load the data file

pub fn generate_row_selection(distribution: DataDistribution) -> String

Generate k6 code for row selection based on distribution strategy

pub fn generate_apply_mappings(mappings: &[DataMapping]) -> String

Generate k6 code to apply mappings from row data

pub fn generate_setup(config: &DataDrivenConfig) -> String

Generate complete data-driven test setup code

pub fn generate_iteration_code(config: &DataDrivenConfig) -> String

Generate code for within the default function

pub fn generate_per_uri_control_code(config: &DataDrivenConfig) -> String

Generate k6 code for per-URI control mode

This mode allows each row in the CSV/JSON to specify:

• HTTP method (GET, POST, PUT, PATCH, DELETE)
• URI/path to call
• Request body (JSON string)
• Query parameters (JSON string or key=value&key=value format)
• Additional headers (JSON string)
• Attack type for security testing
• Expected status code for validation

Example CSV:

method,uri,body,query_params,headers,attack_type,expected_status
GET,/virtualservice,,include_name=true,,sqli,200
POST,/virtualservice,"{\"name\":\"test-vs\",\"port\":80}",,,,201
PUT,/virtualservice/{uuid},"{\"name\":\"updated-vs\"}",,,,200
DELETE,/virtualservice/{uuid},,,,xss,204

pub fn generate_per_uri_metrics(operations: &[(String, String)]) -> String

Generate metrics declarations for per-URI control mode

pub fn generate_per_uri_control_script( config: &DataDrivenConfig, target_url: &str, custom_headers: &HashMap<String, String>, skip_tls_verify: bool, ) -> String

Generate a complete per-URI control mode script