Don't throw an exception if no argument is passed to new

oscardelben · May 1, 2012 · 9addd4a · 9addd4a
1 parent df6d990
commit 9addd4a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/lib/sheet/write.rb b/lib/sheet/write.rb
@@ -3,11 +3,12 @@ class Sheet::Write
 
   attr_accessor :name
 
-  def initialize(name)
+  def initialize(name=nil)
     @name = name
   end
 
   def write
+    return Sheet.write("Please specify a name") unless name
     create_dir_if_doesnt_exist
     if editor_is_set?
       Sheet.exec("#{Sheet.editor} #{Sheet.sheet_path(name)}", true)

diff --git a/spec/sheet/write_spec.rb b/spec/sheet/write_spec.rb
@@ -8,6 +8,11 @@
     Sheet.stub(:sheets_directory_exists?) { true }
   end
 
+  it 'shows an error message if no argument is passed' do
+    Sheet.should_receive(:write).with('Please specify a name')
+    Sheet::Write.new.write
+  end
+
   it 'opens a new file for writing' do
     cmd = "#{editor} #{Sheet.sheet_path('tmux')}"
     Sheet.should_receive(:exec).with(cmd, true)