[testing] add TestDoImport

commands.go

@@ -1,7 +1,6 @@
 package main
 
 import (
-	"bufio"
 	"errors"
 	"fmt"
 	"os"
@@ -13,7 +12,6 @@ import (
 	"github.com/motemen/ghq/cmdutil"
 	"github.com/motemen/ghq/logger"
 	"github.com/urfave/cli"
-	"golang.org/x/sync/errgroup"
 	"golang.org/x/xerrors"
 )
 
@@ -331,46 +329,6 @@ func doLook(c *cli.Context) error {
 	}
 }
 
-func doImport(c *cli.Context) error {
-	var parallel = c.Bool("parallel")
-	g := &getter{
-		update:  c.Bool("update"),
-		shallow: c.Bool("shallow"),
-		ssh:     c.Bool("p"),
-		vcs:     c.String("vcs"),
-		silent:  c.Bool("silent"),
-	}
-	if parallel {
-		// force silent in parallel import
-		g.silent = true
-	}
-
-	eg := &errgroup.Group{}
-	sem := make(chan struct{}, 6)
-	scanner := bufio.NewScanner(os.Stdin)
-	for scanner.Scan() {
-		line := scanner.Text()
-		if parallel {
-			eg.Go(func() error {
-				sem <- struct{}{}
-				defer func() { <-sem }()
-				if err := g.get(line); err != nil {
-					logger.Log("error", err.Error())
-				}
-				return nil
-			})
-		} else {
-			if err := g.get(line); err != nil {
-				logger.Log("error", err.Error())
-			}
-		}
-	}
-	if err := scanner.Err(); err != nil {
-		return fmt.Errorf("While reading input: %s", err)
-	}
-	return eg.Wait()
-}
-
 func doRoot(c *cli.Context) error {
 	var (
 		w   = c.App.Writer

helpers_test.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"fmt"
 	"io/ioutil"
 	"net/url"
 	"os"
@@ -89,3 +90,19 @@ func capture(block func()) (string, string, error) {
 
 	return string(bufOut), string(bufErr), nil
 }
+
+func captureWithInput(in []string, block func()) (string, string, error) {
+	rIn, wIn, err := os.Pipe()
+	if err != nil {
+		return "", "", err
+	}
+	defer rIn.Close()
+	var stdin *os.File
+	os.Stdin, stdin = rIn, os.Stdin
+	defer func() { os.Stdin = stdin }()
+	for _, line := range in {
+		fmt.Fprintln(wIn, line)
+	}
+	wIn.Close()
+	return capture(block)
+}

import.go

@@ -0,0 +1,51 @@
+package main
+
+import (
+	"bufio"
+	"fmt"
+	"os"
+
+	"github.com/motemen/ghq/logger"
+	"github.com/urfave/cli"
+	"golang.org/x/sync/errgroup"
+)
+
+func doImport(c *cli.Context) error {
+	var parallel = c.Bool("parallel")
+	g := &getter{
+		update:  c.Bool("update"),
+		shallow: c.Bool("shallow"),
+		ssh:     c.Bool("p"),
+		vcs:     c.String("vcs"),
+		silent:  c.Bool("silent"),
+	}
+	if parallel {
+		// force silent in parallel import
+		g.silent = true
+	}
+
+	eg := &errgroup.Group{}
+	sem := make(chan struct{}, 6)
+	scanner := bufio.NewScanner(os.Stdin)
+	for scanner.Scan() {
+		line := scanner.Text()
+		if parallel {
+			eg.Go(func() error {
+				sem <- struct{}{}
+				defer func() { <-sem }()
+				if err := g.get(line); err != nil {
+					logger.Log("error", err.Error())
+				}
+				return nil
+			})
+		} else {
+			if err := g.get(line); err != nil {
+				logger.Log("error", err.Error())
+			}
+		}
+	}
+	if err := scanner.Err(); err != nil {
+		return fmt.Errorf("While reading input: %s", err)
+	}
+	return eg.Wait()
+}

logger/log.go

@@ -1,6 +1,7 @@
 package logger
 
 import (
+	"io"
 	"os"
 
 	"github.com/motemen/go-colorine"
@@ -28,7 +29,12 @@ var logger = colorine.NewLogger(
 	}, colorine.Info)
 
 func init() {
-	logger.SetOutput(os.Stderr)
+	SetOutput(os.Stderr)
+}
+
+// SetOutput sets log output writer
+func SetOutput(w io.Writer) {
+	logger.SetOutput(w)
 }
 
 // Log output