Пример #1
0
    def setUp(self):
        """Verifies that each of the model classes derived from declarative_base can be created"""

        Base.metadata.create_all(bind=Session.bind)

        # Should be bound to the session we've set up in __init__
        self._service = DatasetService()

        user = User()
        user.name = "testing"

        another_user = User()
        another_user.name = "Someone else"

        session = Session()
        session.add(user)
        session.add(another_user)
        session.flush()

        self._user_id = user.id
        self._another_user_id = another_user.id

        session.commit()
        session.close()

        self._populate_session()
Пример #2
0
    def test_app_user_can_create(self):
        """Can the application user perform inserts?"""

        with session_scope(Session) as session:

            user = User()
            user.name = "Test User"
            session.add(user)

        with session_scope(Session) as another_session:

            count = another_session.query(User).count()
            self.assertEqual(count, 1, "Expected a single user")
Пример #3
0
    def test_create_user(self):

        # Important - don't instantiate the mock class,
        # as the session creation function in the service
        # will do that for us

        sample_user = User()
        sample_user.username = "******"
        sample_user.name = "Test User"
        sample_user.email = "*****@*****.**"
        sample_user.access_level = "External"

        self._mock_session.add = MagicMock()
        self._mock_session.commit = MagicMock()

        user_service = UserService(self._mock_session)
        user_service.create(sample_user.username, sample_user.name, sample_user.email, sample_user.access_level)

        self._mock_session.add.assert_called_once_with(ANY)
        self._mock_session.commit.assert_called_once_with()
Пример #4
0
    def create(self, username, first_name, last_name, email, access_level):
        """Creates a user (if the user doesn't already exist)
            Params:
                username: The login name of the user
                first_name: User's first name
                last_name: User's last name
                email: User's email address
                access_level: Set to 'Admin' for administrative functions
        """

        with self.transaction_scope() as session:

            user = User()
            user.username = username
            user.name = " ".join([first_name, last_name])
            user.email = email
            user.access_level = access_level
            user.first_name = first_name
            user.last_name = last_name

            session.add(user)
Пример #5
0
    def _populate_session(self):

        with self._service.transaction_scope() as session:

            user = User()
            user.username = "******"
            user.name = "Test User"
            user.email = "*****@*****.**"
            user.access_level = "CEH"

            session.add(user)

            pointDst = DatasetType()
            pointDst.type = "Point"

            coverDst = DatasetType()
            coverDst.type = "Coverage"

            resultDst = DatasetType()
            resultDst.type = "Result"

            session.add(pointDst)
            session.add(coverDst)
            session.add(resultDst)

            dataset_a = Dataset()
            dataset_a.dataset_type = pointDst
            dataset_a.viewable_by_user_id = self._user_id
            dataset_a.name = "Dataset1"

            session.add(dataset_a)

            dataset_b = Dataset()
            dataset_b.dataset_type = pointDst
            dataset_b.name = "Dataset2"

            session.add(dataset_b)

            dataset_c = Dataset()
            dataset_c.dataset_type = pointDst
            dataset_c.viewable_by_user_id = self._another_user_id
            dataset_c.name = "Dataset3"

            session.add(dataset_c)

            dataset_d = Dataset()
            dataset_d.dataset_type = resultDst
            dataset_d.name = "Results Dataset 1"
            dataset_d.viewable_by_user_id = 1

            session.add(dataset_d)

            analysis_a = Analysis()
            analysis_a.point_dataset = dataset_a
            analysis_a.coverage_datasets.append(AnalysisCoverageDataset(dataset_b))
            analysis_a.viewable_by = self._user_id
            analysis_a.result_dataset = dataset_d
            analysis_a.deleted = False

            analysis_b = Analysis()
            analysis_b.point_dataset = dataset_a
            analysis_b.coverage_datasets.append(AnalysisCoverageDataset(dataset_b))
            analysis_b.run_by = self._user_id
            analysis_b.result_dataset = dataset_d
            analysis_b.deleted = False

            analysis_c = Analysis()
            analysis_c.point_dataset = dataset_a
            analysis_c.coverage_datasets.append(AnalysisCoverageDataset(dataset_b))
            analysis_c.viewable_by = self._another_user_id
            analysis_c.result_dataset = dataset_d
            analysis_c.deleted = False

            session.add(analysis_a)
            session.add(analysis_b)
            session.add(analysis_c)
Пример #6
0
def setup_app(command, conf, vars):
    """Place any commands to setup ecomaps here - currently creating db tables"""

    # Don't reload the app if it was loaded under the testing environment
    if not pylons.test.pylonsapp:
        load_environment(conf.global_conf, conf.local_conf)

    # Create the tables if they don't already exist
    Base.metadata.drop_all(bind=Session.bind)
    Base.metadata.create_all(bind=Session.bind)

    with session_scope(Session) as session:

        user = User()
        user.name = "Phil Jenkins"
        user.first_name = "Phil"
        user.last_name = "Jenkins"
        user.username = "******"
        user.email = "*****@*****.**"
        user.access_level = "Admin"

        session.add(user)

        user2 = User()
        user2.name = "Mike Wilson"
        user2.first_name = "Mike"
        user2.last_name = "Wilson"
        user2.username = "******"
        user2.email = "*****@*****.**"
        user2.access_level = "Admin"

        session.add(user2)

        # Model that provides the interface to the R code
        model = Model()
        model.name = "LCM Thredds Model"
        model.id = 1
        model.description = "LCM Thredds model written in R"
        model.code_path = "code_root"

        session.add(model)

        pointDst = DatasetType()
        pointDst.type = "Point"

        coverDst = DatasetType()
        coverDst.type = "Coverage"

        resultDst = DatasetType()
        resultDst.type = "Result"

        session.add(pointDst)
        session.add(coverDst)
        session.add(resultDst)

        # Define a datasetType lookup. This will conver the possible thredds
        # datasets into their EcoMaps equivalents.
        datasetTypes = {"GRID": coverDst, "POINT": pointDst}

        # Populate from thredds
        registerThreddsDatasets("http://thredds.ceh.ac.uk/thredds/ecomaps.xml", datasetTypes, session)