Merge pull request #2 from brandsExclusive/pass-new-env-variable

Pass a Custom database connection path

README.md

@@ -3,12 +3,21 @@ leQuel Library
 
 leQuel provides a simple API to connect to and query postgres database.
 
-### Requires
+### Default database path
 env.DATABASE_URL
 ```
 DATABASE_URL=postgresql://@localhost/my_db
 ```
+### For custom database path
+env.MY_DATABASE_URL
+```
+MY_DATABASE_URL=postgresql://@localhost/my_db_test
+```
 
+```javascript
+const lequel = require('lib-lequel');
+lequel.setDatabaseUrl(process.env.MY_DATABASE_URL)
+```
 
 ### Examples
 ```javascript
@@ -22,5 +31,3 @@ query.paginate(5, 5);
 
 let result = lequel.getRows(...query.finalize());
 ```
-
-

index.js

@@ -1 +1 @@
-module.exports = require('./lib');
+module.exports = require('./lib');

lib/index.js

@@ -1,10 +1,15 @@
 const Pool = require("pg-pool"),
   url = require("url");
 
-if (!process.env.DATABASE_URL) {
-  throw new Error("Environment variable DATABASE_URL is not set");
+let databaseUrl = process.env.DATABASE_URL
+
+if (!databaseUrl) {
+  throw new Error("Environment variable is not set");
 }
-const params = url.parse(process.env.DATABASE_URL);
+module.exports.setDatabaseUrl = (url) => databaseUrl = url
+module.exports.getDatabaseUrl = () => databaseUrl
+
+const params = url.parse(databaseUrl);
 let ssl = (process.env.NODE_ENV === 'production'),
   auth = params.auth.split(':'),
   config = {
@@ -44,4 +49,4 @@ module.exports.getCount = (text, values) => {
 };
 
 module.exports.createGeneric = require('./createGeneric');
-module.exports.queryBuilder = require('./queryBuilder');
+module.exports.queryBuilder = require('./queryBuilder');

test/index.js

@@ -0,0 +1,19 @@
+const expect = require('chai').expect
+
+describe('it can load', () => {
+  afterEach(() => {
+    process.env.DATABASE_URL = null
+  })
+  it('default env variable as database connection', () => {
+    process.env.DATABASE_URL = 'postgresql://@localhost/my_db';
+    const lequel = require('../lib/')
+    expect(lequel.getDatabaseUrl()).to.eql('postgresql://@localhost/my_db')
+  })
+
+  it('custom env variable as database connection', () => {
+    process.env.MY_DATABASE_URL = 'postgresql://@localhost/my_db_test';
+    const lequel = require('../lib/')
+    lequel.setDatabaseUrl(process.env.MY_DATABASE_URL)
+    expect(lequel.getDatabaseUrl()).to.eql('postgresql://@localhost/my_db_test')
+  })
+})