Initial push to public repo

netbox/dcim/tests/test_forms.py

@@ -0,0 +1,71 @@
+from django.test import TestCase
+from dcim.forms import *
+from dcim.models import *
+
+
+def get_id(model, slug):
+    return model.objects.get(slug=slug).id
+
+
+class DeviceTestCase(TestCase):
+
+    fixtures = ['dcim', 'ipam']
+
+    def test_racked_device(self):
+        test = DeviceForm(data={
+            'device_role': get_id(DeviceRole, 'leaf-switch'),
+            'name': 'test',
+            'site': get_id(Site, 'test1'),
+            'face': RACK_FACE_FRONT,
+            'platform': get_id(Platform, 'juniper-junos'),
+            'device_type': get_id(DeviceType, 'qfx5100-48s'),
+            'position': 41,
+            'rack': '1',
+            'manufacturer': get_id(Manufacturer, 'juniper'),
+        })
+        self.assertTrue(test.is_valid(), test.fields['position'].choices)
+        self.assertTrue(test.save())
+
+    def test_racked_device_occupied(self):
+        test = DeviceForm(data={
+            'device_role': get_id(DeviceRole, 'leaf-switch'),
+            'name': 'test',
+            'site': get_id(Site, 'test1'),
+            'face': RACK_FACE_FRONT,
+            'platform': get_id(Platform, 'juniper-junos'),
+            'device_type': get_id(DeviceType, 'qfx5100-48s'),
+            'position': 1,
+            'rack': '1',
+            'manufacturer': get_id(Manufacturer, 'juniper'),
+        })
+        self.assertFalse(test.is_valid())
+
+    def test_non_racked_device(self):
+        test = DeviceForm(data={
+            'device_role': get_id(DeviceRole, 'pdu'),
+            'name': 'test',
+            'site': get_id(Site, 'test1'),
+            'face': None,
+            'platform': None,
+            'device_type': get_id(DeviceType, 'cwg-24vym415c9'),
+            'position': None,
+            'rack': '1',
+            'manufacturer': get_id(Manufacturer, 'servertech'),
+        })
+        self.assertTrue(test.is_valid())
+        self.assertTrue(test.save())
+
+    def test_non_racked_device_with_face(self):
+        test = DeviceForm(data={
+            'device_role': get_id(DeviceRole, 'pdu'),
+            'name': 'test',
+            'site': get_id(Site, 'test1'),
+            'face': RACK_FACE_REAR,
+            'platform': None,
+            'device_type': get_id(DeviceType, 'cwg-24vym415c9'),
+            'position': None,
+            'rack': '1',
+            'manufacturer': get_id(Manufacturer, 'servertech'),
+        })
+        self.assertTrue(test.is_valid())
+        self.assertTrue(test.save())

netbox/dcim/tests/test_models.py

@@ -0,0 +1,96 @@
+from django.test import TestCase
+from dcim.models import *
+
+
+class RackTestCase(TestCase):
+
+    def setUp(self):
+
+        site = Site.objects.create(
+            name='TestSite1',
+            slug='my-test-site'
+        )
+        self.rack = Rack.objects.create(
+            name='TestRack1',
+            facility_id='A101',
+            site=site,
+            u_height=42
+        )
+        self.manufacturer = Manufacturer.objects.create(
+            name='Acme',
+            slug='acme'
+        )
+
+        self.device_type = {
+            'ff2048': DeviceType.objects.create(
+                manufacturer=self.manufacturer,
+                model='FrameForwarder 2048',
+                slug='ff2048'
+            ),
+            'cc5000': DeviceType.objects.create(
+                manufacturer=self.manufacturer,
+                model='CurrentCatapult 5000',
+                slug='cc5000',
+                u_height=0
+            ),
+        }
+        self.role = {
+            'Server': DeviceRole.objects.create(
+                name='Server',
+                slug='server',
+            ),
+            'Switch': DeviceRole.objects.create(
+                name='Switch',
+                slug='switch',
+            ),
+            'Console Server': DeviceRole.objects.create(
+                name='Console Server',
+                slug='console-server',
+            ),
+            'PDU': DeviceRole.objects.create(
+                name='PDU',
+                slug='pdu',
+            ),
+
+        }
+
+    def test_mount_single_device(self):
+
+        rack1 = Rack.objects.get(name='TestRack1')
+        device1 = Device(
+            name='TestSwitch1',
+            device_type=DeviceType.objects.get(manufacturer__slug='acme', slug='ff2048'),
+            device_role=DeviceRole.objects.get(slug='switch'),
+            rack=rack1,
+            position=10,
+            face=RACK_FACE_REAR,
+            )
+        device1.save()
+
+        # Validate rack height
+        self.assertEqual(list(rack1.units), list(reversed(range(1, 43))))
+
+        # Validate inventory (front face)
+        rack1_inventory_front = rack1.get_front_elevation()
+        self.assertEqual(rack1_inventory_front[-10]['device'], device1)
+        del(rack1_inventory_front[-10])
+        for u in rack1_inventory_front:
+            self.assertIsNone(u['device'])
+
+        # Validate inventory (rear face)
+        rack1_inventory_rear = rack1.get_rear_elevation()
+        self.assertEqual(rack1_inventory_rear[-10]['device'], device1)
+        del(rack1_inventory_rear[-10])
+        for u in rack1_inventory_rear:
+            self.assertIsNone(u['device'])
+
+    def test_mount_zero_ru(self):
+        pdu = Device.objects.create(
+            name='TestPDU',
+            device_role=self.role.get('PDU'),
+            device_type=self.device_type.get('cc5000'),
+            rack=self.rack,
+            position=None,
+            face=None,
+        )
+        self.assertTrue(pdu)