From 84e913106894a6ce5df09ed880b5ef70490c946a Mon Sep 17 00:00:00 2001
From: Daniel Milde <daniel@milde.cz>
Date: Sun, 24 Jan 2021 21:31:00 +0100
Subject: [PATCH] some more tests

---
 tui/tui_test.go | 32 ++++++++++++++++++++++++++++----
 1 file changed, 28 insertions(+), 4 deletions(-)

diff --git a/tui/tui_test.go b/tui/tui_test.go
index 79935edc8..db53bef43 100644
--- a/tui/tui_test.go
+++ b/tui/tui_test.go
@@ -113,6 +113,8 @@ func TestDeleteDir(t *testing.T) {
 		time.Sleep(10 * time.Millisecond)
 		simScreen.InjectKey(tcell.KeyRune, 'j', 1)
 		time.Sleep(10 * time.Millisecond)
+		simScreen.InjectKey(tcell.KeyRune, 'l', 1) // test selecting file
+		time.Sleep(10 * time.Millisecond)
 		simScreen.InjectKey(tcell.KeyRune, 'd', 1)
 		time.Sleep(10 * time.Millisecond)
 		simScreen.InjectKey(tcell.KeyRune, 'q', 1)
@@ -203,13 +205,26 @@ func TestShowConfirm(t *testing.T) {
 
 	go func() {
 		time.Sleep(100 * time.Millisecond)
+		simScreen.InjectKey(tcell.KeyRune, 'h', 1) // cannot go up
+		time.Sleep(10 * time.Millisecond)
 		simScreen.InjectKey(tcell.KeyRune, '?', 1)
 		time.Sleep(10 * time.Millisecond)
 		simScreen.InjectKey(tcell.KeyRune, 'q', 1)
 		time.Sleep(10 * time.Millisecond)
-		simScreen.InjectKey(tcell.KeyEnter, '1', 1)
+		simScreen.InjectKey(tcell.KeyRight, '1', 1)
+		time.Sleep(10 * time.Millisecond)
+		simScreen.InjectKey(tcell.KeyRight, '1', 1) // `..` cannot be selected by `l` or `right`
+		time.Sleep(10 * time.Millisecond)
+		simScreen.InjectKey(tcell.KeyRune, 'j', 1)
+		time.Sleep(10 * time.Millisecond)
+		simScreen.InjectKey(tcell.KeyRune, 'j', 1) // select file
+		time.Sleep(10 * time.Millisecond)
 		simScreen.InjectKey(tcell.KeyRune, 'd', 1)
 		time.Sleep(10 * time.Millisecond)
+		simScreen.InjectKey(tcell.KeyRune, 'h', 1) // cannot go up when confirm is shown
+		time.Sleep(10 * time.Millisecond)
+		simScreen.InjectKey(tcell.KeyRune, 'l', 1) // cannot go down when confirm is shown
+		time.Sleep(10 * time.Millisecond)
 		simScreen.InjectKey(tcell.KeyRune, 'q', 1)
 		time.Sleep(10 * time.Millisecond)
 	}()
@@ -309,6 +324,10 @@ func TestSelectDevice(t *testing.T) {
 	ui.ListDevices(getDevicesInfoMock)
 
 	go func() {
+		time.Sleep(100 * time.Millisecond)
+		simScreen.InjectKey(tcell.KeyRune, 'd', 1) // device cannot be deleted
+		time.Sleep(10 * time.Millisecond)
+		simScreen.InjectKey(tcell.KeyRune, 'r', 1) // or refreshed
 		time.Sleep(10 * time.Millisecond)
 		simScreen.InjectKey(tcell.KeyRune, 'l', 1)
 		time.Sleep(10 * time.Millisecond)
@@ -345,7 +364,7 @@ func TestKeys(t *testing.T) {
 		time.Sleep(10 * time.Millisecond)
 		simScreen.InjectKey(tcell.KeyRune, 'd', 1)
 		time.Sleep(10 * time.Millisecond)
-		simScreen.InjectKey(tcell.KeyEnter, 'h', 1)
+		simScreen.InjectKey(tcell.KeyRune, 'h', 1)
 		time.Sleep(10 * time.Millisecond)
 		simScreen.InjectKey(tcell.KeyRune, 'h', 1)
 		time.Sleep(10 * time.Millisecond)
@@ -404,7 +423,12 @@ func analyzeMock(path string, progress *analyze.CurrentProgress, ignore analyze.
 
 func getDevicesInfoMock(_ string) ([]*analyze.Device, error) {
 	item := &analyze.Device{
-		Name: "xxx",
+		Name:       "/dev/root",
+		MountPoint: "/",
+	}
+	item2 := &analyze.Device{
+		Name:       "/dev/boot",
+		MountPoint: "/boot",
 	}
-	return []*analyze.Device{item}, nil
+	return []*analyze.Device{item, item2}, nil
 }