1 files changed, 76 insertions, 0 deletions
diff --git a/web3.extend.js b/web3.extend.js
new file mode 100644
index 00000000..dfa863a2
--- /dev/null
+++ b/web3.extend.js
@@ -0,0 +1,76 @@
+var chai = require('chai');
+var assert = chai.assert;
+var FakeHttpProvider = require('./helpers/FakeHttpProvider');
+var web3 = require('../lib/web3');
+
+
+var tests = [{
+    properties: [new web3._extend.Property({
+        name: 'gasPrice',
+        getter: 'eth_gasPrice',
+        outputFormatter: web3._extend.formatters.outputBigNumberFormatter
+    })]
+},{
+    methods: [new web3._extend.Method({
+        name: 'getBalance',
+        call: 'eth_getBalance',
+        params: 2,
+        inputFormatter: [web3._extend.utils.toAddress, web3._extend.formatters.inputDefaultBlockNumberFormatter],
+        outputFormatter: web3._extend.formatters.outputBigNumberFormatter
+    })]
+},{
+    property: 'admin',
+    properties: [new web3._extend.Property({
+        name: 'gasPrice',
+        getter: 'eth_gasPrice',
+        outputFormatter: web3._extend.formatters.outputBigNumberFormatter
+    })],
+    methods: [new web3._extend.Method({
+        name: 'getBalance',
+        call: 'eth_getBalance',
+        params: 2,
+        inputFormatter: [web3._extend.utils.toAddress, web3._extend.formatters.inputDefaultBlockNumberFormatter],
+        outputFormatter: web3._extend.formatters.outputBigNumberFormatter
+    })]
+}];
+
+describe('web3', function () {
+    describe('_extend', function () {
+        tests.forEach(function (test, index) {
+            it('test no: ' + index, function () {
+                web3._extend(test);
+
+
+                if(test.properties)
+                    test.properties.forEach(function(property){
+
+                        var provider = new FakeHttpProvider();
+                        web3.setProvider(provider);
+                        provider.injectResult('');
+                        provider.injectValidation(function (payload) {
+                            assert.equal(payload.jsonrpc, '2.0');
+                            assert.equal(payload.method, property.getter);
+                        });
+
+                        if(test.property) {
+                            assert.isObject(web3[test.property][property.name]);
+                            assert.isFunction(web3[test.property]['get'+ property.name.charAt(0).toUpperCase() + property.name.slice(1)]);
+                        } else {
+                            assert.isObject(web3[property.name]);
+                            assert.isFunction(web3['get'+ property.name.charAt(0).toUpperCase() + property.name.slice(1)]);                        
+                        }
+                    });
+
+                if(test.methods)
+                    test.methods.forEach(function(property){
+                        if(test.property)
+                            assert.isFunction(web3[test.property][property.name]);
+                        else
+                            assert.isFunction(web3[property.name]);
+                    });
+
+            });
+        });
+    });
+});
+