Update tests to account for PIE files

CybercentreCanada · Feb 10, 2024 · ad28d15 · ad28d15
1 parent 15dcdcf
commit ad28d15
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/test_cuckoo_main.py b/tests/test_cuckoo_main.py
@@ -262,9 +262,11 @@ def test_image_tag_constants():
 
     @staticmethod
     def test_file_constants():
-        assert set(LINUX_x86_FILES) == {"executable/linux/elf32", "executable/linux/so32", "executable/linux/coff32"}
-        assert set(LINUX_x64_FILES) == {"executable/linux/elf64", "executable/linux/so64",
-                                        "executable/linux/ia/coff64", "executable/linux/coff64", "code/shell"}
+        assert set(LINUX_x86_FILES) == {"executable/linux/elf32", "executable/linux/so32",
+                                        "executable/linux/coff32", "executable/linux/pie32"}
+        assert set(LINUX_x64_FILES) == {
+            "executable/linux/elf64", "executable/linux/so64", "executable/linux/ia/coff64", "executable/linux/coff64",
+            "code/shell", "executable/linux/pie64"}
         assert set(WINDOWS_x86_FILES) == {'executable/windows/pe32', 'executable/windows/dll32'}
 
     @staticmethod