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
101
102
103
104
105
106
107
108
109
110
111
112
|
//
// LogRotationTests.swift
// MullvadVPNTests
//
// Created by Jon Petersson on 2024-04-12.
// Copyright © 2025 Mullvad VPN AB. All rights reserved.
//
import XCTest
@testable import MullvadLogging
final class LogRotationTests: XCTestCase {
let fileManager = FileManager.default
var directoryPath: URL!
override func setUpWithError() throws {
directoryPath = FileManager.default.temporaryDirectory
.appendingPathComponent("LogRotationTests", isDirectory: true)
try fileManager.createDirectory(
at: directoryPath,
withIntermediateDirectories: true
)
}
override func tearDownWithError() throws {
try fileManager.removeItem(at: directoryPath)
}
func testRotateLogsByStorageSizeLimit() throws {
let logPaths = [
directoryPath.appendingPathComponent("test1.log"),
directoryPath.appendingPathComponent("test2.log"),
directoryPath.appendingPathComponent("test3.log"),
directoryPath.appendingPathComponent("test4.log"),
directoryPath.appendingPathComponent("test5.log"),
]
try logPaths.forEach { logPath in
try writeDataToDisk(path: logPath, fileSize: 1000)
}
try LogRotation.rotateLogs(
logDirectory: directoryPath,
options: LogRotation.Options(
storageSizeLimit: 5000,
oldestAllowedDate: .distantPast
)
)
var logFileCount = try fileManager.contentsOfDirectory(atPath: directoryPath.relativePath).count
XCTAssertEqual(logFileCount, 5)
try LogRotation.rotateLogs(
logDirectory: directoryPath,
options: LogRotation.Options(
storageSizeLimit: 3999,
oldestAllowedDate: .distantPast
)
)
logFileCount = try fileManager.contentsOfDirectory(atPath: directoryPath.relativePath).count
XCTAssertEqual(logFileCount, 3)
}
func testRotateLogsByOldestAllowedDate() throws {
let firstBatchOflogPaths = [
directoryPath.appendingPathComponent("test1.log"),
directoryPath.appendingPathComponent("test2.log"),
directoryPath.appendingPathComponent("test3.log"),
]
let secondBatchOflogPaths = [
directoryPath.appendingPathComponent("test4.log"),
directoryPath.appendingPathComponent("test5.log"),
]
let oldestDateAllowedForFirstBatch = Date()
try firstBatchOflogPaths.forEach { logPath in
try writeDataToDisk(path: logPath, fileSize: 1000)
}
let oldestDateAllowedForSecondBatch = Date()
try secondBatchOflogPaths.forEach { logPath in
try writeDataToDisk(path: logPath, fileSize: 1000)
}
try LogRotation.rotateLogs(
logDirectory: directoryPath,
options: LogRotation.Options(storageSizeLimit: .max, oldestAllowedDate: oldestDateAllowedForFirstBatch)
)
var logFileCount = try fileManager.contentsOfDirectory(atPath: directoryPath.relativePath).count
XCTAssertEqual(logFileCount, 5)
try LogRotation.rotateLogs(
logDirectory: directoryPath,
options: LogRotation.Options(storageSizeLimit: .max, oldestAllowedDate: oldestDateAllowedForSecondBatch)
)
logFileCount = try fileManager.contentsOfDirectory(atPath: directoryPath.relativePath).count
XCTAssertEqual(logFileCount, 2)
}
}
extension LogRotationTests {
private func stringOfSize(_ size: Int) -> String {
(0..<size).map { "\($0 % 10)" }.joined(separator: "")
}
private func writeDataToDisk(path: URL, fileSize: Int) throws {
let data = Data((0..<fileSize).map { UInt8($0 & 0xff) })
try data.write(to: path)
}
}
|
