swifter/Xcode/Tests/FilesTests.swift at b6edfa355fc8d8f5acfc63641187e9a760a9c43d · httpswift/swifter · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
//
//  FilesTests.swift
//  Swifter
//
//  Created by Michael Enger on 02/09/2021.
//  Copyright © 2021 Damian Kołakowski. All rights reserved.
//

import XCTest
#if os(Linux)
import FoundationNetworking
#endif
@testable import Swifter

class FilesTests: XCTestCase {
    let temporaryDirectoryURL = URL(fileURLWithPath: NSTemporaryDirectory(), isDirectory: true)
    let temporaryFileName = UUID().uuidString + ".png"

    override func setUp() {
        super.setUp()

        let temporaryFileURL = temporaryDirectoryURL.appendingPathComponent(temporaryFileName)
        let data = "This is a file"
        do {
            try data.write(to: temporaryFileURL, atomically: true, encoding: String.Encoding.utf8)
        } catch {
            XCTFail("Failed to create temporary file")
        }
    }

    override func tearDown() {
        let temporaryFileURL = temporaryDirectoryURL.appendingPathComponent(temporaryFileName)
        do {
            try FileManager.default.removeItem(at: temporaryFileURL)
        } catch {
            // no worries
        }

        super.tearDown()
    }

    func testShareFile() {
        let request = HttpRequest()
        let closure = shareFile(temporaryDirectoryURL.appendingPathComponent(temporaryFileName).path)
        let result = closure(request)
        let headers = result.headers()

        XCTAssert(result.statusCode == 200)
        XCTAssert(headers["Content-Type"] == "image/png")
        XCTAssert(headers["Content-Length"] == "14")
    }

    func testShareFileNotFound() {
        let request = HttpRequest()
        let closure = shareFile(temporaryDirectoryURL.appendingPathComponent("does_not_exist").path)
        let result = closure(request)

        XCTAssert(result == .notFound())
    }

    func testShareFilesFromDirectory() {
        let request = HttpRequest()
        request.params = ["path": temporaryFileName]
        let closure = shareFilesFromDirectory(temporaryDirectoryURL.path)
        let result = closure(request)
        let headers = result.headers()

        XCTAssert(result.statusCode == 200)
        XCTAssert(headers["Content-Type"] == "image/png")
        XCTAssert(headers["Content-Length"] == "14")
    }

    func testShareFilesFromDirectoryFileNotFound() {
        let request = HttpRequest()
        request.params = ["path": "does_not_exist.wav"]

        let closure = shareFilesFromDirectory(temporaryDirectoryURL.path)
        let result = closure(request)

        XCTAssert(result == .notFound())
    }

    func testDirectoryBrowser() {
        let request = HttpRequest()
        request.params = ["path": ""]
        let closure = directoryBrowser(temporaryDirectoryURL.path)
        let result = closure(request)

        XCTAssert(result.statusCode == 200)
    }

    func testDirectoryBrowserNotFound() {
        let request = HttpRequest()
        request.params = ["path": "does/not/exist"]
        let closure = directoryBrowser(temporaryDirectoryURL.path)
        let result = closure(request)

        XCTAssert(result == .notFound())
    }
}