1 files changed, 48 insertions, 0 deletions
diff --git a/spec/ruby/language/predefined/data_spec.rb b/spec/ruby/language/predefined/data_spec.rb
new file mode 100644
index 0000000000..921d236ee9
--- /dev/null
+++ b/spec/ruby/language/predefined/data_spec.rb
@@ -0,0 +1,48 @@
+require_relative '../../spec_helper'
+
+describe "The DATA constant" do
+  it "exists when the main script contains __END__" do
+    ruby_exe(fixture(__FILE__, "data1.rb")).chomp.should == "true"
+  end
+
+  it "does not exist when the main script contains no __END__" do
+    ruby_exe("puts Object.const_defined?(:DATA)").chomp.should == 'false'
+  end
+
+  it "does not exist when an included file has a __END__" do
+    ruby_exe(fixture(__FILE__, "data2.rb")).chomp.should == "false"
+  end
+
+  it "does not change when an included files also has a __END__" do
+    ruby_exe(fixture(__FILE__, "data3.rb")).chomp.should == "data 3"
+  end
+
+  it "is included in an otherwise empty file" do
+    ap = fixture(__FILE__, "print_data.rb")
+    str = ruby_exe(fixture(__FILE__, "data_only.rb"), options: "-r#{ap}")
+    str.chomp.should == "data only"
+  end
+
+  it "returns a File object with the right offset" do
+    ruby_exe(fixture(__FILE__, "data_offset.rb")).should == "File\n121\n"
+  end
+
+  it "is set even if there is no data after __END__" do
+    ruby_exe(fixture(__FILE__, "empty_data.rb")).should == "31\n\"\"\n"
+  end
+
+  it "is set even if there is no newline after __END__" do
+    path = tmp("no_newline_data.rb")
+    code = File.binread(fixture(__FILE__, "empty_data.rb"))
+    touch(path, "wb") { |f| f.write code.chomp }
+    begin
+      ruby_exe(path).should == "30\n\"\"\n"
+    ensure
+      rm_r path
+    end
+  end
+
+  it "rewinds to the head of the main script" do
+    ruby_exe(fixture(__FILE__, "data5.rb")).chomp.should == "DATA.rewind"
+  end
+end