nlp_wordsegment_ft.cc 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566
  1. #include <iostream>
  2. #include <cstdlib>
  3. #include <cstring>
  4. #include "gtest/gtest.h"
  5. #include "../utils.h"
  6. #include "alibabacloud/core/AlibabaCloud.h"
  7. #include "alibabacloud/core/CommonClient.h"
  8. #include "alibabacloud/core/HttpRequest.h"
  9. using namespace std;
  10. using namespace AlibabaCloud;
  11. namespace {
  12. TEST(nlp, wordsegment) {
  13. utUtils utils;
  14. string key = utils.get_env("ENV_AccessKeyId");
  15. string secret = utils.get_env("ENV_AccessKeySecret");
  16. AlibabaCloud::InitializeSdk();
  17. ClientConfiguration configuration("cn-shanghai");
  18. CommonClient client(key, secret, configuration);
  19. // 创建API请求并设置参数
  20. CommonRequest request(CommonRequest::RoaPattern);
  21. request.setScheme("http");
  22. request.setDomain("nlp.cn-shanghai.aliyuncs.com");
  23. request.setResourcePath("/nlp/api/wordsegment/general");
  24. request.setHttpMethod(HttpRequest::Post);
  25. const char * data = "{\"lang\":\"ZH\",\"text\":\"Iphone is a bad choice 专用数据线\"}";
  26. request.setContent(data, strlen(data));
  27. request.setHeaderParameter("Content-Type", "application/json;chrset=utf-8");
  28. request.setVersion("2018-04-08");
  29. auto outcome = client.commonResponse(request);
  30. EXPECT_TRUE(outcome.error().errorCode().empty());
  31. const std::string expected = "{\"data\":[{\"id\":0,\"word\":\"Iphone\"},{\"id\":1,\"word\":\" \"},{\"id\":2,\"word\":\"is\"},{\"id\":3,\"word\":\" \"},{\"id\":4,\"word\":\"a\"},{\"id\":5,\"word\":\" \"},{\"id\":6,\"word\":\"bad\"},{\"id\":7,\"word\":\" \"},{\"id\":8,\"word\":\"choice\"},{\"id\":9,\"word\":\" \"},{\"id\":10,\"word\":\"专用\"},{\"id\":11,\"word\":\"数据线\"}]}";
  32. EXPECT_TRUE(outcome.result().payload() == expected);
  33. AlibabaCloud::ShutdownSdk();
  34. }
  35. TEST(nlp, wordsegment_error) {
  36. utUtils utils;
  37. string key = utils.get_env("ENV_AccessKeyId");
  38. string secret = utils.get_env("ENV_AccessKeySecret");
  39. AlibabaCloud::InitializeSdk();
  40. ClientConfiguration configuration("cn-shanghai");
  41. CommonClient client(key, secret, configuration);
  42. // 创建API请求并设置参数
  43. CommonRequest request(CommonRequest::RoaPattern);
  44. request.setScheme("http");
  45. request.setDomain("nlp.cn-shanghai.aliyuncs.com");
  46. request.setResourcePath("/nlp/api/wordsegment/general");
  47. request.setHttpMethod(HttpRequest::Post);
  48. const char * data = "invlaid text";
  49. request.setContent(data, strlen(data));
  50. request.setHeaderParameter("Content-Type", "application/json;chrset=utf-8");
  51. request.setVersion("2018-04-08");
  52. auto outcome = client.commonResponse(request);
  53. const string error = "{\"errorCode\":10007,\"errorMsg\":\"body json format invalid\"}";
  54. cout << "error: " << outcome.error().detail() << endl;
  55. EXPECT_TRUE(outcome.error().detail() == error);
  56. AlibabaCloud::ShutdownSdk();
  57. }
  58. }