| # Licensed to the Apache Software Foundation (ASF) under one |
| # or more contributor license agreements. See the NOTICE file |
| # distributed with this work for additional information |
| # regarding copyright ownership. The ASF licenses this file |
| # to you under the Apache License, Version 2.0 (the |
| # "License"); you may not use this file except in compliance |
| # with the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, |
| # software distributed under the License is distributed on an |
| # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| # KIND, either express or implied. See the License for the |
| # specific language governing permissions and limitations |
| # under the License. |
| |
| '''This module is added for testing the files model ''' |
| |
| from forgefiles.tests.model import FilesTestWithModel |
| from forgefiles.model.files import UploadFolder |
| from forgefiles.model.files import UploadFiles |
| |
| |
| class TestUpload(FilesTestWithModel): |
| ''' Test class for UploadFolder & uploadFiles model''' |
| |
| def test_upload_folder(self): |
| '''Creates an object of the UploadFolder Collection and tests its fields''' |
| upload_folder = UploadFolder |
| upload_folder.folder_name = 'testFolder' |
| upload_folder.published = True |
| upload_folder.remark = 'Publishing new Version' |
| upload_folder.disabled = False |
| assert upload_folder.folder_name == 'testFolder' |
| assert upload_folder.published |
| assert upload_folder.remark == 'Publishing new Version' |
| assert not upload_folder.disabled |
| |
| def test_upload_file(self): |
| '''Creates an object of the UploadFiles Collection and tests its fields''' |
| |
| upload_file = UploadFiles |
| upload_file.filename = 'testFile' |
| upload_file.filetype = 'project_file' |
| upload_file.file_url = 'TestFolder/testFile' |
| upload_file.linked_to_download = True |
| upload_file.published = False |
| assert upload_file.filename == 'testFile' |
| assert upload_file.filetype == 'project_file' |
| assert upload_file.linked_to_download |
| assert not upload_file.published |