From 37faccb93a8e0a6c7beb46259e09fe556cd3b1ea Mon Sep 17 00:00:00 2001
From: Grégoire Duchêne <gduchene@awhk.org>
Date: Sun, 26 Jun 2022 09:34:51 +0100
Subject: Add FlagVarDef and FlagVarSliceDef

---
 flag_test.go | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

(limited to 'flag_test.go')

diff --git a/flag_test.go b/flag_test.go
index 56711d5..a4da69b 100644
--- a/flag_test.go
+++ b/flag_test.go
@@ -20,6 +20,16 @@ func TestFlagVar(s *testing.T) {
 	t.AssertEqual(true, *fl)
 }
 
+func TestFlagVarDef(s *testing.T) {
+	t := core.T{T: s}
+
+	fs := flag.NewFlagSet("", flag.ContinueOnError)
+	fl := core.FlagVarDef(fs, "test", "", strconv.Atoi, 42)
+	t.AssertEqual(42, *fl)
+	t.AssertErrorIs(nil, fs.Parse([]string{"-test=1"}))
+	t.AssertEqual(1, *fl)
+}
+
 func TestFlagVarPtr(s *testing.T) {
 	t := core.T{T: s}
 
@@ -39,6 +49,16 @@ func TestFlagVarSlice(s *testing.T) {
 	t.AssertEqual([]int{1, 2, 42}, *fl)
 }
 
+func TestFlagVarSliceDef(s *testing.T) {
+	t := core.T{T: s}
+
+	fs := flag.NewFlagSet("", flag.ContinueOnError)
+	fl := core.FlagVarSliceDef(fs, "test", "", strconv.Atoi, []int{42})
+	t.AssertEqual([]int{42}, *fl)
+	t.AssertErrorIs(nil, fs.Parse([]string{"-test=1", "-test=2"}))
+	t.AssertEqual([]int{1, 2}, *fl)
+}
+
 func TestFlagVarSlicePtr(s *testing.T) {
 	t := core.T{T: s}
 
-- 
cgit v1.2.3-70-g09d2