fixed mocking in tests

doc_test.go

@@ -7,7 +7,7 @@ import (
 	"atomicgo.dev/keyboard/keys"
 )
 
-func ExampleSimple() {
+func ExampleListen_simple() {
 	Listen(func(key keys.Key) (stop bool, err error) {
 		if key.Code == keys.CtrlC {
 			return true, nil // Stop listener by returning true on Ctrl+C
@@ -18,7 +18,7 @@ func ExampleSimple() {
 	})
 }
 
-func ExampleAdvanced() {
+func ExampleListen_advanced() {
 	// Stop keyboard listener on Escape key press or CTRL+C.
 	// Exit application on "q" key press.
 	// Print every rune key press.
@@ -41,7 +41,7 @@ func ExampleAdvanced() {
 	})
 }
 
-func ExampleMocking() {
+func ExampleSimulateKeyPress() {
 	go func() {
 		SimulateKeyPress("Hello")             // Simulate key press for every letter in string
 		SimulateKeyPress(keys.Enter)          // Simulate key press for Enter

input.go

@@ -167,7 +167,7 @@ func getKeyPress(input io.Reader) (keys.Key, error) {
 	// Read
 	numBytes, err := input.Read(buf[:])
 	if err != nil {
-		return keys.Key{}, fmt.Errorf("could not read input: %w", err)
+		return keys.Key{}, fmt.Errorf("could not read stdin: %w", err)
 	}
 
 	// Check if it's a sequence
@@ -194,7 +194,7 @@ func getKeyPress(input io.Reader) (keys.Key, error) {
 	var runes []rune
 	b := buf[:numBytes]
 
-	// Translate input into runes.
+	// Translate stdin into runes.
 	for i, w := 0, 0; i < len(b); i += w { //nolint:wastedassign
 		r, width := utf8.DecodeRune(b[i:])
 		if r == utf8.RuneError {
@@ -205,7 +205,7 @@ func getKeyPress(input io.Reader) (keys.Key, error) {
 	}
 
 	if len(runes) == 0 {
-		return keys.Key{}, fmt.Errorf("received 0 runes from input")
+		return keys.Key{}, fmt.Errorf("received 0 runes from stdin")
 	} else if len(runes) > 1 {
 		return keys.Key{Code: keys.RuneKey, Runes: runes}, nil
 	}

keyboard.go

@@ -11,16 +11,22 @@ import (
 
 var windowsStdin *os.File
 var con console.Console
-var input = os.Stdin
+var stdin = os.Stdin
 var inputTTY *os.File
 var mockChannel = make(chan keys.Key)
 
+var mocking = false
+
 func startListener() error {
 	err := initInput()
 	if err != nil {
 		return err
 	}
 
+	if mocking {
+		return nil
+	}
+
 	if con != nil {
 		err := con.SetRaw()
 		if err != nil {
@@ -106,7 +112,7 @@ func Listen(onKeyPress func(key keys.Key) (stop bool, err error)) error {
 	return nil
 }
 
-// SimulateKeyPress simulate a key press. It can be used to mock user input and test your application.
+// SimulateKeyPress simulate a key press. It can be used to mock user stdin and test your application.
 //
 // Example:
 //  go func() {

keyboard_test.go

@@ -0,0 +1,31 @@
+package keyboard_test
+
+import (
+	"sync"
+	"testing"
+
+	"atomicgo.dev/keyboard"
+	"atomicgo.dev/keyboard/keys"
+)
+
+var wg sync.WaitGroup
+
+func TestMocking(t *testing.T) {
+	wg.Add(1)
+
+	keyboard.Listen(func(key keys.Key) (stop bool, err error) {
+		if key.Code != keys.Down {
+			t.Errorf("down key should be pressed, but %s is pressed", key)
+		}
+
+		wg.Done()
+
+		return true, nil
+	})
+
+	go func() {
+		keyboard.SimulateKeyPress(keys.Down)
+	}()
+
+	wg.Wait()
+}

tty_unix.go

@@ -17,7 +17,7 @@ func restoreInput() error {
 }
 
 func initInput() error {
-	c, err := console.ConsoleFromFile(input)
+	c, err := console.ConsoleFromFile(stdin)
 	if err != nil {
 		return err
 	}

tty_windows.go

@@ -6,6 +6,7 @@ package keyboard
 import (
 	"fmt"
 	"os"
+	"syscall"
 
 	"github.com/containerd/console"
 )
@@ -20,7 +21,16 @@ func restoreInput() error {
 
 func initInput() error {
 	windowsStdin = os.Stdin
-	os.Stdin = input
+
+	os.Stdin = stdin
+
+	var mode uint32
+	err := syscall.GetConsoleMode(syscall.Stdin, &mode)
+
+	if err != nil {
+		mocking = true
+		return nil
+	}
 
 	con = console.Current()
 
@@ -30,7 +40,7 @@ func initInput() error {
 func openInputTTY() (*os.File, error) {
 	f, err := os.OpenFile("CONIN$", os.O_RDWR, 0644)
 	if err != nil {
-		return nil, fmt.Errorf("failed to open input TTY: %w", err)
+		return nil, fmt.Errorf("failed to open stdin TTY: %w", err)
 	}
 
 	return f, nil