Add extra

nuclia · Aug 22, 2024 · 62dc811 · 62dc811
1 parent e679d7a
commit 62dc811
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/nucliadb/tests/standalone/integration/test_upload_download.py b/nucliadb/tests/standalone/integration/test_upload_download.py
@@ -412,7 +412,7 @@ def test_invalid_content_types(content_type):
         ("data.xml", "application/xml"),
         # Archive files
         ("archive.zip", "application/zip"),
-        ("fooo.rar", "application/x-rar-compressed"),
+        ("fooo.rar", ["application/x-rar-compressed", "application/vnd.rar"]),
         ("archive.tar", "application/x-tar"),
         ("archive.tar.gz", "application/x-tar"),
         # Invalid content types
@@ -422,4 +422,7 @@ def test_invalid_content_types(content_type):
     ],
 )
 def test_guess_content_type(filename, content_type):
-    assert content_types.guess(filename) == content_type
+    if isinstance(content_type, list):
+        assert content_types.guess(filename) in content_type
+    else:
+        assert content_types.guess(filename) == content_type
diff --git a/nucliadb_models/src/nucliadb_models/content_types.py b/nucliadb_models/src/nucliadb_models/content_types.py
@@ -31,10 +31,12 @@
 
 EXTRA_VALID_CONTENT_TYPES = {
     "application/font-woff",
+    "application/mp4",
     "application/toml",
     "application/vnd.jgraph.mxfile",
     "application/vnd.ms-excel.sheet.macroenabled.12",
     "application/vnd.ms-outlook",
+    "application/vnd.ms-word.document.macroenabled.12",
     "application/vnd.rar",
     "application/x-aportisdoc",
     "application/x-archive",
@@ -43,9 +45,11 @@
     "application/x-iwork-pages-sffpages",
     "application/x-mach-binary",
     "application/x-mobipocket-ebook",
+    "application/x-ms-shortcut",
     "application/x-msdownload",
     "application/x-ndjson",
     "application/x-openscad",
+    "application/x-sql",
     "application/x-zip-compressed",
     "application/zstd",
     "audio/vnd.dlna.adts",