Load env in tests

Co-Authored-By: Liza Tsibur <[email protected]>

importer/parser_test.go

@@ -8,12 +8,16 @@ import (
 
 	"github.com/activecm/rita/v5/importer/zeektypes"
 	"github.com/activecm/rita/v5/util"
+	"github.com/joho/godotenv"
 
 	"github.com/spf13/afero"
 	"github.com/stretchr/testify/require"
 )
 
 func TestTruncatedTSV(t *testing.T) {
+	err := godotenv.Load("../.env")
+	require.NoError(t, err)
+
 	path := "../test_data/truncated/conn.log"
 	entries := make(chan zeektypes.Conn)
 	errc := make(chan error)
@@ -60,6 +64,9 @@ func TestTruncatedTSV(t *testing.T) {
 }
 
 func TestTruncatedHeader(t *testing.T) {
+	err := godotenv.Load("../.env")
+	require.NoError(t, err)
+
 	path := "../test_data/truncated/conn_top.log"
 	entries := make(chan zeektypes.Conn)
 	errc := make(chan error)
@@ -104,6 +111,9 @@ func TestTruncatedHeader(t *testing.T) {
 }
 
 func TestTruncatedJSON(t *testing.T) {
+	err := godotenv.Load("../.env")
+	require.NoError(t, err)
+
 	path := "../test_data/truncated/conn_json.log"
 	entries := make(chan zeektypes.Conn)
 	errc := make(chan error)
@@ -150,6 +160,9 @@ func TestTruncatedJSON(t *testing.T) {
 }
 
 func TestHasUnknownFieldTSV(t *testing.T) {
+	err := godotenv.Load("../.env")
+	require.NoError(t, err)
+
 	path := "../test_data/has_unknown_field/http.log"
 
 	entries := make(chan zeektypes.HTTP)
@@ -201,6 +214,9 @@ func TestHasUnknownFieldTSV(t *testing.T) {
 }
 
 func TestPlainTextFile(t *testing.T) {
+	err := godotenv.Load("../.env")
+	require.NoError(t, err)
+
 	path := "../test_data/text_file/conn.log"
 
 	entries := make(chan zeektypes.Conn)

logger/logger_test.go

@@ -4,10 +4,14 @@ import (
 	"sync"
 	"testing"
 
+	"github.com/joho/godotenv"
 	"github.com/stretchr/testify/require"
 )
 
 func TestLoggerNil(t *testing.T) {
+	err := godotenv.Load("../.env")
+	require.NoError(t, err)
+
 	logger := GetLogger()
 	require.NotNil(t, logger, "logger cannot be nil")