test.erp5.testShaDirExternal.py 6.17 KB
# -*- coding: utf-8 -*-
##############################################################################
#
# Copyright (c) 2011 Nexedi SA and Contributors. All Rights Reserved.
#                    Lucas Carvalho <lucas@nexedi.com>
#
# WARNING: This program as such is intended to be used by professional
# programmers who take the whole responsability of assessing all potential
# consequences resulting from its eventual inadequacies and bugs
# End users who are looking for a ready-to-use solution with commercial
# garantees and support are strongly adviced to contract a Free Software
# Service Company
#
# This program is Free Software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
#
##############################################################################


import base64
import json
import os
import httplib
from DateTime import DateTime
from Products.ERP5Type.tests.ERP5TypeLiveTestCase import ERP5TypeTestCase
from erp5.component.test.ShaDirMixin import ShaDirMixin
from erp5.component.test.ShaSecurityMixin import ShaSecurityMixin


class TestShaDirExternal(ShaDirMixin, ShaSecurityMixin, ERP5TypeTestCase):
  """
    ShaDir - HTTP Information Cache server
    We must simulate the real usage of ShaDir using httplib.
  """

  def getTitle(self):
    """
      Return the title of the current test set.
    """
    return "SHADIR External - Real Usage Of ShaDir"

  def afterSetUp(self):
    """
      Initialize the ERP5 site.
    """
    self.group = 'shadir'
    ShaDirMixin.afterSetUp(self)
    ShaSecurityMixin.afterSetUp(self)

    # Define POST headers with Authentication
    self.content_type =  'application/json'
    authentication_string = '%s:%s' % (self.lucas_user, self.lucas_user)
    base64string = base64.encodestring(authentication_string).strip()
    self.header_dict = {'Authorization': 'Basic %s' % base64string,
                        'Content-Type': self.content_type}

    # HTTP Connection properties
    self.host = self.portal.REQUEST.get('SERVER_NAME')
    self.port = self.portal.REQUEST.get('SERVER_PORT')
    self.path = os.path.join(self.shadir.getPath(), self.key)

  def test_external_post(self):
    """
      Test the external usage to POST information
    """
    now = DateTime()
    connection = httplib.HTTPConnection('%s:%s' % (self.host, self.port))
    try:
      connection.request('PUT', self.path, self.data, self.header_dict)
      result = connection.getresponse()
      self.tic()
      data = result.read()
    finally:
      connection.close()
    self.assertEquals('', data)
    self.assertEquals(201, result.status)

    # Check Data Set
    data_set = self.portal.portal_catalog.getResultValue(
                              portal_type='Data Set',
                              reference=self.key)
    self.assertNotEquals(None, data_set)
    self.assertEquals('Published', data_set.getValidationStateTitle())

    # Check Document
    document = self.portal.portal_catalog.getResultValue(portal_type='File',
                                                reference=self.sha512sum,
                                                creation_date=' >= "%s"' % now)
    self.assertNotEquals(None, document)
    self.assertEquals(self.data, document.getData())
    self.assertEquals(str(self.expiration_date),
                               str(document.getExpirationDate()))
    self.assertEquals(data_set, document.getFollowUpValue())
    self.assertEquals('File', document.getPortalType())
    self.assertEquals('Published', document.getValidationStateTitle())

  def test_external_get(self, annonymous=False):
    """
      Test the external usage to retrive the information
    """
    self.test_external_post()
    header_dict = {}
    if not annonymous:
      header_dict = self.header_dict

    connection = httplib.HTTPConnection('%s:%s' % (self.host, self.port))
    try:
      connection.request('GET', self.path, headers=header_dict)
      result = connection.getresponse()
      data = result.read()
    finally:
      connection.close()
    self.assertEquals(json.dumps([json.loads(self.data)]), data)
    self.assertEquals(200, result.status)
    self.assertEquals(self.content_type, result.getheader("content-type"))

  def test_external_get_anonymous(self):
    """
      Test the external usage to retrive information
      accessing as Anonymous User.
    """
    self.test_external_get(annonymous=True)

  def test_external_post_anonymous(self):
    """
    """
    connection = httplib.HTTPConnection('%s:%s' % (self.host, self.port))
    header_dict = {'Content-Type': self.content_type}
    try:
      connection.request('PUT', self.path, self.data, header_dict)
      result = connection.getresponse()
      self.tic()
    finally:
      connection.close()
    self.assertEquals(302, result.status)

  def test_external_post_with_wrong_data(self):
    """
      The data which is sent to the server must follow a JSON schema.
      If the data does not follow the schema it must return the error.
    """
    # Removing a required property
    data = json.loads(self.data)
    data[0] = json.loads(data[0])
    data[0].pop('file')
    data[0] = json.dumps(data[0])
    data = json.dumps(data)

    connection = httplib.HTTPConnection('%s:%s' % (self.host, self.port))
    try:
      connection.request('PUT', self.path, data, self.header_dict)
      result = connection.getresponse()
      self.tic()
      data = result.read()
    finally:
      connection.close()
    self.assertTrue("Required field 'file' is missing" in data, data)
    self.assertEquals(500, result.status)
    self.assertEquals('text/html; charset=utf-8',
                                         result.getheader("content-type"))