From 005bce86625b8fffdb3657103ee5589cbb6eed87 Mon Sep 17 00:00:00 2001 From: Valery Piashchynski Date: Wed, 1 Dec 2021 18:35:28 +0300 Subject: improve codecov Signed-off-by: Valery Piashchynski --- bst/bst_test.go | 43 +++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 43 insertions(+) (limited to 'bst/bst_test.go') diff --git a/bst/bst_test.go b/bst/bst_test.go index 2271508c..2afbee10 100644 --- a/bst/bst_test.go +++ b/bst/bst_test.go @@ -40,6 +40,49 @@ func TestNewBST(t *testing.T) { assert.Len(t, exist3, 100) } +func TestBSTContains(t *testing.T) { + // create a new bst + g := NewBST() + + for i := 0; i < 100; i++ { + g.Insert(uuid.NewString(), "comments") + } + + for i := 0; i < 100; i++ { + g.Insert(uuid.NewString(), "comments2") + } + + for i := 0; i < 100; i++ { + g.Insert(uuid.NewString(), "comments3") + } + + exist := g.Contains("comments") + assert.True(t, exist) + + exist2 := g.Contains("comments2") + assert.True(t, exist2) + + exist3 := g.Contains("comments3") + assert.True(t, exist3) + + exist4 := g.Contains("comments5") + assert.False(t, exist4) + + exist5 := g.Contains("comments6") + assert.False(t, exist5) +} + +func TestBSTRemove(t *testing.T) { + // create a new bst + g := NewBST() + u := uuid.NewString() + g.Insert(u, "comments") + g.Remove(u, "comments") + + res := g.Contains("comments") + assert.False(t, res) +} + func BenchmarkGraph(b *testing.B) { g := NewBST() -- cgit v1.2.3