cdrs-async 0.1.0-alpha.0

use cassandra_proto::frame::AsByte;

const FLAGS_VALUE: u8 = 0x01;
const FLAGS_SKIP_METADATA: u8 = 0x02;
const WITH_PAGE_SIZE: u8 = 0x04;
const WITH_PAGING_STATE: u8 = 0x08;
const WITH_SERIAL_CONSISTENCY: u8 = 0x10;
const WITH_DEFAULT_TIMESTAMP: u8 = 0x20;
const WITH_NAME_FOR_VALUES: u8 = 0x40;

/// Cassandra Query Flags.
#[derive(Clone, Debug)]
pub enum QueryFlags {
  /// If set indicates that Query Params contains value.
  Value,
  /// If set indicates that Query Params does not contain metadata.
  SkipMetadata,
  /// If set indicates that Query Params contains page size.
  PageSize,
  /// If set indicates that Query Params contains paging state.
  WithPagingState,
  /// If set indicates that Query Params contains serial consistency.
  WithSerialConsistency,
  /// If set indicates that Query Params contains default timestamp.
  WithDefaultTimestamp,
  /// If set indicates that Query Params values are named ones.
  WithNamesForValues,
}

impl QueryFlags {
  #[doc(hidden)]
  pub fn has_value(byte: u8) -> bool {
    (byte & FLAGS_VALUE) != 0
  }

  #[doc(hidden)]
  pub fn set_value(byte: u8) -> u8 {
    byte | FLAGS_VALUE
  }

  #[doc(hidden)]
  pub fn has_skip_metadata(byte: u8) -> bool {
    (byte & FLAGS_SKIP_METADATA) != 0
  }

  #[doc(hidden)]
  pub fn set_skip_metadata(byte: u8) -> u8 {
    byte | FLAGS_SKIP_METADATA
  }

  #[doc(hidden)]
  pub fn has_page_size(byte: u8) -> bool {
    (byte & WITH_PAGE_SIZE) != 0
  }

  #[doc(hidden)]
  pub fn set_page_size(byte: u8) -> u8 {
    byte | WITH_PAGE_SIZE
  }

  #[doc(hidden)]
  pub fn has_with_paging_state(byte: u8) -> bool {
    (byte & WITH_PAGING_STATE) != 0
  }

  #[doc(hidden)]
  pub fn set_with_paging_state(byte: u8) -> u8 {
    byte | WITH_PAGING_STATE
  }

  #[doc(hidden)]
  pub fn has_with_serial_consistency(byte: u8) -> bool {
    (byte & WITH_SERIAL_CONSISTENCY) != 0
  }

  #[doc(hidden)]
  pub fn set_with_serial_consistency(byte: u8) -> u8 {
    byte | WITH_SERIAL_CONSISTENCY
  }

  #[doc(hidden)]
  pub fn has_with_default_timestamp(byte: u8) -> bool {
    (byte & WITH_DEFAULT_TIMESTAMP) != 0
  }

  #[doc(hidden)]
  pub fn set_with_default_timestamp(byte: u8) -> u8 {
    byte | WITH_DEFAULT_TIMESTAMP
  }

  #[doc(hidden)]
  pub fn has_with_names_for_values(byte: u8) -> bool {
    (byte & WITH_NAME_FOR_VALUES) != 0
  }

  #[doc(hidden)]
  pub fn set_with_names_for_values(byte: u8) -> u8 {
    byte | WITH_NAME_FOR_VALUES
  }
}

impl AsByte for QueryFlags {
  fn as_byte(&self) -> u8 {
    match *self {
      QueryFlags::Value => FLAGS_VALUE,
      QueryFlags::SkipMetadata => FLAGS_SKIP_METADATA,
      QueryFlags::PageSize => WITH_PAGE_SIZE,
      QueryFlags::WithPagingState => WITH_PAGING_STATE,
      QueryFlags::WithSerialConsistency => WITH_SERIAL_CONSISTENCY,
      QueryFlags::WithDefaultTimestamp => WITH_DEFAULT_TIMESTAMP,
      QueryFlags::WithNamesForValues => WITH_NAME_FOR_VALUES,
    }
  }
}

#[cfg(test)]
mod tests {
  use super::*;

  #[test]
  fn has_value_test() {
    assert!(
      QueryFlags::has_value(FLAGS_VALUE | 0x10),
      "should show that the flag has value"
    );
    assert!(
      !QueryFlags::has_value(FLAGS_SKIP_METADATA),
      "should show that the flag does NOT have value"
    );
  }

  #[test]
  fn set_value_test() {
    assert_eq!(
      QueryFlags::set_value(0),
      FLAGS_VALUE,
      "should set has value flag"
    );
  }

  #[test]
  fn has_skip_metadata_test() {
    assert!(
      QueryFlags::has_skip_metadata(FLAGS_SKIP_METADATA | 0x10),
      "should show that the flag has skip metadata"
    );
    assert!(
      !QueryFlags::has_skip_metadata(FLAGS_VALUE),
      "should show that the flag does NOT have skip metadata"
    );
  }

  #[test]
  fn set_skip_metadata_test() {
    assert_eq!(
      QueryFlags::set_skip_metadata(0),
      FLAGS_SKIP_METADATA,
      "should set has skip metadata flag"
    );
  }

  #[test]
  fn has_page_size_test() {
    assert!(
      QueryFlags::has_page_size(WITH_PAGE_SIZE | 0x10),
      "should show that the flag has with page size"
    );
    assert!(
      !QueryFlags::has_page_size(FLAGS_VALUE),
      "should show that the flag does NOT have with page size"
    );
  }

  #[test]
  fn set_page_size_test() {
    assert_eq!(
      QueryFlags::set_page_size(0),
      WITH_PAGE_SIZE,
      "should set has page size flag"
    );
  }

  #[test]
  fn has_with_paging_state_test() {
    assert!(
      QueryFlags::has_with_paging_state(WITH_PAGING_STATE | 0x10),
      "should show that the flag has with paging state"
    );
    assert!(
      !QueryFlags::has_with_paging_state(FLAGS_VALUE),
      "should show that the flag does NOT have with paging state"
    );
  }

  #[test]
  fn set_with_paging_state_test() {
    assert_eq!(
      QueryFlags::set_with_paging_state(0),
      WITH_PAGING_STATE,
      "should set has with paging state flag"
    );
  }

  #[test]
  fn has_with_serial_consistency_test() {
    assert!(
      QueryFlags::has_with_serial_consistency(WITH_SERIAL_CONSISTENCY | 0x11),
      "should show that the flag has with serial consistency"
    );
    assert!(
      !QueryFlags::has_with_serial_consistency(FLAGS_VALUE),
      "should show that the flag does NOT have with serial consistency"
    );
  }

  #[test]
  fn set_with_serial_consistency_test() {
    assert_eq!(
      QueryFlags::set_with_serial_consistency(0),
      WITH_SERIAL_CONSISTENCY,
      "should set has with serial consistency flag"
    );
  }

  #[test]
  fn has_with_default_timestamp_test() {
    assert!(
      QueryFlags::has_with_default_timestamp(WITH_DEFAULT_TIMESTAMP | 0x10),
      "should show that the flag has with default timestamp"
    );
    assert!(
      !QueryFlags::has_with_default_timestamp(FLAGS_VALUE),
      "should show that the flag does NOT have with default timestamp"
    );
  }

  #[test]
  fn set_with_default_timestamp_test() {
    assert_eq!(
      QueryFlags::set_with_default_timestamp(0),
      WITH_DEFAULT_TIMESTAMP,
      "should set has with serial consistency flag"
    );
  }

  #[test]
  fn has_with_names_for_values_test() {
    assert!(
      QueryFlags::has_with_names_for_values(WITH_NAME_FOR_VALUES | 0x10),
      "should show that the flag has with name for values"
    );
    assert!(
      !QueryFlags::has_with_names_for_values(FLAGS_VALUE),
      "should show that the flag does NOT have with name for values"
    );
  }

  #[test]
  fn set_with_names_for_values_test() {
    assert_eq!(
      QueryFlags::set_with_names_for_values(0),
      WITH_NAME_FOR_VALUES,
      "should set has with name for values flag"
    );
  }

  #[test]
  fn as_byte_test() {
    assert_eq!(
      QueryFlags::Value.as_byte(),
      FLAGS_VALUE,
      "should propery convert values flag"
    );

    assert_eq!(
      QueryFlags::SkipMetadata.as_byte(),
      FLAGS_SKIP_METADATA,
      "should propery convert skip metadata flag"
    );

    assert_eq!(
      QueryFlags::PageSize.as_byte(),
      WITH_PAGE_SIZE,
      "should propery convert with page size flag"
    );

    assert_eq!(
      QueryFlags::WithPagingState.as_byte(),
      WITH_PAGING_STATE,
      "should propery convert with paging state flag"
    );

    assert_eq!(
      QueryFlags::WithSerialConsistency.as_byte(),
      WITH_SERIAL_CONSISTENCY,
      "should propery convert with serial consistency flag"
    );

    assert_eq!(
      QueryFlags::WithDefaultTimestamp.as_byte(),
      WITH_DEFAULT_TIMESTAMP,
      "should propery convert with default timestamp flag"
    );

    assert_eq!(
      QueryFlags::WithNamesForValues.as_byte(),
      WITH_NAME_FOR_VALUES,
      "should propery convert with name for values flag"
    );
  }
}