From a6f899f06258ca0b01eb33d965b10cfcde74a92a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Maciej=20Ma=C5=82ecki?= <me@mmalecki.com>
Date: Wed, 16 Apr 2014 10:12:52 +0200
Subject: [PATCH] Add a unit test for ticker

---
 test/unit/traderTickerTest.js | 55 +++++++++++++++++++++++++++++++++++
 1 file changed, 55 insertions(+)
 create mode 100644 test/unit/traderTickerTest.js

diff --git a/test/unit/traderTickerTest.js b/test/unit/traderTickerTest.js
new file mode 100644
index 00000000..62845a82
--- /dev/null
+++ b/test/unit/traderTickerTest.js
@@ -0,0 +1,55 @@
+'use strict';
+
+var assert = require('chai').assert;
+var hock = require('hock');
+var uuid = require('node-uuid').v4;
+var Trader = require('../../lib/trader.js');
+var PostgresqlInterface = require('../../lib/protocol/db/postgresql_interface.js');
+
+var db = 'psql://lamassu:lamassu@localhost/lamassu-test';
+var psqlInterface = new PostgresqlInterface(db);
+
+var CURRENCY = 'USD';
+
+describe('trader/send', function () {
+  var trader = new Trader(psqlInterface);
+  trader.config = {
+    exchanges: {
+      settings: { currency: CURRENCY }
+    }
+  };
+
+  it('should call `balance` on the transfer exchange', function (done) {
+    trader.transferExchange = {
+      balance: function (callback) {
+        callback(null, 100);
+      }
+    };
+
+    trader.pollBalance(function (err, balance) {
+      assert.notOk(err);
+      assert.equal(trader.balance.transferBalance, 100);
+      assert.ok(trader.balance.timestamp);
+      done();
+    });
+  });
+
+  it('should call `ticker` on the ticker exchange', function (done) {
+    trader.tickerExchange = {
+      ticker: function (currency, callback) {
+        assert.equal(currency, CURRENCY);
+        callback(null, 100);
+      }
+    };
+
+    trader.pollRate(function (err, rate) {
+      var rate;
+
+      assert.notOk(err);
+      rate = trader.rate(CURRENCY);
+      assert.equal(rate.rate, 100);
+      assert.ok(rate.timestamp);
+      done();
+    });
+  });
+});