crosvm/sys_util/poll_token_derive/tests.rs

66 lines
1.9 KiB
Rust
Raw Normal View History

// Copyright 2018 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
use quote::quote;
use syn::{parse_quote, DeriveInput};
#[test]
fn test_variant_bits() {
let mut variants = vec![parse_quote!(A)];
assert_eq!(crate::variant_bits(&variants), 0);
variants.push(parse_quote!(B));
variants.push(parse_quote!(C));
assert_eq!(crate::variant_bits(&variants), 2);
for _ in 0..1021 {
variants.push(parse_quote!(Dynamic));
}
assert_eq!(crate::variant_bits(&variants), 10);
variants.push(parse_quote!(OneMore));
assert_eq!(crate::variant_bits(&variants), 11);
}
#[test]
fn poll_token_e2e() {
let input: DeriveInput = parse_quote! {
enum Token {
A,
B,
C,
D(usize),
E { foobaz: u32 },
}
};
let actual = crate::poll_token_inner(input);
let expected = quote! {
impl PollToken for Token {
fn as_raw_token(&self) -> u64 {
match *self {
Token::A => 0u64,
Token::B => 1u64,
Token::C => 2u64,
Token::D { 0: data } => 3u64 | ((data as u64) << 3u32),
Token::E { foobaz: data } => 4u64 | ((data as u64) << 3u32),
}
}
fn from_raw_token(data: u64) -> Self {
match data & 7u64 {
0u64 => Token::A,
1u64 => Token::B,
2u64 => Token::C,
3u64 => Token::D { 0: (data >> 3u32) as usize },
4u64 => Token::E { foobaz: (data >> 3u32) as u32 },
_ => unreachable!(),
}
}
}
};
assert_eq!(actual.to_string(), expected.to_string());
}