flyweight_es6 + specs

ntedgi · ntedgi · commit fb993490ad1e · 2018-04-09T13:06:32.000+03:00
diff --git a/src/structural/flyweight/flyweight_es6.js b/src/structural/flyweight/flyweight_es6.js
@@ -0,0 +1,19 @@
+class Color {
+    constructor(name){
+        this.name = name
+    }
+}
+
+class colorFactory {
+    constructor(name){
+        this.colors = {};
+    }
+    create(name) {
+        let color = this.colors[name];
+        if(color) return color;
+        this.colors[name] = new Color(name);
+        return this.colors[name];
+    }
+};
+
+module.exports = colorFactory;
diff --git a/test/flyweight_es6-test.js b/test/flyweight_es6-test.js
@@ -0,0 +1,17 @@
+const expect = require('chai').expect;
+
+const colorFactory = require('../src/structural/flyweight/flyweight_es6');
+
+describe('flyweight tests', () => {
+
+    it('sanity', () => {
+        const cf = new colorFactory();
+        cf.create('RED');
+        cf.create('RED');
+        cf.create('RED');
+        cf.create('YELLOW');
+        cf.create('YELLOW');
+        expect(Object.keys(cf.colors)).to.have.lengthOf(2);
+    });
+
+});
