Co-authored-by: Pascal Seitz <pascal.seitz@datadoghq.com>
This commit is contained in:
PSeitz
2025-10-23 10:23:34 +02:00
committed by GitHub
parent 938bfec8b7
commit 60225bdd45
3 changed files with 20 additions and 12 deletions

View File

@@ -781,9 +781,7 @@ mod tests {
use time::{Date, Month}; use time::{Date, Month};
use crate::aggregation::agg_req::Aggregations; use crate::aggregation::agg_req::Aggregations;
use crate::aggregation::intermediate_agg_result::{ use crate::aggregation::intermediate_agg_result::IntermediateAggregationResults;
IntermediateAggregationResult, IntermediateAggregationResults,
};
use crate::aggregation::tests::{ use crate::aggregation::tests::{
exec_request, exec_request_with_query, exec_request_with_query_and_memory_limit, exec_request, exec_request_with_query, exec_request_with_query_and_memory_limit,
get_test_index_from_terms, get_test_index_from_values_and_terms, get_test_index_from_terms, get_test_index_from_values_and_terms,
@@ -791,10 +789,7 @@ mod tests {
use crate::aggregation::{AggregationLimitsGuard, DistributedAggregationCollector}; use crate::aggregation::{AggregationLimitsGuard, DistributedAggregationCollector};
use crate::indexer::NoMergePolicy; use crate::indexer::NoMergePolicy;
use crate::query::AllQuery; use crate::query::AllQuery;
use crate::schema::{ use crate::schema::{IntoIpv6Addr, Schema, FAST, STRING};
IndexRecordOption, IntoIpv6Addr, Schema, TextFieldIndexing, TextOptions, FAST, STORED,
STRING, TEXT,
};
use crate::{Index, IndexWriter}; use crate::{Index, IndexWriter};
#[test] #[test]
@@ -2289,6 +2284,8 @@ mod tests {
// running on different indexes with the same data. // running on different indexes with the same data.
let build_index = || -> crate::Result<Index> { let build_index = || -> crate::Result<Index> {
let mut schema_builder = Schema::builder(); let mut schema_builder = Schema::builder();
let fielda = schema_builder.add_text_field("fielda", FAST);
let fieldb = schema_builder.add_text_field("fieldb", FAST);
let host = schema_builder.add_text_field("host", FAST); let host = schema_builder.add_text_field("host", FAST);
let tags = schema_builder.add_text_field("tags", FAST); let tags = schema_builder.add_text_field("tags", FAST);
let schema = schema_builder.build(); let schema = schema_builder.build();
@@ -2300,6 +2297,8 @@ mod tests {
writer.add_document(doc!( writer.add_document(doc!(
host => "192.168.0.10", host => "192.168.0.10",
tags => "nice", tags => "nice",
fielda => "a",
fieldb => "b",
))?; ))?;
writer.add_document(doc!( writer.add_document(doc!(
host => "192.168.0.1", host => "192.168.0.1",
@@ -2355,7 +2354,9 @@ mod tests {
// --- Aggregations: terms on host and tags --- // --- Aggregations: terms on host and tags ---
let agg_req: Aggregations = serde_json::from_value(json!({ let agg_req: Aggregations = serde_json::from_value(json!({
"hosts": { "terms": { "field": "host" } }, "hosts": { "terms": { "field": "host" } },
"tags": { "terms": { "field": "tags" } } "tags": { "terms": { "field": "tags" } },
"fielda": { "terms": { "field": "fielda" } },
"fieldb": { "terms": { "field": "fieldb" } },
})) }))
.unwrap(); .unwrap();
@@ -2368,12 +2369,21 @@ mod tests {
"tags".to_string(), "tags".to_string(),
serde_json::from_value(json!({ "terms": { "field": "tags" } }))?, serde_json::from_value(json!({ "terms": { "field": "tags" } }))?,
); );
agg_req2.insert(
"fielda".to_string(),
serde_json::from_value(json!({ "terms": { "field": "fielda" } }))?,
);
agg_req2.insert( agg_req2.insert(
"hosts".to_string(), "hosts".to_string(),
serde_json::from_value(json!({ "terms": { "field": "host" } }))?, serde_json::from_value(json!({ "terms": { "field": "host" } }))?,
); );
agg_req2.insert(
"fieldb".to_string(),
serde_json::from_value(json!({ "terms": { "field": "fieldb" } }))?,
);
// make sure the order of the aggregation request is different // make sure the order of the aggregation request is different
assert_ne!(agg_req.keys().next(), agg_req2.keys().next()); // disabled to avoid flaky test with hashmap changes
// assert_ne!(agg_req.keys().next(), agg_req2.keys().next());
let agg_res2 = search(&index2, &agg_req2)?; let agg_res2 = search(&index2, &agg_req2)?;

View File

@@ -41,8 +41,6 @@ const COMPRESSION_BLOCK_SIZE: usize = BitPacker4x::BLOCK_LEN;
#[cfg(test)] #[cfg(test)]
pub(crate) mod tests { pub(crate) mod tests {
use std::iter;
use proptest::prelude::*; use proptest::prelude::*;
use proptest::sample::select; use proptest::sample::select;

View File

@@ -342,7 +342,7 @@ mod tests {
fn test_pack() -> crate::Result<()> { fn test_pack() -> crate::Result<()> {
let mut store_writer = TermInfoStoreWriter::new(); let mut store_writer = TermInfoStoreWriter::new();
let mut term_infos = vec![]; let mut term_infos = vec![];
let offset = |i| (i * 13 + i * i); let offset = |i| i * 13 + i * i;
for i in 0usize..1000usize { for i in 0usize..1000usize {
let term_info = TermInfo { let term_info = TermInfo {
doc_freq: i as u32, doc_freq: i as u32,