From 3dcaefe35ba10a95e235a78ce67a3c2158994359 Mon Sep 17 00:00:00 2001
From: InsanePrawn <insane.prawny@gmail.com>
Date: Mon, 31 Oct 2022 16:35:25 +0100
Subject: [PATCH] devices/test_deviceinfo: add test for device variants
 ("kernels" in pmb)

---
 devices/test_deviceinfo.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/devices/test_deviceinfo.py b/devices/test_deviceinfo.py
index e4aebf4..fa2cfff 100644
--- a/devices/test_deviceinfo.py
+++ b/devices/test_deviceinfo.py
@@ -60,3 +60,14 @@ def test_get_deviceinfo_from_repo():
     assert dev
     info = dev.parse_deviceinfo()
     assert info
+
+
+def test_get_variant_deviceinfo_from_repo():
+    config.try_load_file()
+    dev = get_device('sdm845-xiaomi-beryllium-ebbg')
+    assert dev
+    info = dev.parse_deviceinfo()
+    assert info
+    assert 'dtb' in info  # variant-specific variable, check it has been stripped down from 'dtb_ebbg' to 'dtb'
+    assert 'dtb_tianma' not in info
+    assert info.dtb