diff options
| -rw-r--r-- | Example/Storage/Tests/Unit/FIRStorageMetadataTests.m | 10 | ||||
| -rw-r--r-- | Firebase/Storage/FIRStorageMetadata.m | 4 |
2 files changed, 12 insertions, 2 deletions
diff --git a/Example/Storage/Tests/Unit/FIRStorageMetadataTests.m b/Example/Storage/Tests/Unit/FIRStorageMetadataTests.m index f5fb3b3..92e96bb 100644 --- a/Example/Storage/Tests/Unit/FIRStorageMetadataTests.m +++ b/Example/Storage/Tests/Unit/FIRStorageMetadataTests.m @@ -43,7 +43,8 @@ kFIRStorageMetadataMetageneration : @"67890", kFIRStorageMetadataName : @"path/to/object", kFIRStorageMetadataTimeCreated : @"1992-08-07T17:22:53.108Z", - kFIRStorageMetadataUpdated : @"2016-03-01T20:16:01.673Z" + kFIRStorageMetadataUpdated : @"2016-03-01T20:16:01.673Z", + kFIRStorageMetadataSize : @1337 }; FIRStorageMetadata *metadata = [[FIRStorageMetadata alloc] initWithDictionary:metaDict]; XCTAssertNotNil(metadata); @@ -69,6 +70,8 @@ metaDict[kFIRStorageMetadataTimeCreated]); XCTAssertEqualObjects([metadata RFC3339StringFromDate:metadata.updated], metaDict[kFIRStorageMetadataUpdated]); + NSNumber *size = [NSNumber numberWithLongLong:metadata.size]; + XCTAssertEqualObjects(size, metaDict[kFIRStorageMetadataSize]); } - (void)testDictionaryRepresentation { @@ -85,7 +88,8 @@ kFIRStorageMetadataMetageneration : @"67890", kFIRStorageMetadataName : @"path/to/object", kFIRStorageMetadataTimeCreated : @"1992-08-07T17:22:53.108Z", - kFIRStorageMetadataUpdated : @"2016-03-01T20:16:01.673Z" + kFIRStorageMetadataUpdated : @"2016-03-01T20:16:01.673Z", + kFIRStorageMetadataSize : @1337 }; FIRStorageMetadata *metadata = [[FIRStorageMetadata alloc] initWithDictionary:metaDict]; NSDictionary *dictRepresentation = [metadata dictionaryRepresentation]; @@ -116,6 +120,8 @@ metaDict[kFIRStorageMetadataTimeCreated]); XCTAssertEqualObjects(dictRepresentation[kFIRStorageMetadataUpdated], metaDict[kFIRStorageMetadataUpdated]); + XCTAssertEqualObjects(dictRepresentation[kFIRStorageMetadataSize], + metaDict[kFIRStorageMetadataSize]); } - (void)testInitialzeNoDownloadTokensGetToken { diff --git a/Firebase/Storage/FIRStorageMetadata.m b/Firebase/Storage/FIRStorageMetadata.m index 6c85bbf..4269a45 100644 --- a/Firebase/Storage/FIRStorageMetadata.m +++ b/Firebase/Storage/FIRStorageMetadata.m @@ -181,6 +181,10 @@ metadataDictionary[kFIRStorageMetadataName] = _path; } + if (_size) { + metadataDictionary[kFIRStorageMetadataSize] = [NSNumber numberWithLongLong:_size]; + } + return [metadataDictionary copy]; } |