You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

100 lines
2.4 KiB

  1. package trust
  2. import (
  3. "testing"
  4. "time"
  5. "github.com/stretchr/testify/assert"
  6. )
  7. func TestTrustMetricScores(t *testing.T) {
  8. tm := NewMetric()
  9. tm.Start()
  10. // Perfect score
  11. tm.GoodEvents(1)
  12. score := tm.TrustScore()
  13. assert.Equal(t, 100, score)
  14. // Less than perfect score
  15. tm.BadEvents(10)
  16. score = tm.TrustScore()
  17. assert.NotEqual(t, 100, score)
  18. tm.Stop()
  19. }
  20. func TestTrustMetricConfig(t *testing.T) {
  21. // 7 days
  22. window := time.Minute * 60 * 24 * 7
  23. config := TrustMetricConfig{
  24. TrackingWindow: window,
  25. IntervalLength: 2 * time.Minute,
  26. }
  27. tm := NewMetricWithConfig(config)
  28. tm.Start()
  29. // The max time intervals should be the TrackingWindow / IntervalLen
  30. assert.Equal(t, int(config.TrackingWindow/config.IntervalLength), tm.maxIntervals)
  31. dc := DefaultConfig()
  32. // These weights should still be the default values
  33. assert.Equal(t, dc.ProportionalWeight, tm.proportionalWeight)
  34. assert.Equal(t, dc.IntegralWeight, tm.integralWeight)
  35. tm.Stop()
  36. tm.Wait()
  37. config.ProportionalWeight = 0.3
  38. config.IntegralWeight = 0.7
  39. tm = NewMetricWithConfig(config)
  40. tm.Start()
  41. // These weights should be equal to our custom values
  42. assert.Equal(t, config.ProportionalWeight, tm.proportionalWeight)
  43. assert.Equal(t, config.IntegralWeight, tm.integralWeight)
  44. tm.Stop()
  45. tm.Wait()
  46. }
  47. // XXX: This test fails non-deterministically
  48. func _TestTrustMetricStopPause(t *testing.T) {
  49. // The TestTicker will provide manual control over
  50. // the passing of time within the metric
  51. tt := NewTestTicker()
  52. tm := NewMetric()
  53. tm.SetTicker(tt)
  54. tm.Start()
  55. // Allow some time intervals to pass and pause
  56. tt.NextTick()
  57. tt.NextTick()
  58. tm.Pause()
  59. // could be 1 or 2 because Pause and NextTick race
  60. first := tm.Copy().numIntervals
  61. // Allow more time to pass and check the intervals are unchanged
  62. tt.NextTick()
  63. tt.NextTick()
  64. assert.Equal(t, first, tm.Copy().numIntervals)
  65. // Get the trust metric activated again
  66. tm.GoodEvents(5)
  67. // Allow some time intervals to pass and stop
  68. tt.NextTick()
  69. tt.NextTick()
  70. tm.Stop()
  71. tm.Wait()
  72. second := tm.Copy().numIntervals
  73. // Allow more intervals to pass while the metric is stopped
  74. // and check that the number of intervals match
  75. tm.NextTimeInterval()
  76. tm.NextTimeInterval()
  77. // XXX: fails non-deterministically:
  78. // expected 5, got 6
  79. assert.Equal(t, second+2, tm.Copy().numIntervals)
  80. if first > second {
  81. t.Fatalf("numIntervals should always increase or stay the same over time")
  82. }
  83. }