|
| 1 | +use crate::common::default_agent_options::create_tls_config; |
1 | 2 | use crate::common::test_config::setup_test; |
2 | 3 | use bytes::Bytes; |
3 | | -use couchbase_core::httpx::client::{Client, ClientConfig, ReqwestClient}; |
| 4 | +use couchbase_core::httpx::client::{self, Client, ClientConfig, ReqwestClient}; |
4 | 5 | use couchbase_core::httpx::decoder::Decoder; |
5 | 6 | use couchbase_core::httpx::raw_json_row_streamer::{RawJsonRowItem, RawJsonRowStreamer}; |
6 | 7 | use couchbase_core::httpx::request::{Auth, BasicAuth, Request}; |
@@ -47,22 +48,26 @@ pub struct QueryMetrics { |
47 | 48 | #[test] |
48 | 49 | fn test_row_streamer() { |
49 | 50 | setup_test(async |config| { |
50 | | - let addrs = config.memd_addrs; |
| 51 | + let scheme = if config.use_ssl { "https" } else { "http" }; |
| 52 | + let addrs = &config.http_addrs; |
51 | 53 |
|
52 | | - let host = addrs.first().unwrap().host.clone(); |
| 54 | + let addr = addrs.first().unwrap(); |
| 55 | + let host = &addr.host; |
| 56 | + let port = addr.port; |
53 | 57 |
|
54 | | - let basic_auth = BasicAuth::new(config.username, config.password); |
| 58 | + let basic_auth = BasicAuth::new(&config.username, &config.password); |
55 | 59 |
|
56 | 60 | let request_body = json!({"statement": "select i from array_range(1, 10000) AS i;"}); |
57 | | - let uri = format!("http://{host}:8093/query/service"); |
| 61 | + let uri = format!("{scheme}://{host}:{port}/_p/query/query/service"); |
58 | 62 |
|
59 | 63 | let request = Request::new(Method::POST, uri) |
60 | 64 | .user_agent("rscbcorex".to_string()) |
61 | 65 | .auth(Auth::BasicAuth(basic_auth)) |
62 | 66 | .content_type("application/json".to_string()) |
63 | 67 | .body(Bytes::from(serde_json::to_vec(&request_body).unwrap())); |
64 | 68 |
|
65 | | - let client = ReqwestClient::new(ClientConfig::default()).unwrap(); |
| 69 | + let client_config = ClientConfig::default().tls_config(create_tls_config(&config)); |
| 70 | + let client = ReqwestClient::new(client_config).unwrap(); |
66 | 71 |
|
67 | 72 | let resp = timeout(Duration::from_secs(10), client.execute(request)) |
68 | 73 | .await |
@@ -116,19 +121,23 @@ fn test_row_streamer() { |
116 | 121 | #[test] |
117 | 122 | fn test_json_block_read() { |
118 | 123 | setup_test(async |config| { |
119 | | - let addrs = config.memd_addrs; |
| 124 | + let scheme = if config.use_ssl { "https" } else { "http" }; |
| 125 | + let addrs = &config.http_addrs; |
120 | 126 |
|
121 | | - let host = addrs.first().unwrap().host.clone(); |
| 127 | + let addr = addrs.first().unwrap(); |
| 128 | + let host = &addr.host; |
| 129 | + let port = addr.port; |
122 | 130 |
|
123 | | - let basic_auth = BasicAuth::new(config.username, config.password); |
124 | | - let uri = format!("http://{host}:8091/pools/default/terseClusterInfo"); |
| 131 | + let basic_auth = BasicAuth::new(&config.username, &config.password); |
| 132 | + let uri = format!("{scheme}://{host}:{port}/pools/default/terseClusterInfo"); |
125 | 133 |
|
126 | 134 | let request = Request::new(Method::GET, uri) |
127 | 135 | .user_agent("rscbcorex".to_string()) |
128 | 136 | .auth(Auth::BasicAuth(basic_auth)) |
129 | 137 | .content_type("application/json".to_string()); |
130 | 138 |
|
131 | | - let client = ReqwestClient::new(ClientConfig::default()).expect("could not create client"); |
| 139 | + let client_config = ClientConfig::default().tls_config(create_tls_config(&config)); |
| 140 | + let client = ReqwestClient::new(client_config).expect("could not create client"); |
132 | 141 |
|
133 | 142 | let res = timeout(Duration::from_secs(10), client.execute(request)) |
134 | 143 | .await |
|
0 commit comments