QuietUnrar/Carthage/Checkouts/UnrarKit/Tests/IterateFileInfoTests.m

215 lines
8.5 KiB
Objective-C

//
// IterateFileInfoTests.m
// UnrarKit
//
// Created by Dov Frankel on 5/22/18.
//
//
#import "URKArchiveTestCase.h"
@interface IterateFileInfoTests : URKArchiveTestCase
@end
@implementation IterateFileInfoTests
- (void)testIterateFileInfo
{
NSArray *testArchives = @[@"Test Archive.rar", @"Test Archive (Password).rar"];
NSSet *expectedFileSet = [self.testFileURLs keysOfEntriesPassingTest:^BOOL(NSString *key, id obj, BOOL *stop) {
return ![key hasSuffix:@"rar"];
}];
NSArray *expectedFiles = [[expectedFileSet allObjects] sortedArrayUsingSelector:@selector(compare:)];
for (NSString *testArchiveName in testArchives) {
NSLog(@"Testing list files of archive %@", testArchiveName);
NSURL *testArchiveURL = self.testFileURLs[testArchiveName];
URKArchive *archive = [[URKArchive alloc] initWithURL:testArchiveURL error:nil];
NSError *error = nil;
NSMutableArray *iteratedFiles = [NSMutableArray array];
BOOL success = [archive iterateFileInfo:^(URKFileInfo * _Nonnull fileInfo, BOOL * _Nonnull stop) {
[iteratedFiles addObject:fileInfo.filename];
}
error:&error];
XCTAssertTrue(success, @"Error returned by iterateFileInfo");
XCTAssertNil(error, @"Error returned by iterateFileInfo");
XCTAssertEqual(iteratedFiles.count, expectedFileSet.count,
@"Incorrect number of files listed in archive");
for (NSInteger i = 0; i < iteratedFiles.count; i++) {
NSString *archiveFilename = iteratedFiles[i];
NSString *expectedFilename = expectedFiles[i];
NSLog(@"Testing for file %@", expectedFilename);
XCTAssertEqualObjects(archiveFilename, expectedFilename, @"Incorrect filename listed");
}
}
}
- (void)testIterateFileInfo_Unicode
{
NSSet *expectedFileSet = [self.unicodeFileURLs keysOfEntriesPassingTest:^BOOL(NSString *key, id obj, BOOL *stop) {
return ![key hasSuffix:@"rar"];
}];
NSArray *expectedFiles = [[expectedFileSet allObjects] sortedArrayUsingSelector:@selector(compare:)];
NSURL *testArchiveURL = self.unicodeFileURLs[@"Ⓣest Ⓐrchive.rar"];
URKArchive *archive = [[URKArchive alloc] initWithURL:testArchiveURL error:nil];
NSError *error = nil;
NSMutableArray *iteratedFiles = [NSMutableArray array];
BOOL success = [archive iterateFileInfo:^(URKFileInfo * _Nonnull fileInfo, BOOL * _Nonnull stop) {
[iteratedFiles addObject:fileInfo.filename];
}
error:&error];
XCTAssertTrue(success, @"Error returned by iterateFileInfo");
XCTAssertNil(error, @"Error returned by iterateFileInfo");
XCTAssertEqual(iteratedFiles.count, expectedFileSet.count,
@"Incorrect number of files listed in archive");
for (NSInteger i = 0; i < iteratedFiles.count; i++) {
NSString *archiveFilename = iteratedFiles[i];
NSString *expectedFilename = expectedFiles[i];
XCTAssertEqualObjects(archiveFilename, expectedFilename, @"Incorrect filename listed");
}
}
- (void)testIterateFileInfo_RAR5
{
NSArray *expectedFiles = @[@"yohoho_ws.txt",
@"nopw.txt"];
NSURL *testArchiveURL = self.testFileURLs[@"Test Archive (RAR5).rar"];
URKArchive *archive = [[URKArchive alloc] initWithURL:testArchiveURL error:nil];
NSError *error = nil;
NSMutableArray *iteratedFiles = [NSMutableArray array];
BOOL success = [archive iterateFileInfo:^(URKFileInfo * _Nonnull fileInfo, BOOL * _Nonnull stop) {
[iteratedFiles addObject:fileInfo.filename];
}
error:&error];
XCTAssertTrue(success, @"Error returned by iterateFileInfo");
XCTAssertNil(error, @"Error returned by iterateFileInfo");
XCTAssertEqual(iteratedFiles.count, expectedFiles.count,
@"Incorrect number of files listed in archive");
for (NSInteger i = 0; i < iteratedFiles.count; i++) {
NSString *archiveFilename = iteratedFiles[i];
NSString *expectedFilename = expectedFiles[i];
XCTAssertEqualObjects(archiveFilename, expectedFilename, @"Incorrect filename listed");
}
}
- (void)testIterateFileInfo_HeaderPassword
{
NSArray *testArchives = @[@"Test Archive (Header Password).rar"];
NSSet *expectedFileSet = [self.testFileURLs keysOfEntriesPassingTest:^BOOL(NSString *key, id obj, BOOL *stop) {
return ![key hasSuffix:@"rar"];
}];
NSArray *expectedFiles = [[expectedFileSet allObjects] sortedArrayUsingSelector:@selector(compare:)];
for (NSString *testArchiveName in testArchives) {
NSLog(@"Testing list files of archive %@", testArchiveName);
NSURL *testArchiveURL = self.testFileURLs[testArchiveName];
URKArchive *archiveNoPassword = [[URKArchive alloc] initWithURL:testArchiveURL error:nil];
NSError *passwordError = nil;
NSArray *filesInArchive = [archiveNoPassword listFilenames:&passwordError];
XCTAssertNotNil(passwordError, @"No error returned by listFilenames (no password given)");
XCTAssertNil(filesInArchive, @"List of files returned (no password given)");
URKArchive *archive = [[URKArchive alloc] initWithURL:testArchiveURL password:@"password" error:nil];
NSError *error = nil;
NSMutableArray *iteratedFiles = [NSMutableArray array];
BOOL success = [archive iterateFileInfo:^(URKFileInfo * _Nonnull fileInfo, BOOL * _Nonnull stop) {
[iteratedFiles addObject:fileInfo.filename];
}
error:&error];
XCTAssertTrue(success, @"Error returned by iterateFileInfo");
XCTAssertNil(error, @"Error returned by iterateFileInfo");
XCTAssertEqual(iteratedFiles.count, expectedFiles.count,
@"Incorrect number of files listed in archive");
for (NSInteger i = 0; i < iteratedFiles.count; i++) {
NSString *archiveFilename = iteratedFiles[i];
NSString *expectedFilename = expectedFiles[i];
NSLog(@"Testing for file %@", expectedFilename);
XCTAssertEqualObjects(archiveFilename, expectedFilename, @"Incorrect filename listed");
}
}
}
- (void)testIterateFileInfo_NoHeaderPasswordGiven
{
URKArchive *archive = [[URKArchive alloc] initWithURL:self.testFileURLs[@"Test Archive (Header Password).rar"] error:nil];
NSError *error = nil;
__block BOOL called = NO;
BOOL success = [archive iterateFileInfo:^(URKFileInfo * _Nonnull fileInfo, BOOL * _Nonnull stop) {
called = YES;
}
error:&error];
XCTAssertNotNil(error, @"Iteration without password returned no error");
XCTAssertFalse(success, @"Iteration without password succeeded");
XCTAssertFalse(called, @"Iteration without password called action block");
XCTAssertEqual(error.code, URKErrorCodeMissingPassword, @"Unexpected error code returned");
}
- (void)testIterateFileInfo_NoFilePasswordGiven
{
URKArchive *archive = [[URKArchive alloc] initWithURL:self.testFileURLs[@"Test Archive (Password).rar"] error:nil];
NSError *error = nil;
__block BOOL called = NO;
BOOL success = [archive iterateFileInfo:^(URKFileInfo * _Nonnull fileInfo, BOOL * _Nonnull stop) {
called = YES;
}
error:&error];
XCTAssertNil(error, @"Iteration without file password failed");
XCTAssertTrue(success, @"Iteration without file password failed");
XCTAssertTrue(called, @"Iteration without file password didn't call action block");
}
- (void)testIterateFileInfo_InvalidArchive
{
URKArchive *archive = [[URKArchive alloc] initWithURL:self.testFileURLs[@"Test File A.txt"] error:nil];
NSError *error = nil;
__block BOOL called = NO;
BOOL success = [archive iterateFileInfo:^(URKFileInfo * _Nonnull fileInfo, BOOL * _Nonnull stop) {
called = YES;
}
error:&error];
XCTAssertNotNil(error, @"Iteration of invalid archive succeeded");
XCTAssertFalse(success, @"Iteration for invalid archive succeeded");
XCTAssertFalse(called, @"Iteration for invalid archive called action block");
XCTAssertEqual(error.code, URKErrorCodeBadArchive, @"Unexpected error code returned");
}
@end