pacifica/pacifica-proxy

#!/usr/bin/python
# -*- coding: utf-8 -*-
"""Test the files proxy object."""
import os
import requests
from cherrypy.test import helper
from pacifica.proxy.config import get_config
from .common_test import CommonCPSetup


class TestFilesObject(helper.CPWebCase, CommonCPSetup):
    """Test the files proxy server."""

    NGINX_X_ACCEL_PORT = 8123
    PORT = 8180
    HOST = '127.0.0.1'
    headers = [('Content-Type', 'application/json')]

    def test_files(self):
        """Test for a file."""
        md_url = get_config().get('metadata', 'url')
        resp = requests.get(
            '{}/files'.format(md_url),
            params={'_id': 104}
        )
        self.assertEqual(resp.status_code, 200)
        files = resp.json()
        self.assertTrue(len(files) > 0)
        the_file = files[0]
        url = '/files/{0}/{1}'.format(the_file['hashtype'],
                                      the_file['hashsum'])
        self.getPage(url)
        self.assertStatus('200 OK')
        self.assertTrue(len(self.body) == the_file['size'])

    def test_files_not_found(self):
        """Test for a file that doesn't exist."""
        url = '/files/sha256/somethingthatisnotthere'
        self.getPage(url)
        self.assertStatus('404 Not Found')
        self.assertTrue(len(self.body) > -1)

    def test_files_nginx(self):
        """Test for the nginx headers of we are doing nginx proxy."""
        os.environ['NGINX_ACCEL'] = 'true'
        resp = requests.get(
            '{}/files'.format(get_config().get('metadata', 'url')),
            params={'_id': 104}
        )
        self.assertEqual(resp.status_code, 200)
        files = resp.json()
        self.assertTrue(len(files) > 0)
        the_file = files[0]
        url = '/files/{0}/{1}'.format(the_file['hashtype'],
                                      the_file['hashsum'])
        self.getPage(url)
        self.assertHeader('X-Accel-Redirect',
                          '/archivei_accel/{0}'.format(the_file['_id']))
        resp = requests.get('http://localhost:8123{0}'.format(url))
        self.assertTrue(resp.status_code == 200)
        os.environ['NGINX_ACCEL'] = 'false'