3 files changed, 390 insertions, 0 deletions
diff --git a/src/booking/tests/__init__.py b/src/booking/tests/__init__.py
new file mode 100644
index 0000000..b6fef6c
--- /dev/null
+++ b/src/booking/tests/__init__.py
@@ -0,0 +1,8 @@
+##############################################################################
+# Copyright (c) 2016 Max Breitenfeldt and others.
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+# http://www.apache.org/licenses/LICENSE-2.0
+##############################################################################
diff --git a/src/booking/tests/test_models.py b/src/booking/tests/test_models.py
new file mode 100644
index 0000000..6170295
--- /dev/null
+++ b/src/booking/tests/test_models.py
@@ -0,0 +1,232 @@
+##############################################################################
+# Copyright (c) 2016 Max Breitenfeldt and others.
+# Copyright (c) 2018 Parker Berberian, Sawyer Bergeron, and others.
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+# http://www.apache.org/licenses/LICENSE-2.0
+##############################################################################
+
+
+from datetime import timedelta
+
+from django.contrib.auth.models import Permission, User
+from django.test import TestCase
+from django.utils import timezone
+
+# from booking.models import *
+from booking.models import Booking
+from resource_inventory.models import ResourceBundle, GenericResourceBundle, ConfigBundle
+
+
+class BookingModelTestCase(TestCase):
+
+    count = 0
+
+    def setUp(self):
+        self.owner = User.objects.create(username='owner')
+
+        self.res1 = ResourceBundle.objects.create(
+            template=GenericResourceBundle.objects.create(
+                name="gbundle" + str(self.count)
+            )
+        )
+        self.count += 1
+        self.res2 = ResourceBundle.objects.create(
+            template=GenericResourceBundle.objects.create(
+                name="gbundle2" + str(self.count)
+            )
+        )
+        self.count += 1
+        self.user1 = User.objects.create(username='user1')
+
+        self.add_booking_perm = Permission.objects.get(codename='add_booking')
+        self.user1.user_permissions.add(self.add_booking_perm)
+
+        self.user1 = User.objects.get(pk=self.user1.id)
+        self.config_bundle = ConfigBundle.objects.create(
+            owner=self.user1,
+            name="test config"
+        )
+
+    def test_start_end(self):
+        """
+        if the start of a booking is greater or equal then the end,
+        saving should raise a ValueException
+        """
+        start = timezone.now()
+        end = start - timedelta(weeks=1)
+        self.assertRaises(
+            ValueError,
+            Booking.objects.create,
+            start=start,
+            end=end,
+            resource=self.res1,
+            owner=self.user1,
+            config_bundle=self.config_bundle
+        )
+        end = start
+        self.assertRaises(
+            ValueError,
+            Booking.objects.create,
+            start=start,
+            end=end,
+            resource=self.res1,
+            owner=self.user1,
+            config_bundle=self.config_bundle
+        )
+
+    def test_conflicts(self):
+        """
+        saving an overlapping booking on the same resource
+        should raise a ValueException
+        saving for different resources should succeed
+        """
+        start = timezone.now()
+        end = start + timedelta(weeks=1)
+        self.assertTrue(
+            Booking.objects.create(
+                start=start,
+                end=end,
+                owner=self.user1,
+                resource=self.res1,
+                config_bundle=self.config_bundle
+            )
+        )
+
+        self.assertRaises(
+            ValueError,
+            Booking.objects.create,
+            start=start,
+            end=end,
+            resource=self.res1,
+            owner=self.user1,
+            config_bundle=self.config_bundle
+        )
+
+        self.assertRaises(
+            ValueError,
+            Booking.objects.create,
+            start=start + timedelta(days=1),
+            end=end - timedelta(days=1),
+            resource=self.res1,
+            owner=self.user1,
+            config_bundle=self.config_bundle
+        )
+
+        self.assertRaises(
+            ValueError,
+            Booking.objects.create,
+            start=start - timedelta(days=1),
+            end=end,
+            resource=self.res1,
+            owner=self.user1,
+            config_bundle=self.config_bundle
+        )
+
+        self.assertRaises(
+            ValueError,
+            Booking.objects.create,
+            start=start - timedelta(days=1),
+            end=end - timedelta(days=1),
+            resource=self.res1,
+            owner=self.user1,
+            config_bundle=self.config_bundle
+        )
+
+        self.assertRaises(
+            ValueError,
+            Booking.objects.create,
+            start=start,
+            end=end + timedelta(days=1),
+            resource=self.res1,
+            owner=self.user1,
+            config_bundle=self.config_bundle
+        )
+
+        self.assertRaises(
+            ValueError,
+            Booking.objects.create,
+            start=start + timedelta(days=1),
+            end=end + timedelta(days=1),
+            resource=self.res1,
+            owner=self.user1,
+            config_bundle=self.config_bundle
+        )
+
+        self.assertTrue(
+            Booking.objects.create(
+                start=start - timedelta(days=1),
+                end=start,
+                owner=self.user1,
+                resource=self.res1,
+                config_bundle=self.config_bundle
+            )
+        )
+
+        self.assertTrue(
+            Booking.objects.create(
+                start=end,
+                end=end + timedelta(days=1),
+                owner=self.user1,
+                resource=self.res1,
+                config_bundle=self.config_bundle
+            )
+        )
+
+        self.assertTrue(
+            Booking.objects.create(
+                start=start - timedelta(days=2),
+                end=start - timedelta(days=1),
+                owner=self.user1,
+                resource=self.res1,
+                config_bundle=self.config_bundle
+            )
+        )
+
+        self.assertTrue(
+            Booking.objects.create(
+                start=end + timedelta(days=1),
+                end=end + timedelta(days=2),
+                owner=self.user1,
+                resource=self.res1,
+                config_bundle=self.config_bundle
+            )
+        )
+
+        self.assertTrue(
+            Booking.objects.create(
+                start=start,
+                end=end,
+                owner=self.user1,
+                resource=self.res2,
+                config_bundle=self.config_bundle
+            )
+        )
+
+    def test_extensions(self):
+        """
+        saving a booking with an extended end time is allows to happen twice,
+        and each extension must be a maximum of one week long
+        """
+        start = timezone.now()
+        end = start + timedelta(weeks=1)
+        self.assertTrue(
+            Booking.objects.create(
+                start=start,
+                end=end,
+                owner=self.user1,
+                resource=self.res1,
+                config_bundle=self.config_bundle
+            )
+        )
+
+        booking = Booking.objects.all().first()  # should be only thing in db
+
+        self.assertEquals(booking.ext_count, 2)
+        booking.end = booking.end + timedelta(days=3)
+        try:
+            booking.save()
+        except Exception:
+            self.fail("save() threw an exception")
diff --git a/src/booking/tests/test_quick_booking.py b/src/booking/tests/test_quick_booking.py
new file mode 100644
index 0000000..e445860
--- /dev/null
+++ b/src/booking/tests/test_quick_booking.py
@@ -0,0 +1,150 @@
+##############################################################################
+# Copyright (c) 2018 Parker Berberian, Sawyer Bergeron, and others.
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+# http://www.apache.org/licenses/LICENSE-2.0
+##############################################################################
+
+import datetime
+
+from django.test import TestCase, Client
+
+from booking.models import Booking
+from dashboard.testing_utils import (
+    make_host,
+    make_user,
+    make_user_profile,
+    make_lab,
+    make_installer,
+    make_image,
+    make_scenario,
+    make_os,
+    make_complete_host_profile,
+    make_opnfv_role,
+    make_public_net,
+)
+
+
+class QuickBookingValidFormTestCase(TestCase):
+    @classmethod
+    def setUpTestData(cls):
+        cls.user = make_user(False, username="newtestuser", password="testpassword")
+        make_user_profile(cls.user, True)
+
+        lab_user = make_user(True)
+        cls.lab = make_lab(lab_user)
+
+        cls.host_profile = make_complete_host_profile(cls.lab)
+        cls.scenario = make_scenario()
+        cls.installer = make_installer([cls.scenario])
+        os = make_os([cls.installer])
+        cls.image = make_image(cls.lab, 1, cls.user, os, cls.host_profile)
+        cls.host = make_host(cls.host_profile, cls.lab)
+        cls.role = make_opnfv_role()
+        cls.pubnet = make_public_net(10, cls.lab)
+
+        cls.post_data = cls.build_post_data()
+        cls.client = Client()
+
+    @classmethod
+    def build_post_data(cls):
+        return {
+            'filter_field': '{"hosts":[{"host_' + str(cls.host_profile.id) + '":"true"}], "labs": [{"lab_' + str(cls.lab.lab_user.id) + '":"true"}]}',
+            'purpose': 'my_purpose',
+            'project': 'my_project',
+            'length': '3',
+            'ignore_this': 1,
+            'users': '',
+            'hostname': 'my_host',
+            'image': str(cls.image.id),
+            'installer': str(cls.installer.id),
+            'scenario': str(cls.scenario.id)
+        }
+
+    def post(self, changed_fields={}):
+        payload = self.post_data.copy()
+        payload.update(changed_fields)
+        response = self.client.post('/booking/quick/', payload)
+        return response
+
+    def setUp(self):
+        self.client.login(username=self.user.username, password="testpassword")
+
+    def assertValidBooking(self, booking):
+        self.assertEqual(booking.owner, self.user)
+        self.assertEqual(booking.purpose, 'my_purpose')
+        self.assertEqual(booking.project, 'my_project')
+        delta = booking.end - booking.start
+        delta -= datetime.timedelta(days=3)
+        self.assertLess(delta, datetime.timedelta(minutes=1))
+
+        resource_bundle = booking.resource
+        config_bundle = booking.config_bundle
+
+        opnfv_config = config_bundle.opnfv_config.first()
+        self.assertEqual(self.installer, opnfv_config.installer)
+        self.assertEqual(self.scenario, opnfv_config.scenario)
+
+        host = resource_bundle.hosts.first()
+        self.assertEqual(host.profile, self.host_profile)
+        self.assertEqual(host.template.resource.name, 'my_host')
+
+    def test_with_too_long_length(self):
+        response = self.post({'length': '22'})
+
+        self.assertEqual(response.status_code, 200)
+        self.assertIsNone(Booking.objects.first())
+
+    def test_with_negative_length(self):
+        response = self.post({'length': '-1'})
+
+        self.assertEqual(response.status_code, 200)
+        self.assertIsNone(Booking.objects.first())
+
+    def test_with_invalid_installer(self):
+        response = self.post({'installer': str(self.installer.id + 100)})
+
+        self.assertEqual(response.status_code, 200)
+        self.assertIsNone(Booking.objects.first())
+
+    def test_with_invalid_scenario(self):
+        response = self.post({'scenario': str(self.scenario.id + 100)})
+
+        self.assertEqual(response.status_code, 200)
+        self.assertIsNone(Booking.objects.first())
+
+    def test_with_invalid_host_id(self):
+        response = self.post({'filter_field': '{"hosts":[{"host_' + str(self.host_profile.id + 100) + '":"true"}], "labs": [{"lab_' + str(self.lab.lab_user.id) + '":"true"}]}'})
+
+        self.assertEqual(response.status_code, 200)
+        self.assertIsNone(Booking.objects.first())
+
+    def test_with_invalid_lab_id(self):
+        response = self.post({'filter_field': '{"hosts":[{"host_' + str(self.host_profile.id) + '":"true"}], "labs": [{"lab_' + str(self.lab.lab_user.id + 100) + '":"true"}]}'})
+
+        self.assertEqual(response.status_code, 200)
+        self.assertIsNone(Booking.objects.first())
+
+    def test_with_invalid_empty_filter_field(self):
+        response = self.post({'filter_field': ''})
+
+        self.assertEqual(response.status_code, 200)
+        self.assertIsNone(Booking.objects.first())
+
+    def test_with_garbage_users_field(self):  # expected behavior: treat as though field is empty if it has garbage data
+        response = self.post({'users': 'X�]QP�槰DP�+m���h�U�_�yJA:.rDi��QN|.��C��n�P��F!��D�����5ȅj�9�LV��'})  # output from /dev/urandom
+
+        self.assertEqual(response.status_code, 200)
+        booking = Booking.objects.first()
+        self.assertIsNotNone(booking)
+        self.assertValidBooking(booking)
+
+    def test_with_valid_form(self):
+        response = self.post()
+
+        self.assertEqual(response.status_code, 200)
+        booking = Booking.objects.first()
+        self.assertIsNotNone(booking)
+        self.assertValidBooking(booking)