xos/core/tests.py

#!/usr/bin/env python
from django.test import TestCase
from django.contrib.auth.models import User
from rest_framework import status
from rest_framework.test import APIClient
from rest_framework.test import APITestCase
import json
from django.forms.models import model_to_dict

from core.models import *

print "-------------------------- Let's test!!!!!!!! --------------------------"

from django.apps import apps
from django.test.client import Client
from django.test import testcases
from django.http import SimpleCookie, HttpRequest, QueryDict
from importlib import import_module
from django.conf import settings
class FixedClient(Client):
    def login(self, **credentials):
        """
        Sets the Factory to appear as if it has successfully logged into a site.

        Returns True if login is possible; False if the provided credentials
        are incorrect, or the user is inactive, or if the sessions framework is
        not available.
        """
        from django.contrib.auth import authenticate, login
        user = authenticate(**credentials)
        if (user and user.is_active and
                apps.is_installed('django.contrib.sessions')):
            engine = import_module(settings.SESSION_ENGINE)

            # Create a fake request to store login details.
            request = HttpRequest()

            # XOS's admin.py requires these to be filled in
            request.POST = {"username": credentials["username"],
                            "password": credentials["password"]}

            if self.session:
                request.session = self.session
            else:
                request.session = engine.SessionStore()
            login(request, user)

            # Save the session values.
            request.session.save()

            # Set the cookie to represent the session.
            session_cookie = settings.SESSION_COOKIE_NAME
            self.cookies[session_cookie] = request.session.session_key
            cookie_data = {
                'max-age': None,
                'path': '/',
                'domain': settings.SESSION_COOKIE_DOMAIN,
                'secure': settings.SESSION_COOKIE_SECURE or None,
                'expires': None,
                }
            self.cookies[session_cookie].update(cookie_data)

            return True
        else:
            return False

class FixedAPITestCase(testcases.TestCase):
    client_class = FixedClient

# Environment Tests - Should pass everytime, if not something in the config is broken.
class SimpleTest(TestCase):
    fixtures = []

    def test_basic_addition(self):
        """
        Tests that 1 + 1 always equals 2.
        """
        self.assertEqual(1 + 1, 2)


# Site Test
class SiteTest(TestCase):
    fixtures = []

    def setUp(self):
        Site.objects.create(
            name="Test Site",
            login_base="test_"
        )

    def test_read_site(self):
        """
        Should read a site in the DB.
        """
        site = Site.objects.get(name="Test Site")
        # print(site._meta.get_all_field_names())
        self.assertEqual(site.login_base, "test_")


class UnautheticatedRequest(FixedAPITestCase):
    fixtures = []

    def test_require_authentication(self):
        """
        Ensure that request must be authenticated
        """
        response = self.client.get('/xos/sites/', format='json')
        self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN)


class SiteTestAPI(FixedAPITestCase):
    fixtures = []

    def setUp(self):
        self.site = Site.objects.create(
            name="Test Site",
            login_base="test_"
        )
        self.user = User(
            username='testuser',
            email='test@mail.org',
            password='testing',
            site=self.site,
            is_admin=True
        )
        self.user.save()
        self.client.login(username='test@mail.org', password='testing')

    def test_read_site_API(self):
        """
        Read a Site trough API
        """
        response = self.client.get('/xos/sites/', format='json')
        parsed = json.loads(response.content)
        self.assertEqual(response.status_code, status.HTTP_200_OK)
        self.assertEqual(len(parsed), 1)
        self.assertEqual(parsed[0]['login_base'], 'test_')

    def test_create_site_API(self):
        """
        Create a Site trough API
        """
        data = {
            'name': "Another Test Site",
            'login_base': "another_test_",
            'location': [10, 20],
            'abbreviated_name': 'test'
        }
        response = self.client.post('/xos/sites/', data, format='json')
        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
        self.assertEqual(Site.objects.count(), 2)
        self.assertEqual(Site.objects.filter(name="Another Test Site").count(), 1)


class SliceTestAPI(FixedAPITestCase):
    fixtures = []

    def setUp(self):
        self.site = Site.objects.create(
            name="Test Site",
            login_base="test_"
        )
        self.pi = SiteRole.objects.create(role='pi')
        self.user = User(
            username='testuser',
            email='test@mail.org',
            password='testing',
            site=self.site
        )
        self.user.save()
        self.siteprivileges = SitePrivilege.objects.create(
            user=self.user,
            site=self.site,
            role=self.pi
        )
        self.serviceClass = ServiceClass.objects.create(
            name='Test Service Class'
        )
        self.client.login(username='test@mail.org', password='testing')

    def test_create_site_slice(self):
        """
        Add a slice to a given site
        """
        data = {
            'name': "test_slice",
            'site': self.site.id,
            'serviceClass': self.serviceClass.id
        }
        response = self.client.post('/xos/slices/?no_hyperlinks=1', data, format='json')
        self.assertEqual(response.status_code, status.HTTP_201_CREATED)

    def test_validation_slice_name(self):
        """
        The slice name should start with site.login_base
        curl -H "Accept: application/json; indent=4" -u padmin@vicci.org:letmein 'http://xos:9999/xos/slices/?no_hyperlinks=1' -H "Content-Type: application/json" -X POST --data '{"name": "test", "site":"1", "serviceClass":1}'
        """
        data = {
            'name': "wrong_slice",
            'site': self.site.id,
            'serviceClass': self.serviceClass.id
        }
        response = self.client.post('/xos/slices/?no_hyperlinks=1', data, format='json')
        parsed = json.loads(response.content)
        self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN)
        self.assertEqual(parsed['detail']['specific_error'], "slice name must begin with test_")

    def test_only_admin_can_change_creator(self):
        """
        Only an admin can change the creator of a slice
        """
        slice = Slice.objects.create(
            name="test_slice",
            site=self.site,
            serviceClass=self.serviceClass,
            creator=self.user
        )

        user2 = User(
            username='another_testuser',
            email='another_test@mail.org',
            password='testing',
            site=self.site
        )
        user2.save()

        data = model_to_dict(slice)
        data['creator'] = user2.id

        response = self.client.put('/xos/slices/%s/?no_hyperlinks=1' % slice.id, data, format='json')
        self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN)
        parsed = json.loads(response.content)
        self.assertEqual(parsed['detail']['specific_error'], "Insufficient privileges to change slice creator")

class ServiceTestAPI(FixedAPITestCase):
    fixtures = []

    def setUp(self):
        self.site = Site.objects.create(
            name="Test Site",
            login_base="test_"
        )
        self.admin = User(
            username='testadmin',
            email='admin@mail.org',
            password='testing',
            site=self.site,
            is_admin=True
        )
        self.admin.save()

        self.user = User(
            username='testuser',
            email='user@mail.org',
            password='testing',
            site=self.site
        )
        self.user.save()

        self.service1 = Service.objects.create(
            name="fakeService1",
            versionNumber="1.0.0",
            published=True,
            enabled=True
        )

        self.service2 = Service.objects.create(
            name="fakeService1",
            versionNumber="1.0.0",
            published=True,
            enabled=True
        )

        self.service_role = ServiceRole.objects.create(role='admin')

        self.service_privileges = ServicePrivilege.objects.create(
            user=self.user,
            service=self.service2,
            role=self.service_role
        )

    # TODO
    # [x] admin view all service
    # [ ] user view only service2
    # [x] admin view a specific service service
    # [ ] user view can't view a specific service
    # [ ] user update service2
    # [ ] usercan NOT update service2
    # [x] admin update service1
    def test_admin_read_all_service(self):
        """
        Admin should read all the services
        """
        self.client.login(username='admin@mail.org', password='testing')
        response = self.client.get('/xos/services/', format='json')
        self.assertEqual(response.status_code, status.HTTP_200_OK)
        self.assertEqual(len(json.loads(response.content)), 2)

    # need to understand how slices are related
    def test_user_read_all_service(self):
        """
        User should read only service for which have privileges
        """
        self.client.login(username='user@mail.org', password='testing')
        response = self.client.get('/xos/services/', format='json')
        self.assertEqual(response.status_code, status.HTTP_200_OK)
        self.assertEqual(len(json.loads(response.content)), 1)

    def test_admin_read_one_service(self):
        """
        Read a given service
        """
        self.client.login(username='admin@mail.org', password='testing')
        response = self.client.get('/xos/services/%s/' % self.service1.id, format='json')
        parsed = json.loads(response.content)
        self.assertEqual(response.status_code, status.HTTP_200_OK)
        self.assertEqual(parsed['name'], self.service1.name)

    def test_admin_update_service(self):
        """
        Update a given service
        """
        data = model_to_dict(self.service1)
        data['name'] = "newName"

        self.client.login(username='admin@mail.org', password='testing')
        response = self.client.put('/xos/services/%s/' % self.service1.id, data, format='json')
        parsed = json.loads(response.content)
        self.assertEqual(response.status_code, status.HTTP_200_OK)
        model = Service.objects.get(id=self.service1.id)
        self.assertEqual(model.name, data['name'])