From ee48a63e449ed2a7caff591b8b6556d34998ebc6 Mon Sep 17 00:00:00 2001
From: caffix <caffix@users.noreply.github.com>
Date: Sat, 9 Nov 2024 20:47:29 +0000
Subject: [PATCH] unit test fixes

---
 repository/tag_test.go | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/repository/tag_test.go b/repository/tag_test.go
index 8ffcf7e..5df4828 100644
--- a/repository/tag_test.go
+++ b/repository/tag_test.go
@@ -77,9 +77,8 @@ func TestEntityTag(t *testing.T) {
 	err = store.DeleteEntityTag(ct3.ID)
 	assert.NoError(t, err)
 
-	tag, err = store.FindEntityTagById(ct3.ID, time.Time{})
+	_, err = store.FindEntityTagById(ct3.ID, time.Time{})
 	assert.Error(t, err)
-	assert.Equal(t, tag, nil)
 }
 
 func TestCreateEdgeTag(t *testing.T) {
@@ -91,7 +90,7 @@ func TestCreateEdgeTag(t *testing.T) {
 
 	edge, err := store.Link(&types.Edge{
 		Relation: &relation.BasicDNSRelation{
-			Name:   "owasp.org",
+			Name:   "dns_record",
 			Header: relation.RRHeader{RRType: 5},
 		},
 		FromEntity: e1,
@@ -156,7 +155,6 @@ func TestCreateEdgeTag(t *testing.T) {
 	err = store.DeleteEdgeTag(ct3.ID)
 	assert.NoError(t, err)
 
-	tag, err = store.FindEdgeTagById(ct3.ID, time.Time{})
+	_, err = store.FindEdgeTagById(ct3.ID, time.Time{})
 	assert.Error(t, err)
-	assert.Equal(t, tag, nil)
 }