From b64dccb3247acea7537902281af07f498b0bf0fb Mon Sep 17 00:00:00 2001
From: Naomi Kramer <naomi@activecountermeasures.com>
Date: Wed, 10 Jul 2024 16:10:48 -0400
Subject: [PATCH] Load env in tests

Co-Authored-By: Liza Tsibur <liza@activecountermeasures.com>
---
 importer/parser_test.go | 16 ++++++++++++++++
 logger/logger_test.go   |  4 ++++
 2 files changed, 20 insertions(+)

diff --git a/importer/parser_test.go b/importer/parser_test.go
index 3070ed6..4c47c41 100644
--- a/importer/parser_test.go
+++ b/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)
diff --git a/logger/logger_test.go b/logger/logger_test.go
index 414768f..2cafbad 100644
--- a/logger/logger_test.go
+++ b/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")