Fix bug in test.

jlewi · Sep 20, 2024 · acf5f50 · acf5f50
1 parent 84f5f37
commit acf5f50
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/app/pkg/analyze/session_builder_test.go b/app/pkg/analyze/session_builder_test.go
@@ -3,6 +3,8 @@ package analyze
 import (
 	"context"
 	"database/sql"
+	"os"
+	"path/filepath"
 	"testing"
 	"time"
 
@@ -40,7 +42,14 @@ func setup() (testTuple, error) {
 		return testTuple{}, errors.Wrapf(err, "failed to create config")
 	}
 
-	db, err := sql.Open(SQLLiteDriver, cfg.GetSessionsDB())
+	// If the directory doesn't exit opening the SQLLite database will fail.
+	sessionsDBFile := cfg.GetSessionsDB()
+	dbDir := filepath.Dir(sessionsDBFile)
+	if err := os.MkdirAll(dbDir, 0755); err != nil {
+		return testTuple{}, errors.Wrapf(err, "Failed to create directory: %v", dbDir)
+	}
+
+	db, err := sql.Open(SQLLiteDriver, sessionsDBFile)
 	if err != nil {
 		return testTuple{}, errors.Wrapf(err, "Failed to open database")
 	}

diff --git a/app/pkg/application/app.go b/app/pkg/application/app.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"database/sql"
 	"fmt"
+	"github.com/jlewi/monogo/helpers"
 	"io"
 	"net/http"
 	"os"
@@ -380,7 +381,13 @@ func (a *App) SetupAnalyzer() (*analyze.Analyzer, error) {
 		return nil, errors.New("Config is nil; call LoadConfig first")
 	}
 
-	db, err := sql.Open(analyze.SQLLiteDriver, a.Config.GetSessionsDB())
+	// If the directory doesn't exit opening the SQLLite database will fail.
+	sessionsDBFile := a.Config.GetSessionsDB()
+	dbDir := filepath.Dir(sessionsDBFile)
+	if err := os.MkdirAll(dbDir, helpers.UserGroupAllPerm); err != nil {
+		return nil, errors.Wrapf(err, "Failed to create directory: %v", dbDir)
+	}
+	db, err := sql.Open(analyze.SQLLiteDriver, sessionsDBFile)
 
 	if err != nil {
 		return nil, errors.Wrapf(err, "Failed to open database: %v", a.Config.GetSessionsDB())