Added tests for signature stats

CybercentreCanada · May 21, 2020 · 3aac6f2 · 3aac6f2
1 parent 4a2dafd
commit 3aac6f2
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/test/test_signature.py b/test/test_signature.py
@@ -89,7 +89,7 @@ def test_download_file_handle(datastore, client):
  signature_id = random_id_from_collection(datastore, 'signature', q="type:yara")
  query = "id:{}".format(signature_id)
  output = "/tmp/sigs_{}_obj".format(get_random_id())
- res = client.signature.download(query=query, output=open(output, 'wb'), safe=False)
+ res = client.signature.download(query=query, output=open(output, 'wb'))
  assert res
 
  found = False
@@ -134,6 +134,11 @@ def test_download_raw(datastore, client):
  assert b"suricata/sample_suricata.rules" in res
 
 
+def test_stats(datastore, client):
+ res = client.signature.stats()
+ assert len(res) == datastore.signature.search('id:*')['total']
+
+
 def test_update_available(datastore, client):
  res = client.signature.update_available()
  assert res['update_available']