nlp_wordsegment_ft.cc 2.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465
  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. // create request and assign parameters
  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 good 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\":\"good\"},{\"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. // create request and assign parameters
  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. EXPECT_TRUE(outcome.error().detail() == error);
  55. AlibabaCloud::ShutdownSdk();
  56. }
  57. }