Chromium Code Reviews| OLD | NEW |
|---|---|
| (Empty) | |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #import "chrome/installer/mac/app/Unpacker.h" | |
| 6 | |
| 7 #include "testing/gtest/include/gtest/gtest.h" | |
| 8 | |
| 9 #import "chrome/installer/mac/app/Downloader.h" | |
| 10 | |
| 11 dispatch_semaphore_t mount_semaphore; | |
| 12 | |
| 13 @interface TestDelegate : NSObject<UnpackDelegate> | |
| 14 @end | |
| 15 | |
| 16 @implementation TestDelegate | |
| 17 - (void)onUnpackSuccess { | |
| 18 NSFileManager* fileManager = [NSFileManager defaultManager]; | |
| 19 | |
| 20 if ([fileManager removeItemAtPath:@"/Applications/Google Chromo.app" | |
|
Elly Fong-Jones
2016/08/16 15:26:18
Chrome :)
Anna Zeng
2016/08/16 23:07:40
Acknowledged.
| |
| 21 error:nil]) { | |
| 22 NSLog(@"Application removed!"); | |
| 23 } | |
| 24 | |
| 25 NSLog(@"test succeeded"); | |
| 26 dispatch_semaphore_signal(mount_semaphore); | |
| 27 } | |
| 28 @end | |
| 29 | |
| 30 namespace { | |
| 31 | |
| 32 // TODO: edit the below test to work with the current protocol | |
| 33 TEST(UnpackerTest, MountAndUnmount) { | |
| 34 Unpacker* unpack = [[Unpacker alloc] | |
| 35 // NOTE: purposefully misspelled Chrome in order to avoid overwriting the | |
| 36 // current machine's Google Chrome installation. | |
| 37 initWithFinalAppPath:@"/Applications/Google Chromo.app"]; | |
|
Sidney San Martín
2016/08/13 12:37:19
It would be better if the test used a temporary fo
Anna Zeng
2016/08/16 23:07:40
Good point! Done.
| |
| 38 TestDelegate* test_delegate = [[TestDelegate alloc] init]; | |
| 39 unpack.delegate = test_delegate; | |
| 40 mount_semaphore = dispatch_semaphore_create(0); | |
| 41 | |
| 42 [unpack mountDMG]; | |
| 43 | |
| 44 DASessionRef session = DASessionCreate(nil); | |
| 45 DADiskRef child_disk = DADiskCreateFromVolumePath( | |
| 46 nil, session, | |
| 47 (CFURLRef)[NSURL URLWithString:PATH_FROM_EXECUTABLE(@"tmp")]); | |
| 48 DADiskRef whole_disk = DADiskCopyWholeDisk(child_disk); | |
| 49 | |
| 50 CFDictionaryRef disk_dictionary = DADiskCopyDescription(whole_disk); | |
| 51 CFTypeRef mediaEjectableKey = CFDictionaryGetValue( | |
| 52 disk_dictionary, kDADiskDescriptionMediaEjectableKey); | |
| 53 ASSERT_TRUE(CFBooleanGetValue((CFBooleanRef)mediaEjectableKey)); | |
|
Sidney San Martín
2016/08/13 12:37:19
It would be way simpler to just look for a file (l
Anna Zeng
2016/08/16 23:07:40
Done.
| |
| 54 CFRelease(whole_disk); | |
| 55 CFRelease(child_disk); | |
| 56 | |
| 57 [unpack unmountDMG]; | |
| 58 dispatch_semaphore_wait(mount_semaphore, DISPATCH_TIME_FOREVER); | |
| 59 | |
| 60 DADiskRef disk_check = DADiskCreateFromVolumePath( | |
| 61 nil, session, | |
| 62 (CFURLRef)[NSURL URLWithString:PATH_FROM_EXECUTABLE(@"tmp")]); | |
| 63 EXPECT_FALSE(disk_check); | |
|
Sidney San Martín
2016/08/13 12:37:19
Ditto.
Anna Zeng
2016/08/16 23:07:40
Done.
| |
| 64 } | |
| 65 | |
| 66 } // namespace | |
| OLD | NEW |