From 9f315f1728d54796bb05db52839333643c313b2b Mon Sep 17 00:00:00 2001
From: chrisr3d <chris.studer.68@gmail.com>
Date: Tue, 8 Sep 2020 16:24:41 +0200
Subject: [PATCH] chg: Updated the bgpranking expansion module test

---
 tests/test_expansions.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/tests/test_expansions.py b/tests/test_expansions.py
index a56fbe7..1aa0f7a 100644
--- a/tests/test_expansions.py
+++ b/tests/test_expansions.py
@@ -97,9 +97,16 @@ class TestExpansions(unittest.TestCase):
             self.assertEqual(self.get_errors(response), 'An API key for APIVoid is required.')
 
     def test_bgpranking(self):
-        query = {"module": "bgpranking", "AS": "13335"}
+        query = {
+            "module": "bgpranking",
+            "attribute": {
+                "type": "AS",
+                "value": "13335",
+                "uuid": "ea89a33b-4ab7-4515-9f02-922a0bee333d"
+            }
+        }
         response = self.misp_modules_post(query)
-        self.assertEqual(self.get_values(response)['response']['asn_description'], 'CLOUDFLARENET, US')
+        self.assertEqual(self.get_object(response), 'asn')
 
     def test_btc_steroids(self):
         query = {"module": "btc_steroids", "btc": "1ES14c7qLb5CYhLMUekctxLgc1FV2Ti9DA"}