ocppsim 0.1.1

use serde_json::json;

use super::*;

#[test]
fn get_configuration_reports_unknown_keys() {
  let simulator = simulator_for_tests();
  let payload = json!({ "key": ["HeartbeatInterval", "MissingKey"] });
  let response = simulator.configuration_response_v1_6(&payload);

  assert_eq!(response.configuration_key.len(), 1);
  assert_eq!(response.unknown_key, vec!["MissingKey".to_string()]);
}

#[test]
fn change_configuration_updates_writable_key() {
  let mut simulator = simulator_for_tests();
  let payload = json!({
    "key": "HeartbeatInterval",
    "value": "25"
  });
  let status = simulator.change_configuration_v1_6(&payload);

  assert_eq!(status, ResponseStatus::Accepted);
  let entry = simulator
    .configuration
    .get(&ConfigurationKey::HeartbeatInterval)
    .expect("key must exist");
  assert_eq!(entry.value, "25");
}

#[test]
fn change_configuration_rejects_read_only_key() {
  let mut simulator = simulator_for_tests();
  let payload = json!({
    "key": "NumberOfConnectors",
    "value": "8"
  });
  let status = simulator.change_configuration_v1_6(&payload);

  assert_eq!(status, ResponseStatus::Rejected);
}

#[test]
fn get_variables_v2_0_1_reads_configuration() {
  assert_get_variables_reads_configuration(
    simulator_for_tests_v2_0_1(),
    "schemas/2.0.1/GetVariablesResponse.json",
  );
}

#[test]
fn get_variables_v2_1_reads_configuration() {
  assert_get_variables_reads_configuration(
    simulator_for_tests_v2_1(),
    "schemas/2.1/GetVariablesResponse.json",
  );
}

fn assert_get_variables_reads_configuration(
  simulator: Simulator,
  schema: &str,
) {
  let response = simulator
    .get_variables_v2_x(&json!({
      "getVariableData": [
        get_variable_data("HeartbeatInterval"),
        get_variable_data("MissingKey"),
      ]
    }))
    .expect("get variables response");

  assert_eq!(
    response["getVariableResult"][0]["attributeStatus"],
    "Accepted"
  );
  assert_eq!(response["getVariableResult"][0]["attributeValue"], "10");
  assert_eq!(
    response["getVariableResult"][1]["attributeStatus"],
    "UnknownVariable"
  );
  assert_schema_valid(schema, &response);
}

#[test]
fn get_variables_v2_x_reports_component_and_attribute_errors() {
  let simulator = simulator_for_tests_v2_0_1();
  let response = simulator
    .get_variables_v2_x(&json!({
      "getVariableData": [
        {
          "component": { "name": "Connector" },
          "variable": { "name": "HeartbeatInterval" }
        },
        {
          "component": { "name": "ChargingStation" },
          "variable": { "name": "HeartbeatInterval" },
          "attributeType": "MinSet"
        }
      ]
    }))
    .expect("get variables response");

  assert_eq!(
    response["getVariableResult"][0]["attributeStatus"],
    "UnknownComponent"
  );
  assert_eq!(
    response["getVariableResult"][1]["attributeStatus"],
    "NotSupportedAttributeType"
  );
  assert_schema_valid("schemas/2.0.1/GetVariablesResponse.json", &response);
}

#[test]
fn set_variables_v2_0_1_updates_configuration() {
  assert_set_variables_updates_configuration(
    simulator_for_tests_v2_0_1(),
    "schemas/2.0.1/SetVariablesResponse.json",
  );
}

#[test]
fn set_variables_v2_1_updates_configuration() {
  assert_set_variables_updates_configuration(
    simulator_for_tests_v2_1(),
    "schemas/2.1/SetVariablesResponse.json",
  );
}

fn assert_set_variables_updates_configuration(
  mut simulator: Simulator,
  schema: &str,
) {
  let response = simulator
    .set_variables_v2_x(&json!({
      "setVariableData": [
        set_variable_data("HeartbeatInterval", "20"),
        set_variable_data("NumberOfConnectors", "8"),
      ]
    }))
    .expect("set variables response");

  assert_eq!(
    response["setVariableResult"][0]["attributeStatus"],
    "Accepted"
  );
  assert_eq!(
    response["setVariableResult"][1]["attributeStatus"],
    "Rejected"
  );
  assert_eq!(
    simulator
      .configuration
      .get(&ConfigurationKey::HeartbeatInterval)
      .map(|entry| entry.value.as_str()),
    Some("20")
  );
  assert_schema_valid(schema, &response);
}

#[test]
fn data_transfer_reports_missing_vendor_for_both_protocol_families() {
  let v1_6_simulator = simulator_for_tests();
  let v2_x_simulator = simulator_for_tests_v2_0_1();

  let v1_6_response = v1_6_simulator.data_transfer_v1_6(&json!({}));
  let v2_x_response = v2_x_simulator.data_transfer_v2_x(&json!({}));

  assert_eq!(
    v1_6_response["status"],
    json!(ResponseStatus::UnknownVendorId.as_str())
  );
  assert_eq!(
    v2_x_response["status"],
    json!(ResponseStatus::UnknownVendorId.as_str())
  );
}

#[tokio::test]
async fn set_variables_restarts_active_heartbeat() {
  let mut simulator = simulator_for_tests_v2_0_1();
  simulator.start_heartbeat(10);

  let response = simulator
    .set_variables_v2_x(&json!({
      "setVariableData": [
        set_variable_data("HeartbeatInterval", "22")
      ]
    }))
    .expect("set variables response");

  assert_eq!(
    response["setVariableResult"][0]["attributeStatus"],
    ResponseStatus::Accepted.as_str()
  );
  assert_eq!(
    simulator.heartbeat.as_ref().map(|item| item.seconds),
    Some(22)
  );
  simulator.stop_heartbeat();
}