Mirrors/servo
1
0
Fork 0
mirror of https://github.com/servo/servo.git synced 2025-08-06 14:10:11 +01:00
Code Issues Projects Releases Packages Wiki Activity

fixup! Issue #7382 Use descriptive enums instead of booleans for MIMEClassifier::classifer

Browse source
This commit is contained in:
Mathieu Rheaume 2015-09-12 16:56:55 -04:00
parent 5c807311de
commit ff608de781
2 changed files with 23 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

21
components/net/mime_classifier.rs
View file

@ -26,6 +26,7 @@ pub enum ApacheBugFlag {
OFF
}
#[derive(PartialEq)]
pub enum NoSniffFlag {
ON,
OFF
@ -81,14 +82,18 @@ impl MIMEClassifier {
//some sort of iterator over the classifiers might be better?
fn sniff_unknown_type(&self, no_sniff_flag: NoSniffFlag, data: &[u8]) ->
Option<(String, String)> {
match no_sniff_flag {
NoSniffFlag::OFF => self.scriptable_classifier.classify(data),
_ => None
}.or_else(|| self.plaintext_classifier.classify(data))
.or_else(|| self.image_classifier.classify(data))
.or_else(|| self.audio_video_classifier.classify(data))
.or_else(|| self.archive_classifier.classify(data))
.or_else(|| self.binary_or_plaintext.classify(data))
let should_sniff_scriptable = no_sniff_flag == NoSniffFlag::OFF;
let sniffed = if should_sniff_scriptable {
self.scriptable_classifier.classify(data)
} else {
None
};
sniffed.or_else(|| self.plaintext_classifier.classify(data))
.or_else(|| self.image_classifier.classify(data))
.or_else(|| self.audio_video_classifier.classify(data))
.or_else(|| self.archive_classifier.classify(data))
.or_else(|| self.binary_or_plaintext.classify(data))
}
fn sniff_text_or_data(&self, data: &[u8]) -> Option<(String, String)> {

10
tests/unit/net/mime_classifier.rs
View file

@ -529,3 +529,13 @@ fn test_sniff_octet_stream_apache_flag_on() {
ApacheBugFlag::ON);
}
#[test]
fn test_sniff_mp4_video_apache_flag_on() {
test_sniff_with_flags(&PathBuf::from("video/mp4/test.mp4"),
"video",
"mp4",
None,
NoSniffFlag::OFF,
ApacheBugFlag::ON);
}