mirror of
https://github.com/miniflux/v2.git
synced 2024-09-12 05:57:57 +02:00
207 lines
4.7 KiB
Go
207 lines
4.7 KiB
Go
// Copyright 2018 Frédéric Guillot. All rights reserved.
|
|
// Use of this source code is governed by the Apache 2.0
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package model // import "miniflux.app/model"
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"miniflux.app/config"
|
|
"miniflux.app/http/client"
|
|
)
|
|
|
|
func TestFeedWithResponse(t *testing.T) {
|
|
response := &client.Response{ETag: "Some etag", LastModified: "Some date", EffectiveURL: "Some URL"}
|
|
|
|
feed := &Feed{}
|
|
feed.WithClientResponse(response)
|
|
|
|
if feed.EtagHeader != "Some etag" {
|
|
t.Fatal(`The ETag header should be set`)
|
|
}
|
|
|
|
if feed.LastModifiedHeader != "Some date" {
|
|
t.Fatal(`The LastModified header should be set`)
|
|
}
|
|
|
|
if feed.FeedURL != "Some URL" {
|
|
t.Fatal(`The Feed URL should be set`)
|
|
}
|
|
}
|
|
|
|
func TestFeedCategorySetter(t *testing.T) {
|
|
feed := &Feed{}
|
|
feed.WithCategoryID(int64(123))
|
|
|
|
if feed.Category == nil {
|
|
t.Fatal(`The category field should not be null`)
|
|
}
|
|
|
|
if feed.Category.ID != int64(123) {
|
|
t.Error(`The category ID must be set`)
|
|
}
|
|
}
|
|
|
|
func TestFeedBrowsingParams(t *testing.T) {
|
|
feed := &Feed{}
|
|
feed.WithBrowsingParameters(
|
|
true,
|
|
"Custom User Agent",
|
|
"Username",
|
|
"Secret",
|
|
"Scraper Rule",
|
|
"Rewrite Rule",
|
|
"Block Rule",
|
|
"Allow Rule",
|
|
true,
|
|
)
|
|
|
|
if !feed.Crawler {
|
|
t.Error(`The crawler must be activated`)
|
|
}
|
|
|
|
if feed.UserAgent != "Custom User Agent" {
|
|
t.Error(`The user agent must be set`)
|
|
}
|
|
|
|
if feed.Username != "Username" {
|
|
t.Error(`The username must be set`)
|
|
}
|
|
|
|
if feed.Password != "Secret" {
|
|
t.Error(`The password must be set`)
|
|
}
|
|
|
|
if feed.ScraperRules != "Scraper Rule" {
|
|
t.Errorf(`The scraper rules must be set`)
|
|
}
|
|
|
|
if feed.RewriteRules != "Rewrite Rule" {
|
|
t.Errorf(`The rewrite rules must be set`)
|
|
}
|
|
|
|
if feed.BlocklistRules != "Block Rule" {
|
|
t.Errorf(`The block list rules must be set`)
|
|
}
|
|
|
|
if feed.KeeplistRules != "Allow Rule" {
|
|
t.Errorf(`The keep list rules must be set`)
|
|
}
|
|
|
|
if !feed.FetchViaProxy {
|
|
t.Errorf(`The fetch via proxy is no set`)
|
|
}
|
|
}
|
|
|
|
func TestFeedErrorCounter(t *testing.T) {
|
|
feed := &Feed{}
|
|
feed.WithError("Some Error")
|
|
|
|
if feed.ParsingErrorMsg != "Some Error" {
|
|
t.Error(`The error message must be set`)
|
|
}
|
|
|
|
if feed.ParsingErrorCount != 1 {
|
|
t.Error(`The error counter must be set to 1`)
|
|
}
|
|
|
|
feed.ResetErrorCounter()
|
|
|
|
if feed.ParsingErrorMsg != "" {
|
|
t.Error(`The error message must be removed`)
|
|
}
|
|
|
|
if feed.ParsingErrorCount != 0 {
|
|
t.Error(`The error counter must be set to 0`)
|
|
}
|
|
}
|
|
|
|
func TestFeedCheckedNow(t *testing.T) {
|
|
feed := &Feed{}
|
|
feed.FeedURL = "https://example.org/feed"
|
|
feed.CheckedNow()
|
|
|
|
if feed.SiteURL != feed.FeedURL {
|
|
t.Error(`The site URL must not be empty`)
|
|
}
|
|
|
|
if feed.CheckedAt.IsZero() {
|
|
t.Error(`The checked date must be set`)
|
|
}
|
|
}
|
|
|
|
func TestFeedScheduleNextCheckDefault(t *testing.T) {
|
|
var err error
|
|
parser := config.NewParser()
|
|
config.Opts, err = parser.ParseEnvironmentVariables()
|
|
if err != nil {
|
|
t.Fatalf(`Parsing failure: %v`, err)
|
|
}
|
|
|
|
feed := &Feed{}
|
|
weeklyCount := 10
|
|
feed.ScheduleNextCheck(weeklyCount)
|
|
|
|
if feed.NextCheckAt.IsZero() {
|
|
t.Error(`The next_check_at must be set`)
|
|
}
|
|
}
|
|
|
|
func TestFeedScheduleNextCheckEntryCountBasedMaxInterval(t *testing.T) {
|
|
maxInterval := 5
|
|
minInterval := 1
|
|
os.Clearenv()
|
|
os.Setenv("POLLING_SCHEDULER", "entry_frequency")
|
|
os.Setenv("SCHEDULER_ENTRY_FREQUENCY_MAX_INTERVAL", fmt.Sprintf("%d", maxInterval))
|
|
os.Setenv("SCHEDULER_ENTRY_FREQUENCY_MIN_INTERVAL", fmt.Sprintf("%d", minInterval))
|
|
|
|
var err error
|
|
parser := config.NewParser()
|
|
config.Opts, err = parser.ParseEnvironmentVariables()
|
|
if err != nil {
|
|
t.Fatalf(`Parsing failure: %v`, err)
|
|
}
|
|
feed := &Feed{}
|
|
weeklyCount := maxInterval * 100
|
|
feed.ScheduleNextCheck(weeklyCount)
|
|
|
|
if feed.NextCheckAt.IsZero() {
|
|
t.Error(`The next_check_at must be set`)
|
|
}
|
|
|
|
if feed.NextCheckAt.After(time.Now().Add(time.Minute * time.Duration(maxInterval))) {
|
|
t.Error(`The next_check_at should not be after the now + max interval`)
|
|
}
|
|
}
|
|
|
|
func TestFeedScheduleNextCheckEntryCountBasedMinInterval(t *testing.T) {
|
|
maxInterval := 500
|
|
minInterval := 100
|
|
os.Clearenv()
|
|
os.Setenv("POLLING_SCHEDULER", "entry_frequency")
|
|
os.Setenv("SCHEDULER_ENTRY_FREQUENCY_MAX_INTERVAL", fmt.Sprintf("%d", maxInterval))
|
|
os.Setenv("SCHEDULER_ENTRY_FREQUENCY_MIN_INTERVAL", fmt.Sprintf("%d", minInterval))
|
|
|
|
var err error
|
|
parser := config.NewParser()
|
|
config.Opts, err = parser.ParseEnvironmentVariables()
|
|
if err != nil {
|
|
t.Fatalf(`Parsing failure: %v`, err)
|
|
}
|
|
feed := &Feed{}
|
|
weeklyCount := minInterval / 2
|
|
feed.ScheduleNextCheck(weeklyCount)
|
|
|
|
if feed.NextCheckAt.IsZero() {
|
|
t.Error(`The next_check_at must be set`)
|
|
}
|
|
|
|
if feed.NextCheckAt.Before(time.Now().Add(time.Minute * time.Duration(minInterval))) {
|
|
t.Error(`The next_check_at should not be before the now + min interval`)
|
|
}
|
|
}
|