From 4e7d7282c24c6f4ff67fae6ca535fb259c7f28a7 Mon Sep 17 00:00:00 2001
From: Alessio Greggi <ale_grey_91@hotmail.it>
Date: Mon, 27 Jan 2025 17:51:59 +0100
Subject: [PATCH] test(internal/elfreader): add unit tests

Signed-off-by: Alessio Greggi <ale_grey_91@hotmail.it>
---
 internal/elfreader/symbols_test.go | 99 ++++++++++++++++++++++++++++++
 1 file changed, 99 insertions(+)
 create mode 100644 internal/elfreader/symbols_test.go

diff --git a/internal/elfreader/symbols_test.go b/internal/elfreader/symbols_test.go
new file mode 100644
index 0000000..2a63956
--- /dev/null
+++ b/internal/elfreader/symbols_test.go
@@ -0,0 +1,99 @@
+package elfreader
+
+import (
+	"testing"
+)
+
+func Test_isGoroutine(t *testing.T) {
+	type args struct {
+		s string
+	}
+	tests := []struct {
+		name string
+		args args
+		want bool
+	}{
+		{
+			name: "is not goroutine",
+			args: args{
+				s: "github.com/alegrey91/harpoon/internal/archiver.Convert",
+			},
+			want: false,
+		},
+		{
+			name: "is goroutine",
+			args: args{
+				s: "debug/dwarf.(*Data).readType.func1",
+			},
+			want: true,
+		},
+		{
+			name: "is goroutine",
+			args: args{
+				s: "debug/dwarf.(*Data).readType.func2",
+			},
+			want: true,
+		},
+		{
+			name: "is goroutine",
+			args: args{
+				s: "os.ReadDir.func1",
+			},
+			want: true,
+		},
+		{
+			name: "is goroutine",
+			args: args{
+				s: "github.com/myuser/myproject/pkg/api.ForbiddenListSpec.ExactMatch.SearchStrings.func2",
+			},
+			want: true,
+		},
+		{
+			name: "is goroutine",
+			args: args{
+				s: "github.com/prometheus/client_golang/prometheus.goRuntimeMemStats.func21",
+			},
+			want: true,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := isGoroutine(tt.args.s); got != tt.want {
+				t.Errorf("isGoroutine() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}
+
+func Test_isTestFunction(t *testing.T) {
+	type args struct {
+		s string
+	}
+	tests := []struct {
+		name string
+		args args
+		want bool
+	}{
+		{
+			name: "is test function",
+			args: args{
+				s: "github.com/myuser/myproject/api/v1beta1.TestOwnerListSpec_FindOwner",
+			},
+			want: true,
+		},
+		{
+			name: "is not test function",
+			args: args{
+				s: "net/http.http2got1xxFuncForTests",
+			},
+			want: false,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := isTestFunction(tt.args.s); got != tt.want {
+				t.Errorf("isTestFunction() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}