spec/analyzer_spec.rb in fukuzatsu-0.9.13 vs spec/analyzer_spec.rb in fukuzatsu-0.9.14

@@ -34,26 +34,70 @@
         expect(Analyzer.new(content_5).extract_class_name).to eq("Something")
       end
     end
   end
 
-  context "program_1" do
+  describe "#complexity" do
+    context "program_1" do
 
-    let(:analyzer)  { Analyzer.new(content_1) }
+      let(:analyzer)  { Analyzer.new(content_1) }
 
-    it "matches the manual calculation of complexity of 4" do
-      expect(analyzer.complexity).to eq(4)
+      it "matches the manual calculation of complexity of 4" do
+        expect(analyzer.complexity).to eq(4)
+      end
+
     end
 
-  end
+    context "program_2" do
 
-  context "program_2" do
+      let(:analyzer)  { Analyzer.new(content_2) }
 
-    let(:analyzer)  { Analyzer.new(content_2) }
+      it "matches the manual calculation of complexity of 5" do
+        expect(analyzer.complexity).to eq(5)
+      end
 
-    it "matches the manual calculation of complexity of 5" do
-      expect(analyzer.complexity).to eq(5)
     end
-
+  end
+  
+  describe "extract_methods" do
+    # Note: should implicitly trust private method #methods_from
+    context "from a file with a single method" do
+      let(:analyzer) { Analyzer.new(File.read("spec/fixtures/single_method.rb")) }
+      it "should return a single method" do
+        expect(analyzer.extract_methods.count).to eq 1
+      end
+      it "should extract the method name" do
+        expect(analyzer.extract_methods[0].name).to eq "read_poem"
+      end
+      it "should extract the method content" do
+        expect(analyzer.extract_methods[0].content).to eq 'def read_poem
+  return "I meant to find her when I came/Death had the same design"
+end'
+      end
+      it "should set type to :instance" do
+        expect(analyzer.extract_methods[0].type).to eq :instance
+      end
+    end
+    context "from a file with multiple methods" do
+      let(:analyzer) { Analyzer.new(File.read("spec/fixtures/multiple_methods.rb")) }
+      it "should return multiple methods" do
+        expect(analyzer.extract_methods.map { |m| m.name }).to eq ["bake_treats", "lower_from_window"]
+      end
+    end
+    context "from a file with nested methods" do
+      let(:analyzer) { Analyzer.new(File.read("spec/fixtures/nested_methods.rb")) }
+      it "should return the root method, and its child" do
+        expect(analyzer.extract_methods.map { |m| m.name }).to eq ["grow_flowers", "water_earth"]
+      end
+    end
+    context "from a file with a class" do
+      let(:analyzer) { Analyzer.new(File.read("spec/fixtures/single_class.rb")) }
+      it "should return the class and its methods" do
+        expect(analyzer.extract_methods.map { |m| m.name }).to eq ["Gown", "initialize", "color"]
+      end
+      it "should set the type of class to :class" do
+        expect(analyzer.extract_methods[0].type).to eq :class
+      end
+    end
   end
 
 end