TestMain.go 1.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566
  1. package main
  2. import (
  3. "git.swzry.com/zry/zTinyPasswd"
  4. "fmt"
  5. )
  6. func main() {
  7. ztpm,err :=zTinyPasswd.NewPasswdManagerFromFileName("test.json")
  8. if err != nil{
  9. fmt.Println("Failed: ", err)
  10. return
  11. }
  12. ztpm.Add("root", []byte("hello,zry"), "admin")
  13. ztpm.Add("zry", []byte("hello,zry"), "admin")
  14. ztpm.Add("test1", []byte("qwertyuiop"), "user")
  15. ztpm.Add("test2", []byte("asdfghjkl"), "user")
  16. ztpm.Add("test3", []byte("zxcvbnm"), "user")
  17. u1r,u1e := ztpm.GetUserInfo("zry")
  18. fmt.Println("User zry Info:", u1r, u1e)
  19. fmt.Println("\nList Users @ T1:")
  20. ul := ztpm.ListUser()
  21. for i,v := range ul {
  22. utr,ute := ztpm.GetUserInfo(v)
  23. fmt.Println(i, v, utr, ute)
  24. }
  25. ztpm.ChangeRole("zry", "master")
  26. ztpm.ChangeExternal("test1", "ExtMsgForTest1")
  27. fmt.Println("\nList Users @ T2:")
  28. ul = ztpm.ListUser()
  29. for i,v := range ul {
  30. utr,ute := ztpm.GetUserInfo(v)
  31. fmt.Println(i, v, utr, ute)
  32. }
  33. ztpm.DeleteUser("test3")
  34. ztpm.RenameUser("test1", "etest")
  35. fmt.Println("\nList Users @ T3:")
  36. ul = ztpm.ListUser()
  37. for i,v := range ul {
  38. utr,ute := ztpm.GetUserInfo(v)
  39. fmt.Println(i, v, utr, ute)
  40. }
  41. ztpm.Add("hhh", []byte("??????"), "guest")
  42. fmt.Println("Auth Test 1:")
  43. suc := ztpm.Auth("zry", []byte("hello,zry"))
  44. if suc {
  45. fmt.Println("User zry Logged In.")
  46. }else {
  47. fmt.Println("User zry Auth Failed.")
  48. }
  49. fmt.Println("Auth Test 2:")
  50. suc = ztpm.Auth("zry", []byte("hellozry"))
  51. if suc {
  52. fmt.Println("User zry Logged In.")
  53. }else {
  54. fmt.Println("User zry Auth Failed.")
  55. }
  56. ztpm.ChangePassword("zry", []byte("hellozry"))
  57. fmt.Println("Auth Test 3:")
  58. suc = ztpm.Auth("zry", []byte("hello,zry"))
  59. if suc {
  60. fmt.Println("User zry Logged In.")
  61. }else {
  62. fmt.Println("User zry Auth Failed.")
  63. }
  64. }