From 0342a80d26ccfab37deb0de6b3fe9f25ceb55c84 Mon Sep 17 00:00:00 2001 From: suzaku Date: Fri, 4 Dec 2020 14:48:19 +0800 Subject: [PATCH] More unit test --- entry_test.go | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) diff --git a/entry_test.go b/entry_test.go index 4e562e4..61b1ea4 100644 --- a/entry_test.go +++ b/entry_test.go @@ -130,3 +130,26 @@ func TestUpdateEntryScore(t *testing.T) { t.Errorf("Entity score is wrong: %f", e.score) } } + +func TestLoadEntries(t *testing.T) { + t.Run("Should make sure the entries are sorted by score", func(t *testing.T) { + f, err := ioutil.TempFile("", "entries") + if err != nil { + t.Fatal(err) + } + content := []byte("20\t/a\n22\t/a/b\n12.5\t/c\n") + err = ioutil.WriteFile(f.Name(), content, 0666) + if err != nil { + t.Fatal(err) + } + entries := loadEntries(f.Name()) + if len(entries) != 3 { + t.Errorf("Expect 3 entries, got: %v ", entries) + } + paths := make([]string, len(entries)) + for i, e := range entries { + paths[i] = e.val + } + assertItemsEqual(t, paths, []string{"/a/b", "/a", "/c"}) + }) +}