feat(classes): Add counting of Scala classes

There was no support for collecting Scala class number so I
added it. Tested with the unit tests and with SonarQube
6.3.1. Everything seemed OK.

src/main/scala/com/sagacify/sonar/scala/Measures.scala

@@ -1,17 +1,20 @@
 package com.sagacify.sonar.scala
 
 import scala.annotation.tailrec
-
 import scalariform.lexer.ScalaLexer
 import scalariform.lexer.Token
-import scalariform.lexer.Tokens.LINE_COMMENT
-import scalariform.lexer.Tokens.MULTILINE_COMMENT
-import scalariform.lexer.Tokens.XML_COMMENT
-import scalariform.lexer.Tokens.WS
-import scalariform.lexer.Tokens.EOF
+import scalariform.lexer.Tokens._
 
 object Measures {
 
+  final def count_classes(tokens: List[Token], i: Int = 0): Int = {
+    var count = 0
+    tokens.foreach(token => if (token.tokenType == CLASS || token.tokenType == OBJECT) count += 1)
+
+    count
+  }
+
+
   /* applied on raw source code */
 
   /* applied on lines of code */

src/main/scala/com/sagacify/sonar/scala/ScalaSensor.scala

@@ -36,7 +36,7 @@ class ScalaSensor(scala: Scala, fs: FileSystem) extends Sensor {
                           CM.NCLOC,
                           Measures.count_ncloc(tokens))
 
-      // context.saveMeasure(input, CM.CLASSES, classes)
+      context.saveMeasure(inputFile, CM.CLASSES, Measures.count_classes(tokens))
       // context.saveMeasure(input, CM.FUNCTIONS, methods)
       // context.saveMeasure(input, CM.ACCESSORS, accessors)
       // context.saveMeasure(input, CM.COMPLEXITY_IN_FUNCTIONS, complexityInMethods)

src/test/resources/ScalaFile2.scala

@@ -7,3 +7,6 @@ class ScalaFile2 {
     println("function called.")
   }
 }
+
+object ScalaFile2 {
+}

src/test/scala/com/sagacify/sonar/scala/BaseMetricSensorSpec.scala

@@ -67,6 +67,8 @@ class ScalaSensorSpec extends FlatSpec with Matchers {
           .saveMeasure(file, CM.FILES, 1)
       verify(sensorContext, times(1))
           .saveMeasure(file, CM.COMMENT_LINES, 0)
+      verify(sensorContext, times(1))
+          .saveMeasure(file, CM.CLASSES, 1)
 
     }
   }
@@ -87,7 +89,8 @@ class ScalaSensorSpec extends FlatSpec with Matchers {
           .saveMeasure(file, CM.FILES, 1)
       verify(sensorContext, times(1))
           .saveMeasure(file, CM.COMMENT_LINES, 1)
-
+      verify(sensorContext, times(1))
+        .saveMeasure(file, CM.CLASSES, 2)
     }
   }
 }