add tests

Author: geof90

CodePush.h

@@ -22,6 +22,11 @@
 
 + (NSString *)getApplicationSupportDirectory;
 
+// The below methods are only used during tests.
++ (BOOL)isUsingTestConfiguration;
++ (void)setUsingTestConfiguration:(BOOL)shouldUseTestConfiguration;
++ (void)clearTestUpdates;
+
 @end
 
 @interface CodePushConfig : NSObject
@@ -77,6 +82,10 @@ failCallback:(void (^)(NSError *err))failCallback;
 
 + (void)rollbackPackage;
 
+// The below methods are only used during tests.
++ (void)downloadAndReplaceCurrentBundle:(NSString *)remoteBundleUrl;
++ (void)clearTestUpdates;
+
 @end
 
 typedef NS_ENUM(NSInteger, CodePushInstallMode) {

CodePush.js

@@ -20,75 +20,75 @@ function checkForUpdate(deploymentKey = null) {
    * different from the CodePush update they have already installed.
    */
   return getConfiguration()
-          .then((configResult) => {            
-            /*
-             * If a deployment key was explicitly provided,
-             * then let's override the one we retrieved
-             * from the native-side of the app. This allows
-             * dynamically "redirecting" end-users at different
-             * deployments (e.g. an early access deployment for insiders).
-             */
-            if (deploymentKey) {
-              config = Object.assign({}, configResult, { deploymentKey });
-            } else {
-              config = configResult;
-            }
-            
-            sdk = getSDK(config);
-            
-            // Allow dynamic overwrite of function. This is only to be used for tests.
-            return module.exports.getCurrentPackage();
-          })
-          .then((localPackage) => {
-            var queryPackage = { appVersion: config.appVersion };
-            
-            /*
-             * If the app has a previously installed update, and that update
-             * was targetted at the same app version that is currently running,
-             * then we want to use its package hash to determine whether a new
-             * release has been made on the server. Otherwise, we only need
-             * to send the app version to the server, since we are interested
-             * in any updates for current app store version, regardless of hash.
-             */
-            if (localPackage && localPackage.appVersion && semver.compare(localPackage.appVersion, config.appVersion) === 0) {
-              queryPackage = localPackage;
-            }
-            
-            return new Promise((resolve, reject) => {
-              sdk.queryUpdateWithCurrentPackage(queryPackage, (err, update) => {
-                if (err) {
-                  return reject(err);
-                }
-                
-                /*
-                 * There are three cases where checkForUpdate will resolve to null:
-                 * ----------------------------------------------------------------
-                 * 1) The server said there isn't an update. This is the most common case.
-                 * 2) The server said there is an update but it requires a newer binary version.
-                 *    This would occur when end-users are running an older app store version than
-                 *    is available, and CodePush is making sure they don't get an update that
-                 *    potentially wouldn't be compatible with what they are running.
-                 * 3) The server said there is an update, but the update's hash is the same as
-                 *    the currently running update. This should _never_ happen, unless there is a
-                 *    bug in the server, but we're adding this check just to double-check that the
-                 *    client app is resilient to a potential issue with the update check.
-                 */
-                if (!update || update.updateAppVersion || (update.packageHash === localPackage.packageHash)) {
-                  return resolve(null);
-                } 
+    .then((configResult) => {            
+      /*
+       * If a deployment key was explicitly provided,
+       * then let's override the one we retrieved
+       * from the native-side of the app. This allows
+       * dynamically "redirecting" end-users at different
+       * deployments (e.g. an early access deployment for insiders).
+       */
+      if (deploymentKey) {
+        config = Object.assign({}, configResult, { deploymentKey });
+      } else {
+        config = configResult;
+      }
+      
+      sdk = getSDK(config);
+      
+      // Allow dynamic overwrite of function. This is only to be used for tests.
+      return module.exports.getCurrentPackage();
+    })
+    .then((localPackage) => {
+      var queryPackage = { appVersion: config.appVersion };
+      
+      /*
+       * If the app has a previously installed update, and that update
+       * was targetted at the same app version that is currently running,
+       * then we want to use its package hash to determine whether a new
+       * release has been made on the server. Otherwise, we only need
+       * to send the app version to the server, since we are interested
+       * in any updates for current app store version, regardless of hash.
+       */
+      if (localPackage && localPackage.appVersion && semver.compare(localPackage.appVersion, config.appVersion) === 0) {
+        queryPackage = localPackage;
+      }
+      
+      return new Promise((resolve, reject) => {
+        sdk.queryUpdateWithCurrentPackage(queryPackage, (err, update) => {
+          if (err) {
+            return reject(err);
+          }
+          
+          /*
+           * There are three cases where checkForUpdate will resolve to null:
+           * ----------------------------------------------------------------
+           * 1) The server said there isn't an update. This is the most common case.
+           * 2) The server said there is an update but it requires a newer binary version.
+           *    This would occur when end-users are running an older app store version than
+           *    is available, and CodePush is making sure they don't get an update that
+           *    potentially wouldn't be compatible with what they are running.
+           * 3) The server said there is an update, but the update's hash is the same as
+           *    the currently running update. This should _never_ happen, unless there is a
+           *    bug in the server, but we're adding this check just to double-check that the
+           *    client app is resilient to a potential issue with the update check.
+           */
+          if (!update || update.updateAppVersion || (update.packageHash === localPackage.packageHash)) {
+            return resolve(null);
+          } 
 
-                update = Object.assign(update, PackageMixins.remote);
-                
-                NativeCodePush.isFailedUpdate(update.packageHash)
-                  .then((isFailedHash) => {
-                    update.failedInstall = isFailedHash;
-                    resolve(update);
-                  })
-                  .catch(reject)
-                  .done();
-              })
-            });
-          });
+          update = Object.assign(update, PackageMixins.remote);
+          
+          NativeCodePush.isFailedUpdate(update.packageHash)
+            .then((isFailedHash) => {
+              update.failedInstall = isFailedHash;
+              resolve(update);
+            })
+            .catch(reject)
+            .done();
+        })
+      });
+    });
 }
 
 var getConfiguration = (() => {
@@ -130,11 +130,8 @@ function getCurrentPackage() {
 }
 
 function getSDK(config) {
-   if (testSdk) {
-      return testSdk;
-   } else {
-      return new Sdk(requestFetchAdapter, config);
-   }
+  // Allow dynamic overwrite of acquisition SDK. This is only to be used for tests.
+  return new module.exports.AcquisitionSdk(requestFetchAdapter, config);
 }
 
 /* Logs messages to console with the [CodePush] prefix */
@@ -143,16 +140,15 @@ function log(message) {
 }
 
 var testConfig;
-var testSdk;
 
 // This function is only used for tests. Replaces the default SDK, configuration and native bridge
-function setUpTestDependencies(providedTestSdk, providedTestConfig, testNativeBridge) {
-  if (providedTestSdk) testSdk = providedTestSdk;
+function setUpTestDependencies(testSdk, providedTestConfig, testNativeBridge) {
+  if (testSdk) module.exports.AcquisitionSdk = testSdk;
   if (providedTestConfig) testConfig = providedTestConfig;
   if (testNativeBridge) NativeCodePush = testNativeBridge;
 }
 
-/**
+/*
  * The sync method provides a simple, one-line experience for
  * incorporating the check, download and application of an update.
  * 
@@ -300,6 +296,7 @@ function sync(options = {}, syncStatusChangeCallback, downloadProgressCallback)
 };
 
 var CodePush = {
+  AcquisitionSdk: Sdk,
   checkForUpdate: checkForUpdate,
   getConfiguration: getConfiguration,
   getCurrentPackage: getCurrentPackage,

CodePush.m

@@ -13,8 +13,7 @@ @implementation CodePush {
 
 RCT_EXPORT_MODULE()
 
-static NSTimer *_timer;
-static BOOL usingTestFolder = NO;
+static BOOL testConfigurationFlag = NO;
 
 static NSString *const FailedUpdatesKey = @"CODE_PUSH_FAILED_UPDATES";
 static NSString *const PendingUpdateKey = @"CODE_PUSH_PENDING_UPDATE";
@@ -24,6 +23,8 @@ @implementation CodePush {
 static NSString *const PendingUpdateHashKey = @"hash";
 static NSString *const PendingUpdateIsLoadingKey = @"isLoading";
 
+id saveTestModule = nil;
+
 @synthesize bridge = _bridge;
 
 // Public Obj-C API (see header for method comments)
@@ -68,6 +69,40 @@ + (NSString *)getApplicationSupportDirectory
     return applicationSupportDirectory;
 }
 
+/*
+ * This returns a boolean value indicating whether CodePush has
+ * been set to run under a test configuration.
+ */
++ (BOOL)isUsingTestConfiguration
+{
+    return testConfigurationFlag;
+}
+
+/* 
+ * This is used to enable an environment in which tests can be run.
+ * Specifically, it flips a boolean flag that causes bundles to be
+ * saved to a test folder and enables the ability to modify
+ * installed bundles on the fly from JavaScript.
+ */
++ (void)setUsingTestConfiguration:(BOOL)shouldUseTestConfiguration
+{
+    testConfigurationFlag = shouldUseTestConfiguration;
+}
+
+/*
+ * This is used to clean up all test updates. It can only be used
+ * when the testConfigurationFlag is set to YES, otherwise it will
+ * simply no-op.
+ */
++ (void)clearTestUpdates
+{
+    if ([CodePush isUsingTestConfiguration]) {
+        [CodePushPackage clearTestUpdates];
+        [self removePendingUpdate];
+    }
+}
+
+
 // Private API methods
 
 /*
@@ -155,10 +190,12 @@ - (void)loadBundle
         // is debugging and therefore, shouldn't be redirected to a local
         // file (since Chrome wouldn't support it). Otherwise, update
         // the current bundle URL to point at the latest update
-        if (![_bridge.bundleURL.scheme hasPrefix:@"http"]) {
-            _bridge.bundleURL = [CodePush bundleURL];
+        if ([CodePush isUsingTestConfiguration] || ![_bridge.bundleURL.scheme hasPrefix:@"http"]) {
+            NSURL *url = [CodePush bundleURL];
+            _bridge.bundleURL = url;
         }
 
+        saveTestModule = _bridge.modules[@"RCTTestModule"];
         [_bridge reload];
     });
 }
@@ -180,7 +217,7 @@ - (void)rollbackPackage
 
     // Rollback to the previous version and de-register the new update
     [CodePushPackage rollbackPackage];
-    [self removePendingUpdate];
+    [CodePush removePendingUpdate];
     [self loadBundle];
 }
 
@@ -207,10 +244,10 @@ - (void)saveFailedUpdate:(NSString *)packageHash
 }
 
 /*
- * This method  is used to register the fact that a pending
+ * This method is used to register the fact that a pending
  * update succeeded and therefore can be removed.
  */
-- (void)removePendingUpdate
++ (void)removePendingUpdate
 {
     NSUserDefaults *preferences = [NSUserDefaults standardUserDefaults];
     [preferences removeObjectForKey:PendingUpdateKey];
@@ -321,9 +358,7 @@ - (void)savePendingUpdate:(NSString *)packageHash
             [self savePendingUpdate:updatePackage[@"packageHash"]
                           isLoading:NO];
 
-            if (installMode == CodePushInstallModeImmediate) {
-                [self loadBundle];
-            } else if (installMode == CodePushInstallModeOnNextResume) {
+            if (installMode == CodePushInstallModeOnNextResume) {
                 // Ensure we do not add the listener twice.
                 if (!_hasResumeListener) {
                     // Register for app resume notifications so that we
@@ -376,7 +411,7 @@ - (void)savePendingUpdate:(NSString *)packageHash
 RCT_EXPORT_METHOD(notifyApplicationReady:(RCTPromiseResolveBlock)resolve
                                 rejecter:(RCTPromiseRejectBlock)reject)
 {
-    [self removePendingUpdate];
+    [CodePush removePendingUpdate];
     resolve([NSNull null]);
 }
 
@@ -388,9 +423,16 @@ - (void)savePendingUpdate:(NSString *)packageHash
     [self loadBundle];
 }
 
-RCT_EXPORT_METHOD(setUsingTestFolder:(BOOL)shouldUseTestFolder)
+/*
+ * This method is the native side of the CodePush.downloadAndReplaceCurrentBundle()
+ * method, which is only to be used during tests and no-ops if the test configuration
+ * flag is not set.
+ */
+RCT_EXPORT_METHOD(downloadAndReplaceCurrentBundle:(NSString *)remoteBundleUrl)
 {
-    usingTestFolder = shouldUseTestFolder;
+    if ([CodePush isUsingTestConfiguration]) {
+        [CodePushPackage downloadAndReplaceCurrentBundle:remoteBundleUrl];
+    }
 }
 
 @end

CodePushPackage.m

@@ -14,7 +14,12 @@ @implementation CodePushPackage
 
 + (NSString *)getCodePushPath
 {
-    return [[CodePush getApplicationSupportDirectory] stringByAppendingPathComponent:@"CodePush"];
+    NSString* codePushPath = [[CodePush getApplicationSupportDirectory] stringByAppendingPathComponent:@"CodePush"];
+    if ([CodePush isUsingTestConfiguration]) {
+        codePushPath = [codePushPath stringByAppendingPathComponent:@"TestPackages"];
+    }
+    
+    return codePushPath;
 }
 
 + (NSString *)getDownloadFilePath
@@ -481,4 +486,31 @@ + (void)rollbackPackage
     [self updateCurrentPackageInfo:info error:&error];
 }
 
++ (void)downloadAndReplaceCurrentBundle:(NSString *)remoteBundleUrl
+{
+    NSURL *urlRequest = [NSURL URLWithString:remoteBundleUrl];
+    NSError *error = nil;
+    NSString *downloadedBundle = [NSString stringWithContentsOfURL:urlRequest
+                                                encoding:NSUTF8StringEncoding
+                                                   error:&error];
+    
+    if (error) {
+        NSLog(@"Error downloading from URL %@", remoteBundleUrl);
+    } else {
+        NSString *currentPackageBundlePath = [self getCurrentPackageBundlePath:&error];
+        [downloadedBundle writeToFile:currentPackageBundlePath
+                           atomically:YES
+                             encoding:NSUTF8StringEncoding
+                                error:&error];
+    }
+}
+
++ (void)clearTestUpdates
+{
+    if ([CodePush isUsingTestConfiguration]) {
+        [[NSFileManager defaultManager] removeItemAtPath:[self getCodePushPath] error:nil];
+        [[NSFileManager defaultManager] removeItemAtPath:[self getStatusFilePath] error:nil];
+    }
+}
+
 @end