<gh_stars>1-10
import mock
import re
import os

from dmutils.user import User
from dmutils.forms import FakeCsrf
from flask import json

from app import create_app
from app import login_manager
from unittest import TestCase


class BaseApplicationTest(TestCase):
    def setUp(self):
        self.app = create_app('test')
        self.client = self.app.test_client()

        self._s3_patch = mock.patch('app.main.views.services.S3')
        self.s3 = self._s3_patch.start()
        self.s3.return_value = mock.Mock(
            bucket_name="digitalmarketplace-documents-testing-testing",
            bucket_short_name="documents")

        self._default_suffix_patch = mock.patch(
            'dmutils.documents.default_file_suffix',
            return_value='2015-01-01-1200'
        )
        self._default_suffix_patch.start()

    def tearDown(self):
        self._s3_patch.stop()
        self._default_suffix_patch.stop()

    def load_example_listing(self, name):
        file_path = os.path.join("example_responses", "{}.json".format(name))
        with open(file_path) as f:
            return json.load(f)

    @staticmethod
    def strip_all_whitespace(content):
        pattern = re.compile(r'\s+')
        return re.sub(pattern, '', content)


class Response:
    def __init__(self, status_code=503, message=None):
        self._status_code = status_code
        self._message = message

    @property
    def message(self):
        return self._message

    @property
    def status_code(self):
        return self._status_code


class LoggedInApplicationTest(BaseApplicationTest):
    user_role = 'admin'

    def setUp(self):
        super(LoggedInApplicationTest, self).setUp()

        patch_config = {
            'authenticate_user.return_value': {
                'users': {
                    'id': 1234,
                    'emailAddress': '<EMAIL>',
                    'role': 'admin',
                    'locked': False,
                    'passwordChangedAt': '2015-01-01T00:00:00Z',
                    'active': True,
                    'name': 'tester',
                    'termsAcceptedAt': '2015-01-01T00:00:00.0Z',
                }
            }
        }
        self._data_api_client = mock.patch(
            'app.main.views.login.data_api_client',
            **patch_config
        )
        self._data_api_client.start()
        res = self.client.post('/admin/login', data={
            'email_address': '<EMAIL>',
            'password': '<PASSWORD>',
            'csrf_token': FakeCsrf.valid_token,
        })
        assert res.status_code < 400

        self._user_callback = login_manager.user_callback
        login_manager.user_loader(self.user_loader)

    def user_loader(self, user_id):
        if user_id:
            return User(
                user_id, '<EMAIL>', None, None, False, True, 'tester', self.user_role, '2015-01-01T00:00:00.0Z'
            )

    def tearDown(self):
        self._data_api_client.stop()
        login_manager.user_loader(self._user_callback)

    def _replace_whitespace(self, string, replacement_substring=""):
        # Replace all runs of whitespace with replacement_substring
        return re.sub(r"\s+", replacement_substring, string)

import mock
import re
import os

from dmutils.user import User
from dmutils.forms import FakeCsrf
from flask import json

from app import create_app
from app import login_manager
from unittest import TestCase


class BaseApplicationTest(TestCase):
    def setUp(self):
        self.app = create_app('test')
        self.client = self.app.test_client()

        self._s3_patch = mock.patch('app.main.views.services.S3')
        self.s3 = self._s3_patch.start()
        self.s3.return_value = mock.Mock(
            bucket_name="digitalmarketplace-documents-testing-testing",
            bucket_short_name="documents")

        self._default_suffix_patch = mock.patch(
            'dmutils.documents.default_file_suffix',
            return_value='2015-01-01-1200'
        )
        self._default_suffix_patch.start()

    def tearDown(self):
        self._s3_patch.stop()
        self._default_suffix_patch.stop()

    def load_example_listing(self, name):
        file_path = os.path.join("example_responses", "{}.json".format(name))
        with open(file_path) as f:
            return json.load(f)

    @staticmethod
    def strip_all_whitespace(content):
        pattern = re.compile(r'\s+')
        return re.sub(pattern, '', content)


class Response:
    def __init__(self, status_code=503, message=None):
        self._status_code = status_code
        self._message = message

    @property
    def message(self):
        return self._message

    @property
    def status_code(self):
        return self._status_code


class LoggedInApplicationTest(BaseApplicationTest):
    user_role = 'admin'

    def setUp(self):
        super(LoggedInApplicationTest, self).setUp()

        patch_config = {
            'authenticate_user.return_value': {
                'users': {
                    'id': 1234,
                    'emailAddress': '<EMAIL>',
                    'role': 'admin',
                    'locked': False,
                    'passwordChangedAt': '2015-01-01T00:00:00Z',
                    'active': True,
                    'name': 'tester',
                    'termsAcceptedAt': '2015-01-01T00:00:00.0Z',
                }
            }
        }
        self._data_api_client = mock.patch(
            'app.main.views.login.data_api_client',
            **patch_config
        )
        self._data_api_client.start()
        res = self.client.post('/admin/login', data={
            'email_address': '<EMAIL>',
            'password': '<PASSWORD>',
            'csrf_token': FakeCsrf.valid_token,
        })
        assert res.status_code < 400

        self._user_callback = login_manager.user_callback
        login_manager.user_loader(self.user_loader)

    def user_loader(self, user_id):
        if user_id:
            return User(
                user_id, '<EMAIL>', None, None, False, True, 'tester', self.user_role, '2015-01-01T00:00:00.0Z'
            )

    def tearDown(self):
        self._data_api_client.stop()
        login_manager.user_loader(self._user_callback)

    def _replace_whitespace(self, string, replacement_substring=""):
        # Replace all runs of whitespace with replacement_substring
        return re.sub(r"\s+", replacement_substring, string)
# Replace all runs of whitespace with replacement_substring