userLib_test.go 2.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182
  1. package test
  2. import (
  3. "app.yhyue.com/moapp/jy_docs/rpc/userlib/userlib"
  4. "app.yhyue.com/moapp/jy_docs/rpc/userlib/userlibclient"
  5. "context"
  6. "github.com/tal-tech/go-zero/core/discov"
  7. "github.com/tal-tech/go-zero/zrpc"
  8. "log"
  9. "testing"
  10. )
  11. //用户收藏单元测试
  12. func Test_UserDocCollect(t *testing.T) {
  13. client := zrpc.MustNewClient(zrpc.RpcClientConf{
  14. Etcd: discov.EtcdConf{
  15. Hosts: []string{"192.168.3.240:2379"},
  16. Key: "jydocs.userlib.rpc",
  17. },
  18. })
  19. userLib := userlibclient.NewUserLib(client)
  20. resp, err := userLib.DocCollect(context.Background(), &userlibclient.UserCollectRequest{
  21. /*DocId: "1",
  22. UserId: "1",
  23. SourceUserId: "2",
  24. DocName: "2020招标文件白皮书.docx",
  25. DocFileType: userlib.UserDocFileType_Word,
  26. DocFileSize: 1002,
  27. DocPageSize: 20,
  28. DocCategory: userlib.UserDocCategory_Collect,
  29. DocFileSuffix:"docx",
  30. Cost:0,*/
  31. })
  32. if err != nil {
  33. log.Println("UserDoc Collect call error:", err)
  34. }
  35. log.Print("UserDoc Collect call response:", resp)
  36. }
  37. func Test_UserDocDelete(t *testing.T) {
  38. client := zrpc.MustNewClient(zrpc.RpcClientConf{
  39. Etcd: discov.EtcdConf{
  40. Hosts: []string{"192.168.3.240:2379"},
  41. Key: "jydocs.userlib.rpc",
  42. },
  43. })
  44. userLib := userlibclient.NewUserLib(client)
  45. resp, err := userLib.DocDelete(context.Background(), &userlibclient.UserDeleteRequest{
  46. //DocId: "8",
  47. })
  48. if err != nil {
  49. log.Println("UserDoc Collect call error:", err)
  50. }
  51. log.Print("UserDoc Collect call response:", resp)
  52. }
  53. //用户转存单元测试
  54. func Test_UserDocDwnload(t *testing.T) {
  55. client := zrpc.MustNewClient(zrpc.RpcClientConf{
  56. Etcd: discov.EtcdConf{
  57. Hosts: []string{"127.0.0.1:2380"},
  58. Key: "jydocs.userlib.rpc",
  59. },
  60. })
  61. userLib := userlibclient.NewUserLib(client)
  62. resp, err := userLib.DocDownload(context.Background(), &userlibclient.UserCollectRequest{
  63. DocId: "123",
  64. UserId: "wh5",
  65. SourceUserId: "2",
  66. DocName: "2020招标文件白皮书.docx",
  67. DocFileType: userlib.UserDocFileType_Word,
  68. DocFileSize: 1002,
  69. DocPageSize: 20,
  70. DocCategory: userlib.UserDocCategory_Collect,
  71. DocFileSuffix:"docx",
  72. Cost:1000,
  73. })
  74. if err != nil {
  75. log.Println("UserDoc Collect call error:", err)
  76. }
  77. log.Print("UserDoc Collect call response:", resp.Message,resp.Code)
  78. }