feed_test.go 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163
  1. // SPDX-FileCopyrightText: Copyright The Miniflux Authors. All rights reserved.
  2. // SPDX-License-Identifier: Apache-2.0
  3. package model // import "miniflux.app/v2/internal/model"
  4. import (
  5. "fmt"
  6. "os"
  7. "testing"
  8. "time"
  9. "miniflux.app/v2/internal/config"
  10. )
  11. func TestFeedCategorySetter(t *testing.T) {
  12. feed := &Feed{}
  13. feed.WithCategoryID(int64(123))
  14. if feed.Category == nil {
  15. t.Fatal(`The category field should not be null`)
  16. }
  17. if feed.Category.ID != int64(123) {
  18. t.Error(`The category ID must be set`)
  19. }
  20. }
  21. func TestFeedErrorCounter(t *testing.T) {
  22. feed := &Feed{}
  23. feed.WithTranslatedErrorMessage("Some Error")
  24. if feed.ParsingErrorMsg != "Some Error" {
  25. t.Error(`The error message must be set`)
  26. }
  27. if feed.ParsingErrorCount != 1 {
  28. t.Error(`The error counter must be set to 1`)
  29. }
  30. feed.ResetErrorCounter()
  31. if feed.ParsingErrorMsg != "" {
  32. t.Error(`The error message must be removed`)
  33. }
  34. if feed.ParsingErrorCount != 0 {
  35. t.Error(`The error counter must be set to 0`)
  36. }
  37. }
  38. func TestFeedCheckedNow(t *testing.T) {
  39. feed := &Feed{}
  40. feed.FeedURL = "https://example.org/feed"
  41. feed.CheckedNow()
  42. if feed.SiteURL != feed.FeedURL {
  43. t.Error(`The site URL must not be empty`)
  44. }
  45. if feed.CheckedAt.IsZero() {
  46. t.Error(`The checked date must be set`)
  47. }
  48. }
  49. func TestFeedScheduleNextCheckDefault(t *testing.T) {
  50. var err error
  51. parser := config.NewParser()
  52. config.Opts, err = parser.ParseEnvironmentVariables()
  53. if err != nil {
  54. t.Fatalf(`Parsing failure: %v`, err)
  55. }
  56. feed := &Feed{}
  57. weeklyCount := 10
  58. feed.ScheduleNextCheck(weeklyCount)
  59. if feed.NextCheckAt.IsZero() {
  60. t.Error(`The next_check_at must be set`)
  61. }
  62. if feed.NextCheckAt.After(time.Now().Add(time.Minute * time.Duration(config.Opts.PollingFrequency()))) {
  63. t.Error(`The next_check_at should not be after the now + polling frequency`)
  64. }
  65. }
  66. func TestFeedScheduleNextCheckEntryCountBasedMaxInterval(t *testing.T) {
  67. maxInterval := 5
  68. minInterval := 1
  69. os.Clearenv()
  70. os.Setenv("POLLING_SCHEDULER", "entry_frequency")
  71. os.Setenv("SCHEDULER_ENTRY_FREQUENCY_MAX_INTERVAL", fmt.Sprintf("%d", maxInterval))
  72. os.Setenv("SCHEDULER_ENTRY_FREQUENCY_MIN_INTERVAL", fmt.Sprintf("%d", minInterval))
  73. var err error
  74. parser := config.NewParser()
  75. config.Opts, err = parser.ParseEnvironmentVariables()
  76. if err != nil {
  77. t.Fatalf(`Parsing failure: %v`, err)
  78. }
  79. feed := &Feed{}
  80. weeklyCount := maxInterval * 100
  81. feed.ScheduleNextCheck(weeklyCount)
  82. if feed.NextCheckAt.IsZero() {
  83. t.Error(`The next_check_at must be set`)
  84. }
  85. if feed.NextCheckAt.After(time.Now().Add(time.Minute * time.Duration(maxInterval))) {
  86. t.Error(`The next_check_at should not be after the now + max interval`)
  87. }
  88. }
  89. func TestFeedScheduleNextCheckEntryCountBasedMinInterval(t *testing.T) {
  90. maxInterval := 500
  91. minInterval := 100
  92. os.Clearenv()
  93. os.Setenv("POLLING_SCHEDULER", "entry_frequency")
  94. os.Setenv("SCHEDULER_ENTRY_FREQUENCY_MAX_INTERVAL", fmt.Sprintf("%d", maxInterval))
  95. os.Setenv("SCHEDULER_ENTRY_FREQUENCY_MIN_INTERVAL", fmt.Sprintf("%d", minInterval))
  96. var err error
  97. parser := config.NewParser()
  98. config.Opts, err = parser.ParseEnvironmentVariables()
  99. if err != nil {
  100. t.Fatalf(`Parsing failure: %v`, err)
  101. }
  102. feed := &Feed{}
  103. weeklyCount := minInterval / 2
  104. feed.ScheduleNextCheck(weeklyCount)
  105. if feed.NextCheckAt.IsZero() {
  106. t.Error(`The next_check_at must be set`)
  107. }
  108. if feed.NextCheckAt.Before(time.Now().Add(time.Minute * time.Duration(minInterval))) {
  109. t.Error(`The next_check_at should not be before the now + min interval`)
  110. }
  111. }
  112. func TestFeedScheduleNextCheckEntryFrequencyFactor(t *testing.T) {
  113. factor := 2
  114. os.Clearenv()
  115. os.Setenv("POLLING_SCHEDULER", "entry_frequency")
  116. os.Setenv("SCHEDULER_ENTRY_FREQUENCY_FACTOR", fmt.Sprintf("%d", factor))
  117. var err error
  118. parser := config.NewParser()
  119. config.Opts, err = parser.ParseEnvironmentVariables()
  120. if err != nil {
  121. t.Fatalf(`Parsing failure: %v`, err)
  122. }
  123. feed := &Feed{}
  124. weeklyCount := 7
  125. feed.ScheduleNextCheck(weeklyCount)
  126. if feed.NextCheckAt.IsZero() {
  127. t.Error(`The next_check_at must be set`)
  128. }
  129. if feed.NextCheckAt.After(time.Now().Add(time.Minute * time.Duration(config.Opts.SchedulerEntryFrequencyMaxInterval()/factor))) {
  130. t.Error(`The next_check_at should not be after the now + factor * count`)
  131. }
  132. }