more test cases

cloudflare · Dec 26, 2023 · fc74b5a · fc74b5a
1 parent a42d836
commit fc74b5a
Showing 1 changed file with 28 additions and 11 deletions.
diff --git a/tests/test_ips.py b/tests/test_ips.py
@@ -6,15 +6,32 @@
 
 import CloudFlare
 
-class TestCloudflare:
-    def test_ips(self):
-        # no auth required
-        cf = CloudFlare.CloudFlare()
-        assert isinstance(cf, CloudFlare.CloudFlare)
-        ips = cf.ips()
-        assert isinstance(ips, dict)
-        assert isinstance(ips['ipv4_cidrs'], list)
-        assert isinstance(ips['ipv6_cidrs'], list)
-        assert len(ips['ipv4_cidrs']) > 0
-        assert len(ips['ipv6_cidrs']) > 0
+cf = None
 
+def test_cloudflare():
+    global cf
+    cf = CloudFlare.CloudFlare()
+    assert isinstance(cf, CloudFlare.CloudFlare)
+
+def test_ips():
+    # no auth required
+    cf = CloudFlare.CloudFlare()
+    assert isinstance(cf, CloudFlare.CloudFlare)
+    ips = cf.ips()
+    assert isinstance(ips, dict)
+    assert isinstance(ips['ipv4_cidrs'], list)
+    assert isinstance(ips['ipv6_cidrs'], list)
+    assert len(ips['ipv4_cidrs']) > 0
+    assert len(ips['ipv6_cidrs']) > 0
+
+def test_ips_plus_jdcloud():
+    # no auth required
+    params = {'networks':'jdcloud'}
+    ips = cf.ips(params=params)
+    assert isinstance(ips, dict)
+    assert isinstance(ips['ipv4_cidrs'], list)
+    assert isinstance(ips['ipv6_cidrs'], list)
+    assert isinstance(ips['jdcloud_cidrs'], list)
+    assert len(ips['ipv4_cidrs']) > 0
+    assert len(ips['ipv6_cidrs']) > 0
+    assert len(ips['jdcloud_cidrs']) > 0