You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
1.9 KiB
66 lines
1.9 KiB
// 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());
|
|
}
|