Add createChild method to Resource

Author: s0ph1e

lib/file-handlers/css.js

@@ -1,7 +1,6 @@
 var _ = require('underscore');
 var Promise = require('bluebird');
 var getCssUrls = require('css-url-parser');
-var Resource = require('../resource');
 var utils = require('../utils');
 
 function loadCss (context, resource) {
@@ -12,8 +11,7 @@ function loadCss (context, resource) {
 
 	var promises = _.map(cssUrls, function loadResourceFromCssUrl (cssUrl) {
 		var resourceUrl = utils.getUrl(url, cssUrl);
-		var cssResource = new Resource(resourceUrl);
-		cssResource.setParent(resource);
+		var cssResource = resource.createChild(resourceUrl);
 
 		return context.loadResource(cssResource).then(function handleLoadedSource (loadedResource) {
 			var relativePath = utils.getRelativePath(filename, loadedResource.getFilename());

lib/file-handlers/html.js

@@ -1,7 +1,6 @@
 var cheerio = require('cheerio');
 var Promise = require('bluebird');
 var utils = require('../utils');
-var Resource = require('../resource');
 
 function loadHtml (context, resource) {
 	var sources = context.getHtmlSources();
@@ -50,8 +49,7 @@ function loadResources (context, resource, source) {
 
 		if (attr) {
 			var resourceUrl = utils.getUrl(url, attr);
-			var htmlResource = new Resource(resourceUrl);
-			htmlResource.setParent(resource);
+			var htmlResource = resource.createChild(resourceUrl);
 			htmlResource.setHtmlData({ tagName: el[0].name, attributeName: source.attr });
 
 			return context.loadResource(htmlResource).then(function handleLoadedSource (loadedResource) {

lib/resource.js

@@ -15,6 +15,17 @@ function Resource (url, filename) {
 	this.filename = filename;
 }
 
+Resource.prototype.createChild = function createChild (url, filename) {
+	var child = new Resource(url, filename);
+
+	var currentDepth = this.getDepth();
+
+	child.setParent(this);
+	child.setDepth(++currentDepth);
+
+	return child;
+};
+
 Resource.prototype.getUrl = function getUrl () {
 	return this.url;
 };
@@ -43,6 +54,14 @@ Resource.prototype.setParent = function setParent (parent) {
 	this.parent = parent;
 };
 
+Resource.prototype.getDepth = function getDepth () {
+	return this.depth || 0;
+};
+
+Resource.prototype.setDepth = function setDepth (depth) {
+	this.depth = depth;
+};
+
 /**
  *
  * @param {Object} data - html element data

test/unit/resource-test.js

@@ -4,7 +4,7 @@ var Resource = require('../../lib/resource');
 var types = require('../../lib/config/resource-types');
 
 describe('Resource', function() {
-	describe('#Resource', function() {
+	describe('#getType', function() {
 		it('should return correct type based on extension', function() {
 			var html = new Resource('http://example.com', 'index.html');
 			var htm = new Resource('http://example.com', 'index.htm');
@@ -68,4 +68,56 @@ describe('Resource', function() {
 			res.getType().should.be.eql(types.other);
 		});
 	});
+
+	describe('#setDepth', function () {
+		it('should set depth', function() {
+			var o = new Resource('http://google.com');
+			o.setDepth(555);
+			o.depth.should.be.eql(555);
+		});
+	});
+
+	describe('#getDepth', function () {
+		it('should return depth if object has it', function() {
+			var o = new Resource('http://google.com');
+			o.setDepth(123);
+			o.getDepth().should.be.eql(123);
+		});
+
+		it('should return 0 if object has no depth', function() {
+			var o = new Resource('http://google.com');
+			o.getDepth().should.be.eql(0);
+		});
+
+	});
+
+	describe('#createChild', function () {
+		it('should return Resource', function() {
+			var parent = new Resource('http://example.com');
+			var child = parent.createChild('http://google.com');
+			child.should.be.instanceOf(Resource);
+		});
+
+		it('should set correct url and filename', function() {
+			var parent = new Resource('http://example.com');
+			var child = parent.createChild('http://google.com', 'google.html');
+			child.getUrl().should.be.eql('http://google.com');
+			child.getFilename().should.be.eql('google.html');
+		});
+
+		it('should set parent', function() {
+			var parent = new Resource('http://example.com');
+			var child = parent.createChild('http://google.com');
+			child.parent.should.be.equal(parent);
+		});
+
+		it('should set depth', function() {
+			var parent = new Resource('http://example.com');
+			var child = parent.createChild('http://google.com');
+			child.getDepth().should.be.eql(1);
+
+			var childOfChild = child.createChild('http://google.com.ua');
+			childOfChild.getDepth().should.be.eql(2);
+		});
+	});
 });