104 lines
3.6 KiB
Objective-C
104 lines
3.6 KiB
Objective-C
/**
|
|
* Copyright (c) 2015-present, Facebook, Inc.
|
|
* All rights reserved.
|
|
*
|
|
* This source code is licensed under the BSD-style license found in the
|
|
* LICENSE file in the root directory of this source tree. An additional grant
|
|
* of patent rights can be found in the PATENTS file in the same directory.
|
|
*
|
|
*/
|
|
|
|
#import <XCTest/XCTest.h>
|
|
|
|
#import <React/RCTMultipartStreamReader.h>
|
|
|
|
@interface RCTMultipartStreamReaderTests : XCTestCase
|
|
|
|
@end
|
|
|
|
@implementation RCTMultipartStreamReaderTests
|
|
|
|
- (void)testSimpleCase {
|
|
NSString *response =
|
|
@"preable, should be ignored\r\n"
|
|
@"--sample_boundary\r\n"
|
|
@"Content-Type: application/json; charset=utf-8\r\n"
|
|
@"Content-Length: 2\r\n\r\n"
|
|
@"{}\r\n"
|
|
@"--sample_boundary--\r\n"
|
|
@"epilogue, should be ignored";
|
|
|
|
NSInputStream *inputStream = [NSInputStream inputStreamWithData:[response dataUsingEncoding:NSUTF8StringEncoding]];
|
|
RCTMultipartStreamReader *reader = [[RCTMultipartStreamReader alloc] initWithInputStream:inputStream boundary:@"sample_boundary"];
|
|
__block NSInteger count = 0;
|
|
BOOL success = [reader readAllParts:^(NSDictionary *headers, NSData *content, BOOL done) {
|
|
XCTAssertTrue(done);
|
|
XCTAssertEqualObjects(headers[@"Content-Type"], @"application/json; charset=utf-8");
|
|
XCTAssertEqualObjects([[NSString alloc] initWithData:content encoding:NSUTF8StringEncoding], @"{}");
|
|
count++;
|
|
}];
|
|
XCTAssertTrue(success);
|
|
XCTAssertEqual(count, 1);
|
|
}
|
|
|
|
- (void)testMultipleParts {
|
|
NSString *response =
|
|
@"preable, should be ignored\r\n"
|
|
@"--sample_boundary\r\n"
|
|
@"1\r\n"
|
|
@"--sample_boundary\r\n"
|
|
@"2\r\n"
|
|
@"--sample_boundary\r\n"
|
|
@"3\r\n"
|
|
@"--sample_boundary--\r\n"
|
|
@"epilogue, should be ignored";
|
|
|
|
NSInputStream *inputStream = [NSInputStream inputStreamWithData:[response dataUsingEncoding:NSUTF8StringEncoding]];
|
|
RCTMultipartStreamReader *reader = [[RCTMultipartStreamReader alloc] initWithInputStream:inputStream boundary:@"sample_boundary"];
|
|
__block NSInteger count = 0;
|
|
BOOL success = [reader readAllParts:^(__unused NSDictionary *headers, NSData *content, BOOL done) {
|
|
count++;
|
|
XCTAssertEqual(done, count == 3);
|
|
NSString *expectedBody = [NSString stringWithFormat:@"%ld", (long)count];
|
|
NSString *actualBody = [[NSString alloc] initWithData:content encoding:NSUTF8StringEncoding];
|
|
XCTAssertEqualObjects(actualBody, expectedBody);
|
|
}];
|
|
XCTAssertTrue(success);
|
|
XCTAssertEqual(count, 3);
|
|
}
|
|
|
|
- (void)testNoDelimiter {
|
|
NSString *response = @"Yolo";
|
|
|
|
NSInputStream *inputStream = [NSInputStream inputStreamWithData:[response dataUsingEncoding:NSUTF8StringEncoding]];
|
|
RCTMultipartStreamReader *reader = [[RCTMultipartStreamReader alloc] initWithInputStream:inputStream boundary:@"sample_boundary"];
|
|
__block NSInteger count = 0;
|
|
BOOL success = [reader readAllParts:^(__unused NSDictionary *headers, __unused NSData *content, __unused BOOL done) {
|
|
count++;
|
|
}];
|
|
XCTAssertFalse(success);
|
|
XCTAssertEqual(count, 0);
|
|
}
|
|
|
|
- (void)testNoCloseDelimiter {
|
|
NSString *response =
|
|
@"preable, should be ignored\r\n"
|
|
@"--sample_boundary\r\n"
|
|
@"Content-Type: application/json; charset=utf-8\r\n"
|
|
@"Content-Length: 2\r\n\r\n"
|
|
@"{}\r\n"
|
|
@"--sample_boundary\r\n"
|
|
@"incomplete message...";
|
|
|
|
NSInputStream *inputStream = [NSInputStream inputStreamWithData:[response dataUsingEncoding:NSUTF8StringEncoding]];
|
|
RCTMultipartStreamReader *reader = [[RCTMultipartStreamReader alloc] initWithInputStream:inputStream boundary:@"sample_boundary"];
|
|
__block NSInteger count = 0;
|
|
BOOL success = [reader readAllParts:^(__unused NSDictionary *headers, __unused NSData *content, __unused BOOL done) {
|
|
count++;
|
|
}];
|
|
XCTAssertFalse(success);
|
|
XCTAssertEqual(count, 1);
|
|
}
|
|
|
|
@end
|