♻️ test: implement Cmd interface

budougumi0617 · Sep 28, 2020 · 4e81999 · 4e81999
1 parent 3c3318e
commit 4e81999
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 2 deletions.
diff --git a/generate.go b/generate.go
@@ -39,7 +39,7 @@ var _ Cmd = &GenerateCmd{}
 type GenerateCmd struct{}
 
 func (g *GenerateCmd) MaxArg() int {
-	return 2
+	return 1
 }
 
 func (g *GenerateCmd) Usage() string {

diff --git a/test.go b/test.go
@@ -5,9 +5,26 @@ import (
 	"fmt"
 	"io/ioutil"
 	"os"
+	"strconv"
 )
 
-func TestCmd(ctx context.Context, id int) error {
+var _ Cmd = &TestCmd{}
+
+type TestCmd struct{}
+
+func (c *TestCmd) MaxArg() int {
+	return 1
+}
+
+func (c *TestCmd) Usage() string {
+	return "submit codes and "
+}
+
+func (c *TestCmd) Run(ctx context.Context, args []string) error {
+	id, err := strconv.Atoi(args[0])
+	if err != nil {
+		return err
+	}
 	session := os.Getenv("LEETCODE_SESSION")
 	token := os.Getenv("LEETCODE_TOKEN")