menu_test.go 5.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231
  1. package door
  2. import (
  3. "flag"
  4. "fmt"
  5. "io/ioutil"
  6. "net"
  7. "os"
  8. "strings"
  9. "testing"
  10. "time"
  11. )
  12. func TestMenuRender(t *testing.T) {
  13. bracket := ColorText("BLUE")
  14. option := ColorText("BRI GREEN")
  15. upper := ColorText("CYAN")
  16. lower := ColorText("MAGENTA")
  17. render := MakeMenuRender(bracket, option, upper, lower)
  18. // Fake menu line
  19. input := "[X] BUGZ test"
  20. got := render(input)
  21. expected := bracket + "[" + option + "X" + bracket + "]"
  22. expected += lower + " " + upper + "BUGZ" + lower + " test"
  23. if got != expected {
  24. t.Errorf("MenuRender expected %#v, got %#v", expected, got)
  25. }
  26. }
  27. func TestMenuConnection(t *testing.T) {
  28. tmpFile, err := ioutil.TempFile(os.TempDir(), "test-")
  29. if err != nil {
  30. panic("Cannot create temporary file")
  31. }
  32. // Remember to clean up the file afterwards
  33. defer os.Remove(tmpFile.Name())
  34. // establish network socket connection to set Comm_handle
  35. server, client := setupSockets()
  36. // We're not testing closed connections, so:
  37. defer server.Close()
  38. defer client.Close()
  39. // unicode 90x40 response
  40. buffer := []byte("\x1b[1;1R\x1b[2;3R\x1b[40;90R")
  41. server.Write(buffer)
  42. // Access Fd (File descriptor) of client for dropfile
  43. client_conn := client.(*net.TCPConn)
  44. client_file, err := client_conn.File()
  45. var fd int = int(client_file.Fd())
  46. // Create door32.sys file
  47. dfc := DropfileConfig{2, fd, 1800, "Test BBSID", 1701, "Real Username", "Handle", 880, 28, 0, 12}
  48. tmpFile.WriteString(fmt.Sprintf("%d\n%d\n%d\n%s\n%d\n%s\n%s\n%d\n%d\n%d\n%d\n",
  49. dfc.Comm_type, dfc.Comm_handle, dfc.Baudrate, dfc.BBSID, dfc.User_number, dfc.Real_name, dfc.Handle,
  50. dfc.Security_level, dfc.Time_left, dfc.Emulation, dfc.Node))
  51. tmpFile.Close()
  52. d := Door{}
  53. // If I call d.Init() more then once flag complains about flag redefined.
  54. // Reset flags
  55. flag.CommandLine = flag.NewFlagSet(os.Args[0], flag.ExitOnError)
  56. // preset commandline args so door can init
  57. os.Args = []string{"door", "-d", tmpFile.Name()}
  58. d.Init("menu-test")
  59. // clean up log file - not needed in tests
  60. defer os.Remove("menu-test-12.log")
  61. // Ok!
  62. if !Unicode {
  63. t.Errorf("Unicode not true %t", Unicode)
  64. }
  65. if Width != 90 {
  66. t.Errorf("Width not 90: %d", Width)
  67. }
  68. if Height != 40 {
  69. t.Errorf("Height not 40: %d", Height)
  70. }
  71. // These are the commands sent to detect ... throw this all away.
  72. buffer = make([]byte, 1024)
  73. server.SetReadDeadline(time.Now().Add(time.Millisecond * 20))
  74. _, err = server.Read(buffer)
  75. // t.Errorf("Buffer : %#v\n", buffer[:r])
  76. server.SetReadDeadline(time.Time{})
  77. // Ready to test!
  78. m := Menu{Panel: Panel{Width: 10,
  79. X: 2,
  80. Y: 2,
  81. Style: DOUBLE,
  82. }}
  83. // Use simple renders for testing
  84. m.SelectedR = func(text string) string {
  85. return ColorText("BLACK ON WHITE") + text
  86. }
  87. m.UnselectedR = func(text string) string {
  88. return ColorText("WHI ON BLA") + text
  89. }
  90. m.AddSelection("A", "ART")
  91. m.AddSelection("B", "BOO")
  92. m.AddSelection("C", "Cat")
  93. var choice int
  94. // preload the key buffer with commands
  95. // BUG: I can't use \x1b[A because of the way extended codes work there.
  96. keys := "\x00\x50\x00\x50\r" // down, down, ENTER
  97. server.Write([]byte(keys))
  98. choice = m.Choose(&d)
  99. option := m.GetOption(choice)
  100. if choice != 3 {
  101. t.Errorf("Expected menu option 3, got %d / %c", choice, option)
  102. }
  103. if option != 'C' {
  104. t.Errorf("Expected menu option C, got %c / %d", option, choice)
  105. }
  106. // Read the display output
  107. server.SetReadDeadline(time.Now().Add(time.Millisecond * 50))
  108. r, err := server.Read(buffer)
  109. server.SetReadDeadline(time.Time{})
  110. output := string(buffer[:r])
  111. // t.Errorf("OUTPUT: (len %d) %#v", r, output)
  112. parts := strings.SplitAfter(output, "H")
  113. for _, part := range parts {
  114. t.Logf("Got: %#v", part)
  115. }
  116. keys = "\x00\x48\r" // up, ENTER
  117. server.Write([]byte(keys))
  118. m.Chosen = 1
  119. choice = m.Choose(&d)
  120. option = m.GetOption(choice)
  121. if choice != 1 {
  122. t.Errorf("Expected menu option 1, got %d / %c", choice, option)
  123. }
  124. if option != 'A' {
  125. t.Errorf("Expected menu option A, got %c / %d", option, choice)
  126. }
  127. // Read the display output
  128. server.SetReadDeadline(time.Now().Add(time.Millisecond * 50))
  129. r, err = server.Read(buffer)
  130. server.SetReadDeadline(time.Time{})
  131. output = string(buffer[:r])
  132. // Test HOME, END, selecting an option, 8 & 2.
  133. keys = "\x00\x4f\r" // END, ENTER
  134. server.Write([]byte(keys))
  135. m.Chosen = 1
  136. choice = m.Choose(&d)
  137. option = m.GetOption(choice)
  138. if choice != 3 || option != 'C' {
  139. t.Errorf("Expected 3/C, got %d/%c", choice, option)
  140. }
  141. // Read the display output
  142. server.SetReadDeadline(time.Now().Add(time.Millisecond * 50))
  143. r, err = server.Read(buffer)
  144. server.SetReadDeadline(time.Time{})
  145. output = string(buffer[:r])
  146. keys = "\x00\x47\r" // HOME, ENTER
  147. server.Write([]byte(keys))
  148. m.Chosen = 2
  149. choice = m.Choose(&d)
  150. option = m.GetOption(choice)
  151. if choice != 1 || option != 'A' {
  152. t.Errorf("Expected 1/A, got %d/%c", choice, option)
  153. }
  154. // Read the display output
  155. server.SetReadDeadline(time.Now().Add(time.Millisecond * 50))
  156. r, err = server.Read(buffer)
  157. server.SetReadDeadline(time.Time{})
  158. output = string(buffer[:r])
  159. server.Write([]byte("B"))
  160. m.Chosen = 0
  161. choice = m.Choose(&d)
  162. option = m.GetOption(choice)
  163. if choice != 2 || option != 'B' {
  164. t.Errorf("Expected 2/B, got %d/%c", choice, option)
  165. }
  166. // Read the display output
  167. server.SetReadDeadline(time.Now().Add(time.Millisecond * 50))
  168. r, err = server.Read(buffer)
  169. server.SetReadDeadline(time.Time{})
  170. output = string(buffer[:r])
  171. keys = "2\r" // "Down", ENTER
  172. server.Write([]byte(keys))
  173. m.Chosen = 1
  174. choice = m.Choose(&d)
  175. option = m.GetOption(choice)
  176. if choice != 3 || option != 'C' {
  177. t.Errorf("Expected 3/C, got %d/%c", choice, option)
  178. }
  179. // Read the display output
  180. server.SetReadDeadline(time.Now().Add(time.Millisecond * 50))
  181. r, err = server.Read(buffer)
  182. server.SetReadDeadline(time.Time{})
  183. output = string(buffer[:r])
  184. }