crosvm/sys_util/poll_token_derive/tests.rs
David Tolnay fe3ef7d998 edition: Update absolute paths to 2018 style
This is an easy step toward adopting 2018 edition eventually, and will
make any future CL that sets `edition = "2018"` this much smaller.

The module system changes in Rust 2018 are described here:

https://doc.rust-lang.org/edition-guide/rust-2018/module-system/path-clarity.html

Generated by running:

    cargo fix --edition --all

in each workspace, followed by bin/fmt.

TEST=cargo check
TEST=cargo check --all-features
TEST=cargo check --target aarch64-unknown-linux-gnu

Change-Id: I000ab5e69d69aa222c272fae899464bbaf65f6d8
Reviewed-on: https://chromium-review.googlesource.com/1513054
Commit-Ready: ChromeOS CL Exonerator Bot <chromiumos-cl-exonerator@appspot.gserviceaccount.com>
Tested-by: David Tolnay <dtolnay@chromium.org>
Tested-by: kokoro <noreply+kokoro@google.com>
Reviewed-by: David Tolnay <dtolnay@chromium.org>
2019-03-13 21:05:03 -07:00

65 lines
1.9 KiB
Rust

// 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());
}