diff --git a/lib/orthoses/missing_name.rb b/lib/orthoses/missing_name.rb index 26fd710..657957f 100644 --- a/lib/orthoses/missing_name.rb +++ b/lib/orthoses/missing_name.rb @@ -51,7 +51,7 @@ def call end missings.uniq! missings.each do |missing| - @store[missing].header = "module #{missing}" + @store[missing] end end diff --git a/lib/orthoses/missing_name_test.rb b/lib/orthoses/missing_name_test.rb index ec47f38..d7a5be5 100644 --- a/lib/orthoses/missing_name_test.rb +++ b/lib/orthoses/missing_name_test.rb @@ -17,6 +17,10 @@ class C3 extend M2 end end + class CC + module MM + end + end } def test_missing_name(t) store = Orthoses::MissingName.new( @@ -25,10 +29,15 @@ def test_missing_name(t) Orthoses::Utils.new_store.tap do |store| store["MissingNameTest::M1::M2"] << "CONST: 1" store["MissingNameTest::C1::C2"].header = "class MissingNameTest::C1::C2 < ::MissingNameTest::C1" + store["MissingNameTest::CC::MM"] end } ).call + store.each do |_, content| + content.auto_header + end + unless store.has_key?("MissingNameTest::M1::M2") t.error("MissingNameTest::M1::M2 not found in store") end @@ -59,5 +68,11 @@ def test_missing_name(t) unless store.has_key?("MissingNameTest") t.error("MissingNameTest not found in store") end + unless store["MissingNameTest::CC::MM"].header == "module MissingNameTest::CC::MM" + t.error("MissingNameTest::CC::MM should be module") + end + unless store["MissingNameTest::CC"].header == "class MissingNameTest::CC" + t.error("MissingNameTest::CC should be class") + end end end