Refactored unit tests

how-to-create-encrypted-paper-backup/tests/bip39.exp

@@ -1,190 +1,104 @@
 #!/usr/bin/expect
 
-# exp_internal 1
-# log_file -noappend ~/expect.log
-
-set timeout 30
-
-proc bold {} {
-  return [exec tput bold]
-}
-proc red {} {
-  return [exec tput setaf 1]
-}
-proc green {} {
-  return [exec tput setaf 2]
-}
-proc normal {} {
-  return [exec tput sgr0]
-}
+source ./test.exp
 
 set bip39_mnemonic "online pipe enough dutch decorate want moment scheme rigid enlist blast boat purse sick chalk shop brush all return betray jacket salon abandon retire"
 set electrum_mnemonic "month pet slice know amused leave now vague palm guess cattle element cross truly auto put among craft repeat van sample nephew sad family"
-set failed "[bold][red]Failed[normal]"
-set ok "[bold][green]OK[normal]"
 
-puts "[bold]Should create BIP39 mnemonic[normal]"
+test_label "Should create BIP39 mnemonic"
 
 spawn qr-backup.sh --create-bip39-mnemonic
 
 expect {
-  default {
-    puts $failed
-    exit 1
-  }
   -re {Format USB flash drive \(y or n\)\?} {
-    send "n\r"
+    test_send "n\r"
   }
 }
 
 expect {
-  default {
-    puts $failed
-    exit 1
-  }
   -re {\[sudo\] password for pi:} {
-    send "$env(password)\r"
+    test_send "$env(password)\r"
   }
 }
 
 expect {
-  default {
-    puts $failed
-    exit 1
-  }
   -re {Creating BIP39 mnemonic…}
 }
 
 expect {
-  default {
-    puts $failed
-    exit 1
-  }
-  -re {(([a-z]+ ?){24})} {
-    exit 0
+  -re {([a-z]+ ?){24}} {
+    test_ok true
   }
 }
 
-puts "[bold]Should validate valid BIP39 mnemonic[normal]"
+test_label "Should validate valid BIP39 mnemonic"
 
 spawn qr-backup.sh --validate-bip39-mnemonic
 
 expect {
-  default {
-    puts $failed
-    exit 1
-  }
   -re {Format USB flash drive \(y or n\)\?} {
-    send "n\r"
+    test_send "n\r"
   }
 }
 
 expect {
-  default {
-    puts $failed
-    exit 1
-  }
   -re {\[sudo\] password for pi:} {
-    send "$env(password)\r"
+    test_send "$env(password)\r"
   }
 }
 
 expect {
-  default {
-    puts $failed
-    exit 1
-  }
   -re {Please type secret and press enter, then ctrl\+d} {
-    sleep 0.1
-    send "$bip39_mnemonic\r"
-    sleep 0.1
-    send "\x04"
+    test_send "$bip39_mnemonic\r"
+    test_send "\x04"
   }
 }
 
 expect {
-  default {
-    puts $failed
-    exit 1
-  }
-  -re {Please type bip39_mnemonic and press enter, then ctrl\+d \(again\)} {
-    sleep 0.1
-    send "$secret\r"
-    sleep 0.1
-    send "\x04"
+  -re {Please type secret and press enter, then ctrl\+d \(again\)} {
+    test_send "$bip39_mnemonic\r"
+    test_send "\x04"
   }
 }
 
 expect {
-  default {
-    puts $failed
-    exit 1
-  }
   -re {Please type passphrase and press enter} {
-    puts $ok
-    exit 0
+    test_ok true
   }
 }
 
-puts "[bold]Should fail to validate invalid BIP39 mnemonic[normal]"
+test_label "Should fail to validate invalid BIP39 mnemonic"
 
 spawn qr-backup.sh --validate-bip39-mnemonic
 
 expect {
-  default {
-    puts $failed
-    exit 1
-  }
   -re {Format USB flash drive \(y or n\)\?} {
-    send "n\r"
+    test_send "n\r"
   }
 }
 
 expect {
-  default {
-    puts $failed
-    exit 1
-  }
   -re {\[sudo\] password for pi:} {
-    send "$env(password)\r"
+    test_send "$env(password)\r"
   }
 }
 
 expect {
-  default {
-    puts $failed
-    exit 1
-  }
   -re {Please type secret and press enter, then ctrl\+d} {
-    sleep 0.1
-    send "$electrum_mnemonic\r"
-    sleep 0.1
-    send "\x04"
+    test_send "$electrum_mnemonic\r"
+    test_send "\x04"
   }
 }
 
 expect {
-  default {
-    puts $failed
-    exit 1
-  }
-  -re {Please type electrum_mnemonic and press enter, then ctrl\+d \(again\)} {
-    sleep 0.1
-    send "$secret\r"
-    sleep 0.1
-    send "\x04"
+  -re {Please type secret and press enter, then ctrl\+d \(again\)} {
+    test_send "$electrum_mnemonic\r"
+    test_send "\x04"
   }
 }
 
 expect {
-  default {
-    puts $failed
-    exit 1
-  }
   -re {Invalid BIP39 mnemonic} {
-    puts $ok
-    exit 0
+    test_ok true
   }
 }
-
-interact