# (c) cavaliba.com - tests / app_data.ipam.ip_list_filename

from django.test import SimpleTestCase

from app_data.ipam import ip_list_filename


class IpamIpListFilenameTest(SimpleTestCase):
    def test_cidr_flattened_into_one_safe_filename(self):
        self.assertEqual(ip_list_filename("10.1.0.0/16"), "ipam_ip_list_10-1-0-0_16.csv")

    def test_no_path_separator_survives(self):
        # the "/" of the CIDR must not turn the download name into a path
        self.assertNotIn("/", ip_list_filename("0.0.0.0/0"))
        self.assertEqual(ip_list_filename("0.0.0.0/0"), "ipam_ip_list_0-0-0-0_0.csv")

    def test_single_host_prefix(self):
        self.assertEqual(ip_list_filename("203.0.113.7/32"), "ipam_ip_list_203-0-113-7_32.csv")