From 8be5f93f5b2d4b6f438ca84094937a0f7101c59b Mon Sep 17 00:00:00 2001
From: T <t@tjp.lol>
Date: Sat, 2 Aug 2025 17:25:59 -0600
Subject: Initial commit of punchcard.

Contains working time tracking commands, and the stub of a command to
generate reports.
---
 internal/commands/test_utils.go | 48 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)
 create mode 100644 internal/commands/test_utils.go

(limited to 'internal/commands/test_utils.go')

diff --git a/internal/commands/test_utils.go b/internal/commands/test_utils.go
new file mode 100644
index 0000000..282e472
--- /dev/null
+++ b/internal/commands/test_utils.go
@@ -0,0 +1,48 @@
+package commands
+
+import (
+	"bytes"
+	"context"
+	"database/sql"
+	"testing"
+
+	punchctx "punchcard/internal/context"
+	"punchcard/internal/database"
+	"punchcard/internal/queries"
+)
+
+func setupTestDB(t *testing.T) (*queries.Queries, func()) {
+	db, err := sql.Open("sqlite", ":memory:")
+	if err != nil {
+		t.Fatalf("Failed to open in-memory sqlite db: %v", err)
+	}
+	if err := database.InitializeDB(db); err != nil {
+		t.Fatalf("Failed to initialize in-memory sqlite db: %v", err)
+	}
+	q := queries.New(db)
+
+	// Return cleanup function that restores environment immediately
+	cleanup := func() {
+		if err := q.DBTX().(*sql.DB).Close(); err != nil {
+			t.Logf("error closing database: %v", err)
+		}
+	}
+
+	return q, cleanup
+}
+
+func executeCommandWithDB(t *testing.T, q *queries.Queries, args ...string) (string, error) {
+	buf := new(bytes.Buffer)
+
+	// Create context with provided database
+	ctx := punchctx.WithDB(context.Background(), q)
+
+	// Use factory functions to create fresh command instances for each test
+	testRootCmd := NewRootCmd()
+	testRootCmd.SetOut(buf)
+	testRootCmd.SetErr(buf)
+	testRootCmd.SetArgs(args)
+
+	err := testRootCmd.ExecuteContext(ctx)
+	return buf.String(), err
+}
-- 
cgit v1.2.3