2 files changed, 34 insertions, 0 deletions
diff --git a/third_party/rust/cpufeatures/tests/aarch64.rs b/third_party/rust/cpufeatures/tests/aarch64.rs
new file mode 100644
index 0000000000..949669bed0
--- /dev/null
+++ b/third_party/rust/cpufeatures/tests/aarch64.rs
@@ -0,0 +1,17 @@
+//! ARM64 tests
+
+#![cfg(target_arch = "aarch64")]
+
+cpufeatures::new!(armcaps, "aes", "sha2", "sha3");
+
+#[test]
+fn init() {
+    let token: armcaps::InitToken = armcaps::init();
+    assert_eq!(token.get(), armcaps::get());
+}
+
+#[test]
+fn init_get() {
+    let (token, val) = armcaps::init_get();
+    assert_eq!(val, token.get());
+}
diff --git a/third_party/rust/cpufeatures/tests/x86.rs b/third_party/rust/cpufeatures/tests/x86.rs
new file mode 100644
index 0000000000..0d81242389
--- /dev/null
+++ b/third_party/rust/cpufeatures/tests/x86.rs
@@ -0,0 +1,17 @@
+//! x86/x86_64 tests
+
+#![cfg(any(target_arch = "x86", target_arch = "x86_64"))]
+
+cpufeatures::new!(cpuid, "aes", "sha");
+
+#[test]
+fn init() {
+    let token: cpuid::InitToken = cpuid::init();
+    assert_eq!(token.get(), cpuid::get());
+}
+
+#[test]
+fn init_get() {
+    let (token, val) = cpuid::init_get();
+    assert_eq!(val, token.get());
+}