Abstract worseConn comparisons and add tests

[btrtrc.git] / worse-conns_test.go

diff --git a/worse-conns_test.go b/worse-conns_test.go
new file mode 100644 (file)
index 0000000..39eecf7
--- /dev/null
+++ b/worse-conns_test.go
@@ -0,0 +1,30 @@
+package torrent
+
+import (
+       "testing"
+       "time"
+
+       qt "github.com/frankban/quicktest"
+)
+
+func TestWorseConnLastHelpful(t *testing.T) {
+       c := qt.New(t)
+       c.Check(worseConnInput{}.Less(worseConnInput{LastHelpful: time.Now()}), qt.IsTrue)
+       c.Check(worseConnInput{}.Less(worseConnInput{CompletedHandshake: time.Now()}), qt.IsTrue)
+       c.Check(worseConnInput{LastHelpful: time.Now()}.Less(worseConnInput{CompletedHandshake: time.Now()}), qt.IsFalse)
+       c.Check(worseConnInput{
+               LastHelpful: time.Now(),
+       }.Less(worseConnInput{
+               LastHelpful:        time.Now(),
+               CompletedHandshake: time.Now(),
+       }), qt.IsTrue)
+       now := time.Now()
+       c.Check(worseConnInput{
+               LastHelpful: now,
+       }.Less(worseConnInput{
+               LastHelpful:        now.Add(-time.Nanosecond),
+               CompletedHandshake: now,
+       }), qt.IsFalse)
+       c.Check(worseConnInput{}.Less(worseConnInput{Pointer: 1}), qt.IsTrue)
+       c.Check(worseConnInput{Pointer: 2}.Less(worseConnInput{Pointer: 1}), qt.IsFalse)
+}