diff --git a/app_generators/watircraft/watircraft_generator.rb b/app_generators/watircraft/watircraft_generator.rb index 65de7a6..188b39d 100644 --- a/app_generators/watircraft/watircraft_generator.rb +++ b/app_generators/watircraft/watircraft_generator.rb @@ -69,6 +69,7 @@ def manifest lib/init lib/pages lib/steps + lib/methods test test/specs test/features diff --git a/spec/method_generator_spec.rb b/spec/method_generator_spec.rb index 34ad3ed..b2f7427 100644 --- a/spec/method_generator_spec.rb +++ b/spec/method_generator_spec.rb @@ -7,7 +7,7 @@ include Helpers::Taza before :each do - generate_project + generate_project ["--site=frito"] end after :each do @@ -15,7 +15,13 @@ end it "should be able to generate a method" do - run_generator('spec', ['empty_shopping_cart'], generator_sources) + run_generator('method', ['empty_shopping_cart'], generator_sources) end - -end \ No newline at end of file + + it "should work when when the provided name includes a space" do + MethodGenerator.any_instance.stubs(:configured_site).returns('frito') + generator = MethodGenerator.new(['check out']) + generator.name.should == 'check_out' + end + +end diff --git a/watircraft_generators/method/method_generator.rb b/watircraft_generators/method/method_generator.rb index c5a3c64..b2205b3 100644 --- a/watircraft_generators/method/method_generator.rb +++ b/watircraft_generators/method/method_generator.rb @@ -12,7 +12,7 @@ class MethodGenerator < RubiGen::Base def initialize(runtime_args, runtime_options = {}) super usage if args.empty? - @name = args.shift + @name = args.shift.computerize @site_name = configured_validated_site extract_options end