python-bgpstuff.net/test_bgpstuff.py at main · mellowdrifter/python-bgpstuff.net · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import bgpstuff
import ipaddress
import unittest


class ClientTest(unittest.TestCase):
    def setUp(self):
        self.client = bgpstuff.Client()

    def test_get_route(self):
        self.client.get_route("8.8.8.8")
        self.assertEqual(ipaddress.ip_network(
            '8.8.8.0/24'), self.client.route)
        self.client.get_route("1.2.3.4")
        self.assertIsNone(self.client.route)
        self.client.get_route("2600::")
        self.assertEqual(ipaddress.ip_network(
            '2600::/48'), self.client.route)
        with self.assertRaises(ValueError):
            self.client.get_route("10.0.0.0")

    def test_get_origin(self):
        self.client.get_origin("8.8.8.8")
        self.assertEqual(15169, self.client.origin)
        with self.assertRaises(ValueError):
            self.client.get_route("10.0.0.0")

    def test_get_as_path(self):
        self.client.get_as_path("8.8.8.8")
        self.assertEqual(15169, self.client.as_path[-1])

    def test_get_roa(self):
        self.client.get_roa("1.1.1.1")
        self.assertEqual("VALID", self.client.roa)
        with self.assertRaises(ValueError):
            self.client.get_route("10.0.0.0")

    def test_as_name(self):
        self.client.get_as_name(15169)
        self.assertEqual("GOOGLE", self.client.as_name)
        with self.assertRaises(ValueError):
            self.client.get_route("hi")

    def test_get_sourced_prefixes(self):
        self.client.get_sourced_prefixes(15169)
        dns = ipaddress.ip_network("8.8.4.0/24")
        if dns not in self.client.sourced:
            self.assert_("8.8.4.0/24 network not found")
        with self.assertRaises(ValueError):
            self.client.get_sourced_prefixes("hi")

    def test_get_totals(self):
        self.client.get_totals()
        self.assertGreater(self.client.total_v4, 800000)
        self.assertGreater(self.client.total_v6, 100000)

    def test_get_all_as_names(self):
        self.client.get_as_names()
        self.assertGreater(len(self.client.all_as_names), 100000)
        self.assertEqual("GOOGLE", self.client.all_as_names[15169])
        self.client.get_as_name(3356)
        self.assertTrue(self.client.exists)
        self.assertEqual(200, self.client.status_code)
        self.client.get_as_name(4100000000)
        self.assertFalse(self.client.exists)
        self.assertEqual(200, self.client.status_code)

    def test_get_vrps(self):
        self.client.get_vrps(15169)
        dns = ipaddress.ip_network("8.8.4.0/24")
        if dns not in self.client.vrps:
            self.assert_("8.8.4.0/24 vrp not found")
        v6dns = ipaddress.ip_network("2001:4860::/32")
        if v6dns not in self.client.vrps:
            self.assert_("2001:4860::/32 vrp not found")
        with self.assertRaises(ValueError):
            self.client.get_vrps("sup")

    def tearDown(self):
        self.client._close_session()