コード例 #1
0
ファイル: test_context.py プロジェクト: mikrosimage/tk-core
    def test_from_entity_dictionary(self, get_current_user, from_entity):
        """
        Test context.from_entity_dictionary - this can contruct a context from
        an entity dictionary looking at linked entities where available.

        Falls back to 'from_entity' if the entity dictionary doesn't contain
        everything needed.
        """
        get_current_user.return_value = self.current_user

        # overload from_entity to ensure it causes a test fail if from_entity_dictionary
        # falls back to it:
        from_entity.return_value = {}

        ent_dict = {"type":self.shot["type"], "id":self.shot["id"], "code":self.shot["code"]}
        ent_dict["project"] = self.project

        result = context.from_entity_dictionary(self.tk, ent_dict)
        self.assertIsNotNone(result)

        self.check_entity(self.project, result.project)
        self.assertEquals(3, len(result.project))

        self.check_entity(self.shot, result.entity)
        self.assertEquals(3, len(result.entity))
        
        self.check_entity(self.current_user, result.user)
コード例 #2
0
ファイル: test_context.py プロジェクト: mikrosimage/tk-core
    def test_from_linked_entity_types(self, get_current_user):
        get_current_user.return_value = self.current_user

        # PublishedFile and Version entities are special cased in the context
        # factories. We need to make sure they create a context object that is
        # built from what those entities are linked to, but with the original
        # entity kept as the source_entity of the context.
        result = context.from_entity(self.tk, self.publishedfile["type"], self.publishedfile["id"])
        self.check_entity(self.project, result.project, check_name=False)
        self.check_entity(self.shot, result.entity, check_name=False)
        self.check_entity(self.task, result.task, check_name=False)
        self.check_entity(
            result.source_entity,
            dict(
                type=self.publishedfile["type"],
                id=self.publishedfile["id"],
            ),
            check_name=False,
        )

        result = context.from_entity_dictionary(self.tk, self.publishedfile)
        self.check_entity(self.project, result.project, check_name=False)
        self.check_entity(self.shot, result.entity, check_name=False)
        self.check_entity(self.task, result.task, check_name=False)
        self.check_entity(
            result.source_entity,
            dict(
                type=self.publishedfile["type"],
                id=self.publishedfile["id"],
            ),
            check_name=False,
        )
コード例 #3
0
ファイル: test_context.py プロジェクト: satishgoda/tk-core
    def test_from_entity_dictionary(self, get_current_user, from_entity):
        """
        Test context.from_entity_dictionary - this can contruct a context from
        an entity dictionary looking at linked entities where available.

        Falls back to 'from_entity' if the entity dictionary doesn't contain
        everything needed.
        """
        get_current_user.return_value = self.current_user

        # overload from_entity to ensure it causes a test fail if from_entity_dictionary
        # falls back to it:
        from_entity.return_value = {}

        ent_dict = {
            "type": self.shot["type"],
            "id": self.shot["id"],
            "code": self.shot["code"]
        }
        ent_dict["project"] = self.project

        result = context.from_entity_dictionary(self.tk, ent_dict)
        self.assertIsNotNone(result)

        self.check_entity(self.project, result.project)
        self.assertEquals(3, len(result.project))

        self.check_entity(self.shot, result.entity)
        self.assertEquals(3, len(result.entity))

        self.check_entity(self.current_user, result.user)
コード例 #4
0
ファイル: test_context.py プロジェクト: satishgoda/tk-core
    def test_from_linked_entity_types(self, get_current_user):
        get_current_user.return_value = self.current_user

        # PublishedFile and Version entities are special cased in the context
        # factories. We need to make sure they create a context object that is
        # built from what those entities are linked to, but with the original
        # entity kept as the source_entity of the context.
        result = context.from_entity(self.tk, self.publishedfile["type"],
                                     self.publishedfile["id"])
        self.check_entity(self.project, result.project, check_name=False)
        self.check_entity(self.shot, result.entity, check_name=False)
        self.check_entity(self.task, result.task, check_name=False)
        self.check_entity(
            result.source_entity,
            dict(
                type=self.publishedfile["type"],
                id=self.publishedfile["id"],
            ),
            check_name=False,
        )

        result = context.from_entity_dictionary(self.tk, self.publishedfile)
        self.check_entity(self.project, result.project, check_name=False)
        self.check_entity(self.shot, result.entity, check_name=False)
        self.check_entity(self.task, result.task, check_name=False)
        self.check_entity(
            result.source_entity,
            dict(
                type=self.publishedfile["type"],
                id=self.publishedfile["id"],
            ),
            check_name=False,
        )
コード例 #5
0
ファイル: test_context.py プロジェクト: satishgoda/tk-core
    def test_from_entity_dictionary_additional_entities(
            self, get_current_user, from_entity):
        """
        Test context.from_entity_dictionary - this can contruct a context from
        an entity dictionary looking at linked entities where available.
        
        Falls back to 'from_entity' if the entity dictionary doesn't contain
        everything needed.
        """
        get_current_user.return_value = self.current_user

        # overload from_entity to ensure it causes a test fail if from_entity_dictionary
        # falls back to it:
        from_entity.return_value = {}

        add_value = {"name": "additional", "id": 3, "type": "add_type"}
        ent_dict = {
            "type": "Task",
            "id": self.task["id"],
            "content": self.task["content"],
            "additional_field": add_value
        }
        ent_dict["project"] = self.project
        ent_dict["entity"] = self.shot
        ent_dict["step"] = self.step

        result = context.from_entity_dictionary(self.tk, ent_dict)
        self.assertIsNotNone(result)

        self.check_entity(self.project, result.project)
        self.assertEquals(3, len(result.project))

        self.check_entity(self.shot, result.entity)
        self.assertEquals(3, len(result.entity))

        self.check_entity(self.current_user, result.user)

        self.check_entity(self.step, result.step)
        self.assertEquals(3, len(result.step))

        self.assertEquals(self.task["type"], result.task["type"])
        self.assertEquals(self.task["id"], result.task["id"])
        self.assertEquals(self.task["content"], result.task["name"])
        self.assertEquals(3, len(result.task))
コード例 #6
0
ファイル: test_context.py プロジェクト: mikrosimage/tk-core
    def test_from_entity_dictionary_additional_entities(self, get_current_user, from_entity):
        """
        Test context.from_entity_dictionary - this can contruct a context from
        an entity dictionary looking at linked entities where available.
        
        Falls back to 'from_entity' if the entity dictionary doesn't contain
        everything needed.
        """
        get_current_user.return_value = self.current_user
        
        # overload from_entity to ensure it causes a test fail if from_entity_dictionary
        # falls back to it:
        from_entity.return_value = {}

        add_value = {"name":"additional", "id": 3, "type": "add_type"}
        ent_dict = {"type":"Task", "id":self.task["id"], "content":self.task["content"], "additional_field":add_value}
        ent_dict["project"] = self.project
        ent_dict["entity"] = self.shot
        ent_dict["step"] = self.step

        result = context.from_entity_dictionary(self.tk, ent_dict)
        self.assertIsNotNone(result)

        self.check_entity(self.project, result.project)
        self.assertEquals(3, len(result.project))

        self.check_entity(self.shot, result.entity)
        self.assertEquals(3, len(result.entity))
        
        self.check_entity(self.current_user, result.user)
        
        self.check_entity(self.step, result.step)
        self.assertEquals(3, len(result.step))
        
        self.assertEquals(self.task["type"], result.task["type"])
        self.assertEquals(self.task["id"], result.task["id"])
        self.assertEquals(self.task["content"], result.task["name"])
        self.assertEquals(3, len(result.task))