mirror of
https://gitee.com/johng/gf.git
synced 2024-11-30 03:07:45 +08:00
92 lines
2.6 KiB
Go
92 lines
2.6 KiB
Go
// Copyright GoFrame Author(https://goframe.org). All Rights Reserved.
|
|
//
|
|
// This Source Code Form is subject to the terms of the MIT License.
|
|
// If a copy of the MIT was not distributed with this file,
|
|
// You can obtain one at https://github.com/gogf/gf.
|
|
|
|
// go test *.go -bench=".*" -benchmem
|
|
|
|
package gcmd_test
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/gogf/gf/v2/os/gcmd"
|
|
"github.com/gogf/gf/v2/test/gtest"
|
|
)
|
|
|
|
func Test_Parse(t *testing.T) {
|
|
gtest.C(t, func(t *gtest.T) {
|
|
os.Args = []string{"gf", "--force", "remove", "-fq", "-p=www", "path", "-n", "root"}
|
|
p, err := gcmd.Parse(map[string]bool{
|
|
"n, name": true,
|
|
"p, prefix": true,
|
|
"f,force": false,
|
|
"q,quiet": false,
|
|
})
|
|
t.AssertNil(err)
|
|
t.Assert(len(p.GetArgAll()), 3)
|
|
t.Assert(p.GetArg(0), "gf")
|
|
t.Assert(p.GetArg(1), "remove")
|
|
t.Assert(p.GetArg(2), "path")
|
|
t.Assert(p.GetArg(2).String(), "path")
|
|
|
|
t.Assert(len(p.GetOptAll()), 8)
|
|
t.Assert(p.GetOpt("n"), "root")
|
|
t.Assert(p.GetOpt("name"), "root")
|
|
t.Assert(p.GetOpt("p"), "www")
|
|
t.Assert(p.GetOpt("prefix"), "www")
|
|
t.Assert(p.GetOpt("prefix").String(), "www")
|
|
|
|
t.Assert(p.GetOpt("n") != nil, true)
|
|
t.Assert(p.GetOpt("name") != nil, true)
|
|
t.Assert(p.GetOpt("p") != nil, true)
|
|
t.Assert(p.GetOpt("prefix") != nil, true)
|
|
t.Assert(p.GetOpt("f") != nil, true)
|
|
t.Assert(p.GetOpt("force") != nil, true)
|
|
t.Assert(p.GetOpt("q") != nil, true)
|
|
t.Assert(p.GetOpt("quiet") != nil, true)
|
|
t.Assert(p.GetOpt("none") != nil, false)
|
|
|
|
_, err = p.MarshalJSON()
|
|
t.AssertNil(err)
|
|
})
|
|
}
|
|
|
|
func Test_ParseArgs(t *testing.T) {
|
|
gtest.C(t, func(t *gtest.T) {
|
|
p, err := gcmd.ParseArgs(
|
|
[]string{"gf", "--force", "remove", "-fq", "-p=www", "path", "-n", "root"},
|
|
map[string]bool{
|
|
"n, name": true,
|
|
"p, prefix": true,
|
|
"f,force": false,
|
|
"q,quiet": false,
|
|
})
|
|
t.AssertNil(err)
|
|
t.Assert(len(p.GetArgAll()), 3)
|
|
t.Assert(p.GetArg(0), "gf")
|
|
t.Assert(p.GetArg(1), "remove")
|
|
t.Assert(p.GetArg(2), "path")
|
|
t.Assert(p.GetArg(2).String(), "path")
|
|
|
|
t.Assert(len(p.GetOptAll()), 8)
|
|
t.Assert(p.GetOpt("n"), "root")
|
|
t.Assert(p.GetOpt("name"), "root")
|
|
t.Assert(p.GetOpt("p"), "www")
|
|
t.Assert(p.GetOpt("prefix"), "www")
|
|
t.Assert(p.GetOpt("prefix").String(), "www")
|
|
|
|
t.Assert(p.GetOpt("n") != nil, true)
|
|
t.Assert(p.GetOpt("name") != nil, true)
|
|
t.Assert(p.GetOpt("p") != nil, true)
|
|
t.Assert(p.GetOpt("prefix") != nil, true)
|
|
t.Assert(p.GetOpt("f") != nil, true)
|
|
t.Assert(p.GetOpt("force") != nil, true)
|
|
t.Assert(p.GetOpt("q") != nil, true)
|
|
t.Assert(p.GetOpt("quiet") != nil, true)
|
|
t.Assert(p.GetOpt("none") != nil, false)
|
|
})
|
|
}
|