Skip to content

Commit d032798

Browse files
Merge pull request #165 from splunk/update_host_range
Minor Fixes in the plugin.
2 parents 54bff88 + 92b5a45 commit d032798

File tree

2 files changed

+13
-11
lines changed

2 files changed

+13
-11
lines changed

pytest_splunk_addon/standard_lib/index_tests/test_templates.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -185,11 +185,7 @@ def test_indextime_time(
185185
if splunk_indextime_time.get("identifier"):
186186
extra_filter = splunk_indextime_time.get("identifier")
187187
else:
188-
extra_filter = (
189-
"(host="
190-
+ " OR host=".join(splunk_indextime_time.get("hosts"))
191-
+ ")"
192-
)
188+
extra_filter = "host IN (\""+"\",\"".join(set(splunk_indextime_time.get("hosts")))+"\")"
193189

194190
if splunk_indextime_time["tokenized_event"].time_values:
195191
extra_filter += " | eval e_time=_time"
@@ -208,7 +204,8 @@ def test_indextime_time(
208204
retries=splunk_search_util.search_retry,
209205
)
210206
results = list(results)
211-
207+
if not results:
208+
assert False, "No Events found for query: " + search
212209
result_fields = {
213210
key: [ceil(float(item[key])) for item in results]
214211
for key in results[0].keys()
@@ -276,4 +273,4 @@ def test_indextime_line_breaker(
276273

277274
assert (
278275
count_from_results == expected_events_count
279-
), f"Expected count: {expected_events_count} Actual Count: {count_from_results}"
276+
), f"Query: {query} \nExpected count: {expected_events_count} Actual Count: {count_from_results}"

pytest_splunk_addon/standard_lib/sample_generation/sample_event.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
src_ipv6, dest_ipv6 = 0, 0
1313
host_count, fqdn_count = 0,0
1414
url_ip_count = 0
15+
host_ipv4_octet_count, dvc_ipv4_octet_count = 0, 0
1516

1617
ip_rules = {
1718
"src":{
@@ -107,13 +108,17 @@ def get_ipv4(self, rule):
107108
addr = [int(src_ipv4 / 256) % 256, src_ipv4 % 256]
108109
return "".join([ip_rules.get(rule)["ipv4"], str(addr[0]), ".", str(addr[1])])
109110
elif rule == "host":
110-
global host_ipv4
111+
global host_ipv4, host_ipv4_octet_count
111112
host_ipv4 += 1
112-
return "".join([ip_rules.get(rule)["ipv4"], str(host_ipv4 % 101), ".0"])
113+
if host_ipv4 == 101:
114+
host_ipv4 = 51
115+
host_ipv4_octet_count += 1
116+
return "".join([ip_rules.get(rule)["ipv4"], str(host_ipv4 % 101), ".", str(host_ipv4_octet_count % 256)])
113117
elif rule == "dvc":
114-
global dvc_ipv4
118+
global dvc_ipv4, dvc_ipv4_octet_count
115119
dvc_ipv4 += 1
116-
return "".join([ip_rules.get(rule)["ipv4"], str(dvc_ipv4 % 51), ".0"])
120+
dvc_ipv4_octet_count += 1
121+
return "".join([ip_rules.get(rule)["ipv4"], str(dvc_ipv4 % 51), ".", str(dvc_ipv4_octet_count % 256)])
117122
elif rule == "dest":
118123
global dest_ipv4
119124
dest_ipv4 += 1

0 commit comments

Comments
 (0)