From fa7e67e394331e3a501fce48f51655bbf325acd2 Mon Sep 17 00:00:00 2001
From: Tobias Adam <tow8ie@gmail.com>
Date: Wed, 14 Jan 2015 17:48:19 +0100
Subject: [PATCH] Raises an error if styleguide section comment contains no
 section definition.

---
 lib/kss/section.rb   | 13 ++++++++++++-
 test/section_test.rb |  6 ++++++
 2 files changed, 18 insertions(+), 1 deletion(-)

diff --git a/lib/kss/section.rb b/lib/kss/section.rb
index ec08a82..844590f 100644
--- a/lib/kss/section.rb
+++ b/lib/kss/section.rb
@@ -1,4 +1,8 @@
 module Kss
+
+  class SectionNotDefinedError < StandardError
+  end
+
   # Public: Represents a styleguide section. Each section describes one UI
   # element. A Section can be thought of as the collection of the description,
   # modifiers, and styleguide reference.
@@ -35,7 +39,14 @@ def section
       return @section unless @section.nil?
 
       cleaned  = section_comment.strip.sub(/\.$/, '') # Kill trailing period
-      @section = cleaned.match(/Styleguide (.+)/i)[1]
+      match_result = cleaned.match(/Styleguide (.+)/i)
+
+      if match_result
+        @section = match_result[1]
+      else
+        raise SectionNotDefinedError.new,
+          'No section string found in comment block.'
+      end
 
       @section
     end
diff --git a/test/section_test.rb b/test/section_test.rb
index 29e4138..021f143 100644
--- a/test/section_test.rb
+++ b/test/section_test.rb
@@ -39,6 +39,12 @@ def setup
     assert_equal '2.1.1', @section.section
   end
 
+  test "raises an error if styleguide reference is not set" do
+    assert_raises Kss::SectionNotDefinedError do
+      Kss::Section.new('', 'example.css').section
+    end
+  end
+
   test "parses word phrases as styleguide references" do
     @comment_text.gsub!('2.1.1', 'Buttons - Truly Lime')
     section = Kss::Section.new(@comment_text, 'example.css')