FileTransferCellTests.swift

//
// Wire
// Copyright (C) 2016 Wire Swiss GmbH
// 
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
// 
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
// 
// You should have received a copy of the GNU General Public License
// along with this program. If not, see http://www.gnu.org/licenses/.
// 


import XCTest
@testable import Wire

class FileTransferCellTests: ZMSnapshotTestCase {
    
    func wrappedCellWithConfig(_ config: ((MockMessage) -> ())?) -> UITableView {
    
        let fileMessage = MockMessageFactory.fileTransferMessage()
        
        if let config = config {
            config(fileMessage!)
        }
        
        let cell = FileTransferCell(style: .default, reuseIdentifier: "test")
        
        let layoutProperties = ConversationCellLayoutProperties()
        layoutProperties.showSender = true
        layoutProperties.showBurstTimestamp = false
        layoutProperties.showUnreadMarker = false
        
        cell.prepareForReuse()
        cell.bounds = CGRect(x: 0.0, y: 0.0, width: 320.0, height: 9999)
        cell.contentView.bounds = CGRect(x: 0.0, y: 0.0, width: 320, height: 9999)
        cell.layer.speed = 0;
        
        cell.layoutMargins = UIEdgeInsetsMake(0, CGFloat(WAZUIMagic.float(forIdentifier: "content.left_margin")),
        0, CGFloat(WAZUIMagic.float(forIdentifier: "content.right_margin")))
        
        cell.configure(for: fileMessage, layoutProperties: layoutProperties)
      
        return cell.wrapInTableView()
    }
    
    override func setUp() {
        super.setUp()
        self.accentColor = .vividRed
    }
   
    func testUploadedCell_fromThisDevice() {
        let cell = self.wrappedCellWithConfig({
            $0.fileMessageData?.transferState = .uploaded
            $0.backingFileMessageData.fileURL = Bundle.main.bundleURL
        })
        verify(view: cell)
    }
    
    func testUploadedCell_fromOtherDevice() {
        let cell = self.wrappedCellWithConfig({
            $0.fileMessageData?.transferState = .uploaded
            $0.backingFileMessageData.fileURL = .none
        })
        verify(view: cell)
    }
    
    func testUploadedCell_fromOtherUser() {
        let cell = self.wrappedCellWithConfig({
            $0.fileMessageData?.transferState = .uploaded
            $0.backingFileMessageData.fileURL = .none
            $0.sender = MockUser.mockUsers().first!
        })
        verify(view: cell)
    }
    
    
    
    func testUploadedCell_fromThisDevice_longFileName() {
        let cell = self.wrappedCellWithConfig({
            $0.fileMessageData?.transferState = .uploaded
            $0.fileMessageData?.filename = "Etiam lacus elit, tempor at blandit sit amet, faucibus in erat. Mauris faucibus scelerisque mattis.pdf"
            $0.backingFileMessageData.fileURL = Bundle.main.bundleURL
        })
        verify(view: cell)
    }
    
    func testUploadedCell_fromThisDevice_bigFileSize() {
        let cell = self.wrappedCellWithConfig({
            $0.fileMessageData?.transferState = .uploaded
            $0.fileMessageData?.size = 1024 * 1024 * 25
            $0.backingFileMessageData.fileURL = .none
        })
        verify(view: cell)
    }
    
    
    func testUploadingCell_fromThisDevice() {
        let cell = self.wrappedCellWithConfig({
            $0.fileMessageData?.transferState = .uploading
            $0.backingFileMessageData.fileURL = Bundle.main.bundleURL
        })
        verify(view: cell)
    }
    
    func testUploadingCell_fromOtherDevice() {
        let cell = self.wrappedCellWithConfig({
            $0.fileMessageData?.transferState = .uploading
            $0.backingFileMessageData.fileURL = .none
        })
        verify(view: cell)
    }
    
    func testUploadingCell_fromOtherUser() {
        let cell = self.wrappedCellWithConfig({
            $0.fileMessageData?.transferState = .uploading
            $0.backingFileMessageData.fileURL = .none
            $0.sender = MockUser.mockUsers().first!
        })
        verify(view: cell)
    }
    
    
    
    func testDownloadingCell_fromThisDevice() {
        let cell = self.wrappedCellWithConfig({
            $0.fileMessageData?.transferState = .downloading
            $0.fileMessageData?.progress = 0.75
            $0.backingFileMessageData.fileURL = Bundle.main.bundleURL
        })
        verify(view: cell)
    }
    
    func testDownloadingCell_fromOtherDevice() {
        let cell = self.wrappedCellWithConfig({
            $0.fileMessageData?.transferState = .downloading
            $0.fileMessageData?.progress = 0.75
            $0.backingFileMessageData.fileURL = .none
        })
        verify(view:cell)
    }
    
    func testDownloadingCell_fromOtherUser() {
        let cell = self.wrappedCellWithConfig({
            $0.fileMessageData?.transferState = .downloading
            $0.backingFileMessageData.fileURL = .none
            $0.fileMessageData?.progress = 0.75
            $0.sender = MockUser.mockUsers().first!
        })
        verify(view: cell)
    }
    
    
    
    func testDownloadedCell_fromThisDevice() {
        let cell = self.wrappedCellWithConfig({
            $0.fileMessageData?.transferState = .downloaded
            $0.backingFileMessageData.fileURL = Bundle.main.bundleURL
        })
        verify(view: cell)
    }
    
    func testDownloadedCell_fromOtherDevice() {
        let cell = self.wrappedCellWithConfig({
            $0.fileMessageData?.transferState = .downloaded
            $0.backingFileMessageData.fileURL = .none
        })
        verify(view: cell)
    }
    
    func testDownloadedCell_fromOtherUser() {
        let cell = self.wrappedCellWithConfig({
            $0.fileMessageData?.transferState = .downloaded
            $0.backingFileMessageData.fileURL = .none
            $0.sender = MockUser.mockUsers().first!
        })
        verify(view: cell)
    }
    
    
    
    func testFailedDownloadCell_fromThisDevice() {
        let cell = self.wrappedCellWithConfig({
            $0.fileMessageData?.transferState = .failedDownload
            $0.backingFileMessageData.fileURL = Bundle.main.bundleURL
        })
        verify(view: cell)
    }
    
    func testFailedDownloadCell_fromOtherDevice() {
        let cell = self.wrappedCellWithConfig({
            $0.fileMessageData?.transferState = .failedDownload
            $0.backingFileMessageData.fileURL = .none
        })
        verify(view: cell)
    }
    
    func testFailedDownloadCell_fromOtherUser() {
        let cell = self.wrappedCellWithConfig({
            $0.fileMessageData?.transferState = .failedDownload
            $0.backingFileMessageData.fileURL = .none
            $0.sender = MockUser.mockUsers().first!
        })
        verify(view: cell)
    }
    
    
    
    
    func testFailedUploadCell_fromThisDevice() {
        let cell = self.wrappedCellWithConfig({
            $0.fileMessageData?.transferState = .failedUpload
            $0.backingFileMessageData.fileURL = Bundle.main.bundleURL
        })
        verify(view: cell)
    }
    
    func testFailedUploadCell_fromOtherDevice() {
        let cell = self.wrappedCellWithConfig({
            $0.fileMessageData?.transferState = .failedUpload
            $0.backingFileMessageData.fileURL = .none
        })
        verify(view: cell)
    }
    
    func testFailedUploadCell_fromOtherUser() {
        let cell = self.wrappedCellWithConfig({
            $0.fileMessageData?.transferState = .failedUpload
            $0.backingFileMessageData.fileURL = .none
            $0.sender = MockUser.mockUsers().first!
        })
        verify(view: cell)
    }
    
    // MARK : Upload Cancelled
    
    func testCancelledUploadCell_fromThisDevice() {
        let cell = wrappedCellWithConfig {
            $0.fileMessageData?.transferState = .cancelledUpload
            $0.backingFileMessageData.fileURL = Bundle.main.bundleURL
        }
        
        verify(view: cell)
    }

    // MARK : Obfuscated

    func testObfuscatedFileTransferCell() {
        verify(view: wrappedCellWithConfig {
            $0.fileMessageData?.transferState = .uploaded
            $0.backingFileMessageData.fileURL = Bundle.main.bundleURL
            $0.isObfuscated = true
        })
    }
}