예제 #1
0
 def test_update_patch(self, mock_serializer):
     """Test update with PATCH"""
     mock_serializer.return_value.is_valid.return_value = True
     mock_serializer.return_value.save.return_value = self.property_state
     mock_request = mock.MagicMock()
     mock_request.method = 'PATCH'
     data = {'org_id': 1, 'cycle': 2, 'state': {'test': 3}, 'property': 4}
     serializer = PropertyViewAsStateSerializer()
     serializer.context = {'request': mock_request}
     serializer.update(self.property_view, data)
     mock_serializer.assert_called_with(self.property_state,
                                        data={'test': 3})
     self.assertTrue(mock_serializer.return_value.save.called)
    def test_update_put(self, mock_serializer):
        """Test update with PUT"""
        mock_serializer.return_value.is_valid.return_value = True
        mock_serializer.return_value.save.return_value = 'mock_state'
        mock_property_view = mock.MagicMock()
        mock_request = mock.MagicMock()
        data = {'org_id': 1, 'cycle': 2, 'state': {'test': 3}, 'property': 4}

        serializer = PropertyViewAsStateSerializer()
        mock_request.METHOD = 'PUT'
        serializer.context = {'request': mock_request}
        serializer.update(mock_property_view, data)
        mock_serializer.assert_called_with(data={'test': 3})
        self.assertTrue(mock_serializer.return_value.save.called)