added unit test

This commit is contained in:
Anastasios Zouzias
2019-11-12 14:06:10 +01:00
parent 324005a126
commit e7ed0fa9ea
4 changed files with 34 additions and 1 deletions
+1
View File
@@ -6,3 +6,4 @@ elasticsearch-async
setuptools
wheel
pymisp
pytest
+1 -1
View File
@@ -294,7 +294,7 @@ class ElasticsearchDSLBackend(RulenameCommentMixin, ElasticsearchWildcardHandlin
"buckets_path": {
"count": count_distinct_agg_name
},
'script': script_limit
"script": script_limit
}
}
}
+30
View File
@@ -0,0 +1,30 @@
from sigma.backends.elasticsearch import ElasticsearchDSLBackend
from sigma.configuration import SigmaConfiguration
from sigma.parser.condition import SigmaAggregationParser
def test_backend_elastic():
sigma_config = SigmaConfiguration()
backend = ElasticsearchDSLBackend(sigma_config)
# setup the aggregator input object without calling __init__()
agg = object.__new__(SigmaAggregationParser)
agg.condition = "3"
agg.cond_op = "<"
agg.aggfunc = SigmaAggregationParser.AGGFUNC_COUNT
agg.aggfield = "aggfield"
agg.groupfield = "groupfield"
# Make queries non-empty
backend.queries = [{}]
backend.generateAggregation(agg)
assert len(backend.queries) == 1, "backend has exactly one query"
assert (
"groupfield_count" in backend.queries[0]["aggs"]["aggs"]
), "groupfield_count is the top aggregation key"
assert (
"aggfield_distinct"
in backend.queries[0]["aggs"]["aggs"]["groupfield_count"]["aggs"]
), "aggfield_distinct is the nested aggregation key"
+2
View File
@@ -1 +1,3 @@
def test_collection():
pass