Python DataController.addContainer方法代码示例

# 需要导入模块: from flashcardapp.datacontroller import DataController [as 别名]
# 或者: from flashcardapp.datacontroller.DataController import addContainer [as 别名]
    def test_addContainer(self):
        """
        Steps this test will run:

        1. Creates a new DataController
        2. Creates a new json string representing the data of a container based
           on the specifications of the DataController's addContainer method.
        3. Passes the json string to the DataController's addContainer method
           and gets the result of the method (response).
        4. Asserts that there is a 'saved' key in the response.
        5. Asserts that the value of 'saved' is True.
        """
        dc = DataController()
        cName = "Test Container"
        clsKey = 1
        numofcontainers = 4
        data = {"title":cName, "classId":clsKey}
        resultStr = dc.addContainer(data, self.user)
        result = json.loads(resultStr)
        self.assertTrue('saved' in result)
        self.assertTrue(result['saved'], result['message'])

        #Check that the associatation was actually made
        clist = Container.objects.filter(classes__Id=clsKey)
        self.assertEqual(len(clist), numofcontainers)

        cls = Class.objects.get(pk=clsKey)
        cnt = Container.objects.get(title=cName)

        self.assertIn(cls, cnt.classes.all())
        self.assertTrue(cnt in cls.container_set.all(), "The container is not associated with the class.")

# 需要导入模块: from flashcardapp.datacontroller import DataController [as 别名]
# 或者: from flashcardapp.datacontroller.DataController import addContainer [as 别名]
def addContainer(request, ContainerData):
    ''' Ajax wrapper for the DataController function 'addContainer'

    *Takes*
      A dictionary representing a container
    *Returns*
      A json string declaring saved data success or failure

    See the DataController documentation for more information
    '''
    dc = DataController()
    return dc.addContainer(ContainerData, request.user)