[FIX] rule34.paheal.net: updatet detection if post is a video

gan-of-culture · Feb 2, 2023 · 1453a70 · 1453a70
1 parent cb77b98
commit 1453a70
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/extractors/rule34/rule34.go b/extractors/rule34/rule34.go
@@ -126,7 +126,7 @@ func extractData(URL string) (*static.Data, error) {
 	}
 
 	dataType := static.DataTypeImage
-	if strings.Contains(htmlString, "#Videomain") {
+	if strings.Contains(htmlString, "Videomain") {
 		dataType = static.DataTypeVideo
 	}
 

diff --git a/extractors/rule34/rule34_test.go b/extractors/rule34/rule34_test.go
@@ -109,7 +109,7 @@ func TestMassExtract(t *testing.T) {
 			config.Amount = 10
 			elements, err := New().Extract(tt.URL)
 			if err != nil {
-				t.Error("elements has error or is too big for single tests")
+				test.CheckError(t, err)
 			}
 			if len(elements) != tt.Want {
 				t.Errorf("Got: %v - Want: %v", len(elements), tt.Want)