image_assertion.js

var ImageAsserter = (function() {

  function ImageAsserter(tuneUpPath, outputPath, refImagesPath) {

    if (!outputPath || !refImagesPath || !tuneUpPath) {

      throw new AssertionException("output, refImages, tuneUp pathes can't be null");
    }

    this.tuneUpPath     = tuneUpPath;
    this.outputPath     = outputPath;
    this.refImagesPath  = refImagesPath;

    var target          = UIATarget.localTarget();
    if (!target) {

      throw new AssertionException("unable to get localTarget");
    }

    this.host           = target.host();
    if (!this.host) {

      throw new AssertionException("unable to get current UAIHost");
    }
  }

  ImageAsserter.prototype.assertImageNamed = function(imageName) {

    var command,
        taskResult,
        assertSuccessfull = false,
        SUCCESS_EXIT_CODE = 0,
        TIMEOUT           = 5,
        args              = [this.outputPath, this.refImagesPath, imageName];

    command     = this.tuneUpPath + '/image_asserter';
    taskResult  = this.host.performTaskWithPathArgumentsTimeout(command,
                                                                args,
                                                                TIMEOUT);

    assertSuccessfull = (taskResult.exitCode === SUCCESS_EXIT_CODE);
    if (!assertSuccessfull) UIALogger.logError(taskResult.stderr);

    return assertSuccessfull;
  };

  return ImageAsserter;
}());

function createImageAsserter(tuneUpPath, outputPath, refImagesPath) {

  this.imageAsserter = new ImageAsserter(tuneUpPath, outputPath, refImagesPath);
}

function assertScreenMatchesImageNamed(imageName, message) {

  if (!this.imageAsserter) {
    throw new AssertionException("imageAsserter isn't created.");
  }

  UIATarget.localTarget().captureAppScreenWithName(imageName);
  UIATarget.localTarget().delay(1); // delay for screenshot to be saved

  var assertionPassed = this.imageAsserter.assertImageNamed(imageName);
  if (!assertionPassed) {

    if (!message) message = 'Assertion of the image ' + imageName + ' failed.';
    throw new AssertionException(message);
  }
}