test output

tillkuhn · Nov 18, 2024 · 4acf0a2 · 4acf0a2
1 parent 57e9ff4
commit 4acf0a2
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 0 deletions.
diff --git a/cmd/punch.go b/cmd/punch.go
@@ -24,6 +24,7 @@ var punchCmd = &cobra.Command{
 				return err
 			}
 		}
+		punchOpts.Out = rootCmd.OutOrStdout()
 		t := tracker.New(&punchOpts)
 		return t.PunchReport(cmd.Context())
 	},

diff --git a/cmd/punch_test.go b/cmd/punch_test.go
@@ -34,6 +34,16 @@ func TestSum(t *testing.T) {
 			"",
 			"cannot parse",
 		},
+		"ok-2-args": {
+			[]string{"4h5m", "2024-11-11"},
+			"Monday",
+			"",
+		},
+		"ok-1-arg": {
+			[]string{"4h5m"},
+			"Monday",
+			"",
+		},
 	}
 
 	for name, te := range tests {
@@ -45,6 +55,8 @@ func TestSum(t *testing.T) {
 			err := rootCmd.Execute()
 			if te.error != "" {
 				assert.ErrorContains(t, err, te.error)
+			} else {
+				assert.NoError(t, err)
 			}
 			assert.Contains(t, actual.String(), te.out)
 		})

diff --git a/pkg/tracker/db.go b/pkg/tracker/db.go
@@ -21,7 +21,11 @@ var initSQL string
 
 // initDB initializes SQLite DB in local filesystem
 func initDB(opts *Options) (*sqlx.DB, error) {
+	if err := os.MkdirAll(opts.AppDir(), os.ModePerm); err != nil {
+		return nil, fmt.Errorf("cannot create app dir %s: %w", opts.AppDir(), err)
+	}
 	dbFile := filepath.Join(opts.AppDir(), "db.sqlite3")
+
 	log.Printf("🥫 Open database file=%s sqlite=%s", dbFile, sqlite3.SQLITE_VERSION)
 	db, err := sqlx.Open("sqlite", dbFile)
 	if err != nil {