def test_verify_variables_multiple(self):
     test_subnet = Subnet()
     test_subnet.network_address = 'THIS IS A TEST'
     test_subnet.ip_range = 'THIS IS A TEST'
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.network_address = '152.2.136.0'
     test_subnet.ip_range = 'THIS IS A TEST'
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.network_address = '152.2.136.0'
     test_subnet.ip_range = '152.2.136.1-152.2.136.62'
     self.assertTrue(test_subnet.verify_variables())
 def test_verify_variables_total_host_count(self):
     test_subnet = Subnet()
     test_subnet.total_host_count = 'THIS IS A TEST'
     self.assertFalse(test_subnet.verify_total_host_count())
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.total_host_count = '123'
     self.assertFalse(test_subnet.verify_total_host_count())
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.total_host_count = '64'
     self.assertTrue(test_subnet.verify_total_host_count())
     self.assertTrue(test_subnet.verify_variables())
 def test_verify_variables_assignable_host_count(self):
     test_subnet = Subnet()
     test_subnet.assignable_host_count = 'THIS IS A TEST'
     self.assertFalse(test_subnet.verify_assignable_host_count())
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.assignable_host_count = '123'
     self.assertFalse(test_subnet.verify_assignable_host_count())
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.assignable_host_count = '62'
     self.assertTrue(test_subnet.verify_assignable_host_count())
     self.assertTrue(test_subnet.verify_variables())
 def test_verify_variables_broadcast(self):
     test_subnet = Subnet()
     test_subnet.broadcast_address = 0
     self.assertFalse(test_subnet.verify_broadcast())
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.broadcast_address = 'THIS IS A TEST'
     self.assertFalse(test_subnet.verify_broadcast())
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.broadcast_address = '152.2.136.0/26'
     self.assertFalse(test_subnet.verify_broadcast())
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.broadcast_address = '152.2.136.63'
     self.assertTrue(test_subnet.verify_broadcast())
     self.assertTrue(test_subnet.verify_variables())
 def test_verify_variables_ip_range(self):
     test_subnet = Subnet()
     test_subnet.ip_range = 0
     self.assertFalse(test_subnet.verify_ip_range())
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.ip_range = 'THIS IS A TEST'
     self.assertFalse(test_subnet.verify_ip_range())
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.ip_range = '152.2.136.0/26'
     self.assertFalse(test_subnet.verify_ip_range())
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.ip_range = '152.2.136.63'
     self.assertFalse(test_subnet.verify_ip_range())
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.ip_range = '152.2.136.0-'
     self.assertFalse(test_subnet.verify_ip_range())
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.ip_range = '152.2.136.0-152.2.136.0'
     self.assertFalse(test_subnet.verify_ip_range())
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.ip_range = '152.2.136.62-152.2.136.1'
     self.assertFalse(test_subnet.verify_ip_range())
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.ip_range = '152.2.136.1-152.2.136.62'
     self.assertTrue(test_subnet.verify_ip_range())
     self.assertTrue(test_subnet.verify_variables())
 def test_verify_variables_network_address(self):
     test_subnet = Subnet()
     test_subnet.network_address = 0
     self.assertFalse(test_subnet.verify_network_address())
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.network_address = 'THIS IS A TEST'
     self.assertFalse(test_subnet.verify_network_address())
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.network_address = '152.2.136.0/26'
     self.assertFalse(test_subnet.verify_network_address())
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.network_address = '152.2.136.0'
     self.assertTrue(test_subnet.verify_network_address())
     self.assertTrue(test_subnet.verify_variables())
 def test_verify_variables_netmask(self):
     test_subnet = Subnet()
     test_subnet.netmask = 0
     self.assertFalse(test_subnet.verify_netmask())
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.netmask = 'THIS IS A TEST'
     self.assertFalse(test_subnet.verify_netmask())
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.netmask = '152.2.136.0'
     self.assertFalse(test_subnet.verify_netmask())
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.netmask = '255.255.255.192'
     self.assertTrue(test_subnet.verify_netmask())
     self.assertTrue(test_subnet.verify_variables())
 def test_verify_variables_cidr(self):
     test_subnet = Subnet()
     test_subnet.cidr_address = 0
     self.assertFalse(test_subnet.verify_cidr_address())
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.cidr_address = 'THIS IS A TEST'
     self.assertFalse(test_subnet.verify_cidr_address())
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.cidr_address = '152.2.136.0'
     self.assertFalse(test_subnet.verify_cidr_address())
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.cidr_address = '152.2.136.0/'
     self.assertFalse(test_subnet.verify_cidr_address())
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.cidr_address = '152.2.136.0//'
     self.assertFalse(test_subnet.verify_cidr_address())
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.cidr_address = '152.2.136.0/Test'
     self.assertFalse(test_subnet.verify_cidr_address())
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.cidr_address = '152.2.136.0/26/'
     self.assertFalse(test_subnet.verify_cidr_address())
     self.assertFalse(test_subnet.verify_variables())
     test_subnet.cidr_address = '152.2.136.0/26'
     self.assertTrue(test_subnet.verify_cidr_address())
     self.assertTrue(test_subnet.verify_variables())
 def test_verify_variables_none(self):
     test_subnet = Subnet()
     self.assertFalse(test_subnet.verify_variable_use())
     self.assertFalse(test_subnet.verify_variables())
Exemplo n.º 10
0
class CIDRCalcGUI:
    def __init__(self):
        self.subnet = Subnet()

        self.cidr = Tk()
        self.cidr.wm_title('CIDR Calculator')

        self.cidr_from_netmask = LabelFrame(
            self.cidr, text='CIDR: Network Address & Netmask')
        self.net_address_label = Label(self.cidr_from_netmask,
                                       text='Network Address:')
        self.netmask_label = Label(self.cidr_from_netmask, text='Netmask:')
        self.net_address_entry = Entry(self.cidr_from_netmask, width=25)
        self.netmask_entry = Entry(self.cidr_from_netmask, width=25)

        self.cidr_from_ip_range = LabelFrame(
            self.cidr, text='CIDR: Network Address & Assignable IP Range')
        self.net_address_label2 = Label(self.cidr_from_ip_range,
                                        text='Network Address:')
        self.ip_range_label = Label(self.cidr_from_ip_range,
                                    text='Assignable IP Range:')
        self.net_address_entry2 = Entry(self.cidr_from_ip_range, width=25)
        self.ip_range_entry = Entry(self.cidr_from_ip_range, width=25)

        self.learning_steps = LabelFrame(self.cidr, text='Learning Steps')

        self.calculate = Button(self.cidr_from_netmask,
                                text='Calculate',
                                command=lambda: self.get_cidr_from_netmask())
        self.calculate2 = Button(self.cidr_from_ip_range,
                                 text='Calculate',
                                 command=lambda: self.get_cidr_from_ip_range())

        self.cidr_from_netmask.grid(row=0)
        self.cidr_from_ip_range.grid(row=1)
        self.learning_steps.grid(row=2, sticky='W')

        self.net_address_label.grid(row=0, column=0, sticky='W')
        self.net_address_entry.grid(row=1, column=0)
        self.netmask_label.grid(row=0, column=1, sticky='W')
        self.netmask_entry.grid(row=1, column=1)
        self.calculate.grid(row=1, column=2)

        self.net_address_label2.grid(row=0, column=0, sticky='W')
        self.net_address_entry2.grid(row=1, column=0)
        self.ip_range_label.grid(row=0, column=1, sticky='W')
        self.ip_range_entry.grid(row=1, column=1)
        self.calculate2.grid(row=1, column=2)

        # self.cidr.mainloop()

    def get_cidr_from_netmask(self):
        if self.subnet.cidr_address != '':
            self.clear_and_reset()
        else:
            self.subnet.network_address = self.net_address_entry.get()
            self.subnet.netmask = self.netmask_entry.get()

            if self.subnet.network_address != '' and self.subnet.netmask != '' and self.subnet.verify_variables(
            ):
                self.subnet.calculate_cidr_from_netmask()
                steps = self.subnet.cidr_address_steps

                step1 = Label(self.learning_steps, text='Step 1:')
                network = Label(self.learning_steps,
                                text='Network Address: {}'.format(
                                    self.subnet.network_address))
                netmask = Label(self.learning_steps,
                                text='NetMask: {}'.format(self.subnet.netmask))
                step2 = Label(self.learning_steps, text='Step 2:')
                binary = Label(self.learning_steps,
                               text='Netmask Binary: {}'.format(
                                   steps[0]['Netmask Binary']))
                step3 = Label(self.learning_steps, text='Step 3:')
                count = Label(self.learning_steps,
                              text="(Count 0's from Step 2)")
                cidr = Label(self.learning_steps,
                             text='CIDR: {}'.format(steps[1]['CIDR']))
                step4 = Label(self.learning_steps, text='Step 4:')
                cidr_address = Label(self.learning_steps,
                                     text='CIDR Address: {}'.format(
                                         self.subnet.cidr_address))

                step1.grid(row=0, column=0, sticky='W')
                network.grid(row=0, column=1, sticky='W')
                netmask.grid(row=0, column=2, sticky='W')
                step2.grid(row=1, column=0, sticky='W')
                binary.grid(row=1, column=1, columnspan=2, sticky='W')
                step3.grid(row=2, column=0, sticky='W')
                count.grid(row=2, column=1, sticky='W')
                cidr.grid(row=2, column=2, sticky='W')
                step4.grid(row=3, column=0, sticky='W')
                cidr_address.grid(row=3, column=1, sticky='W')
            else:
                self.clear_and_reset()
                HelpGUI()

    def get_cidr_from_ip_range(self):
        if self.subnet.cidr_address != '':
            self.clear_and_reset()
        else:
            self.subnet.network_address = self.net_address_entry2.get()
            self.subnet.ip_range = self.ip_range_entry.get()

            if self.subnet.network_address != '' and self.subnet.ip_range != '' and self.subnet.verify_variables(
            ):
                self.subnet.calculate_cidr_from_ip_range()
                steps = self.subnet.cidr_address_steps

                step1 = Label(self.learning_steps, text='Step 1:')
                network = Label(self.learning_steps,
                                text='Network Address: {}'.format(
                                    self.subnet.network_address))
                ip_range = Label(self.learning_steps,
                                 text='Assignable IP Range: {}'.format(
                                     self.subnet.ip_range))
                step2 = Label(self.learning_steps, text='Step 2:')
                front = Label(self.learning_steps,
                              text='Front: {}'.format(steps[0]['Front']))
                back = Label(self.learning_steps,
                             text=' Back: {}'.format(steps[0]['Back']))
                step3 = Label(self.learning_steps, text='Step 3:')
                comparison = Label(self.learning_steps,
                                   text='Comparison: {}'.format(
                                       steps[1]['Comparison']))
                step4 = Label(self.learning_steps, text='Step 4:')
                cidr = Label(self.learning_steps,
                             text='CIDR: {}'.format(steps[2]['CIDR']))
                step5 = Label(self.learning_steps, text='Step 5:')
                cidr_address = Label(self.learning_steps,
                                     text='CIDR Address: {}'.format(
                                         self.subnet.cidr_address))

                step1.grid(row=0, column=0, sticky='W')
                network.grid(row=0, column=1, sticky='W')
                ip_range.grid(row=1, column=1, sticky='W')
                step2.grid(row=2, column=0, sticky='W')
                front.grid(row=2, column=1, sticky='W')
                back.grid(row=3, column=1, sticky='W')
                step3.grid(row=4, column=0, sticky='W')
                comparison.grid(row=4, column=1, columnspan=2, sticky='W')
                step4.grid(row=5, column=0, sticky='W')
                cidr.grid(row=5, column=1, sticky='W')
                step5.grid(row=6, column=0, sticky='W')
                cidr_address.grid(row=6, column=1, sticky='W')
            else:
                self.clear_and_reset()
                HelpGUI()

    def clear_and_reset(self):
        self.net_address_entry.delete(0, 'end')
        self.netmask_entry.delete(0, 'end')
        self.net_address_entry2.delete(0, 'end')
        self.ip_range_entry.delete(0, 'end')
        self.subnet.reset_variables()
        for child in self.learning_steps.winfo_children():
            child.destroy()

    def generate_main(self):
        self.cidr.mainloop()