From 54af5fec96d802dfa9598ea67cb8d6ac40f7cee3 Mon Sep 17 00:00:00 2001
From: Shane Kilkelly <shane@kilkelly.me>
Date: Thu, 16 Mar 2017 16:35:51 +0000
Subject: [PATCH] Upgrade metrics

---
 services/spelling/app.coffee                         |  1 -
 .../spelling/app/coffee/LearnedWordsManager.coffee   | 12 +++++++++++-
 .../test/unit/coffee/LearnedWordsManagerTests.coffee |  1 +
 3 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/services/spelling/app.coffee b/services/spelling/app.coffee
index 7e8c4d69bb..bdefe9224c 100644
--- a/services/spelling/app.coffee
+++ b/services/spelling/app.coffee
@@ -11,7 +11,6 @@ server = express()
 bodyParser = require('body-parser')
 metrics = require("metrics-sharelatex")
 metrics.initialize("spelling")
-metrics.mongodb.monitor(Path.resolve(__dirname + "/node_modules/mongojs/node_modules/mongodb"), logger)
 metrics.memory.monitor(logger)
 HealthCheckController = require("./app/js/HealthCheckController")
 
diff --git a/services/spelling/app/coffee/LearnedWordsManager.coffee b/services/spelling/app/coffee/LearnedWordsManager.coffee
index cf99544876..a9c1326be5 100644
--- a/services/spelling/app/coffee/LearnedWordsManager.coffee
+++ b/services/spelling/app/coffee/LearnedWordsManager.coffee
@@ -29,4 +29,14 @@ module.exports = LearnedWordsManager =
 			mongoCache.set(user_token, words)
 			callback null, words
 
-		
+
+metrics.timeAsyncMethod(
+	LearnedWordsManager, 'learnWord',
+	'LearnedWordsManager.learnWord',
+	logger
+)
+metrics.timeAsyncMethod(
+	LearnedWordsManager, 'getLearnedWords',
+	'LearnedWordsManager.getLearnedWords',
+	logger
+)
diff --git a/services/spelling/test/unit/coffee/LearnedWordsManagerTests.coffee b/services/spelling/test/unit/coffee/LearnedWordsManagerTests.coffee
index 6cbe6af9c9..09d9dc4440 100644
--- a/services/spelling/test/unit/coffee/LearnedWordsManagerTests.coffee
+++ b/services/spelling/test/unit/coffee/LearnedWordsManagerTests.coffee
@@ -23,6 +23,7 @@ describe "LearnedWordsManager", ->
 				log:->
 				err:->
 				info:->
+			'metrics-sharelatex': {timeAsyncMethod: sinon.stub(), inc: sinon.stub()}
 
 	describe "learnWord", ->
 		beforeEach ->