Add some tests

.gitignore

@@ -0,0 +1 @@
+/Gemfile.lock

Gemfile

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+source "https://rubygems.org"
+
+gemspec

Rakefile

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+require 'minitest/test_task'
+
+Minitest::TestTask.create
+
+task default: :test

lib/ndjson.rb

@@ -34,5 +34,8 @@ def write obj
       @output.puts JSON.generate(obj)
     end
 
+    def flush
+      @output.flush
+    end
   end
 end

ndjson.gemspec

@@ -8,5 +8,8 @@ Gem::Specification.new do |s|
   s.email       = '[email protected]'
   s.files       = ['lib/ndjson.rb']
   s.homepage    = 'https://github.com/ndjson/ndjson.rb'
-  s.license       = 'MIT'
-end
+  s.license     = 'MIT'
+
+  s.add_development_dependency "minitest"
+  s.add_development_dependency "rake"
+end

test/example.ndjson

@@ -0,0 +1,3 @@
+ {"some":"thing"}
+ {"foo":17,"bar":false,"quux":true}
+ {"may":{"include":"nested","objects":["and","arrays"]}}

test/ndjson_test.rb

@@ -0,0 +1,48 @@
+# frozen_string_literal: true
+
+require_relative 'test_helper'
+
+module NDJSONTest
+  class ParserTest < Minitest::Test
+    def test_initialize_with_path
+      parser = NDJSON::Parser.new(File.expand_path('example.ndjson', __dir__))
+      index = 0
+      parser.each do |entry|
+        assert_equal({ 'some' => 'thing' }, entry) if index.zero?
+        index += 1
+      end
+      assert_equal(index, 3)
+    end
+
+    def test_initialize_with_array
+      content = File.read(File.expand_path('example.ndjson', __dir__)).lines
+      parser = NDJSON::Parser.new(content)
+      index = 0
+      parser.each do |entry|
+        assert_equal({ 'some' => 'thing' }, entry) if index.zero?
+        index += 1
+      end
+      assert_equal(index, 3)
+    end
+  end
+
+  class GeneratorTest < Minitest::Test
+    def test_initialize_with_path
+      file = Tempfile.new('example.ndjson')
+      assert_equal(String, file.path.class)
+      path = file.path
+      generator = NDJSON::Generator.new(path)
+      generator.write({ 'some' => 'thing' })
+      generator.flush
+      assert_equal(%({"some":"thing"}\n), File.read(path))
+    end
+
+    def test_initialize_with_stringio
+      io = StringIO.new
+      generator = NDJSON::Generator.new(io)
+      generator.write({ 'some' => 'thing' })
+      io.rewind
+      assert_equal(%({"some":"thing"}\n), io.read)
+    end
+  end
+end

test/test_helper.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+$LOAD_PATH.unshift File.expand_path('../lib', __dir__)
+require 'ndjson'
+
+require 'tempfile'
+require 'stringio'
+require 'minitest/autorun'