Skip to content

Commit

Permalink
Define chunk size in before :each clause
Browse files Browse the repository at this point in the history
  • Loading branch information
thisismydesign committed May 29, 2017
1 parent a71c8a7 commit 1a265cc
Showing 1 changed file with 18 additions and 18 deletions.
36 changes: 18 additions & 18 deletions spec/json/streamer/json_streamer_spec.rb
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@
@example_key = 'key'
@example_value = 'value'
@example_hash = {@example_key => @example_value}
@chunk_size = 10
end

describe '#get' do
Expand All @@ -18,7 +19,7 @@

hash = {}
json_file_mock = StringIO.new(JSON.generate(hash))
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, 10)
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, @chunk_size)

objects = []
streamer.get(nesting_level:0) do |object|
Expand All @@ -35,7 +36,7 @@

hash = {'object1':@example_hash, 'object2':@example_hash, 'object3':@example_hash}
json_file_mock = StringIO.new(JSON.generate(hash))
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, 10)
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, @chunk_size)

objects = []
streamer.get(nesting_level:1) do |object|
Expand All @@ -54,7 +55,7 @@

array = Array.new(10) {@example_hash}
json_file_mock = StringIO.new(JSON.generate(array))
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, 10)
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, @chunk_size)

objects = []
streamer.get(nesting_level:1) do |object|
Expand Down Expand Up @@ -99,9 +100,8 @@
objects.push(object)
end

chunk_size = 10
json_file_mock = StringIO.new(JSON.generate(hash))
json_file_mock.each(chunk_size) do |chunk|
json_file_mock.each(@chunk_size) do |chunk|
streamer.parser << chunk
end

Expand All @@ -122,7 +122,7 @@
(1..max_levels).each do |max_level|
hash = NDHash.generate(levels: max_level, values_per_level: 0, hashes_per_level: entries_per_level)
json_file_mock = StringIO.new(JSON.generate(hash))
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, 10)
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, @chunk_size)

objects = []
streamer.get(nesting_level:max_level-1) do |object|
Expand All @@ -139,7 +139,7 @@

hash = {obj:@example_hash}
json_file_mock = StringIO.new(JSON.generate(hash))
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, 10)
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, @chunk_size)

streamer.get(nesting_level:0) {}

Expand All @@ -152,7 +152,7 @@

hash = {obj:@example_hash}
json_file_mock = StringIO.new(JSON.generate(hash))
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, 10)
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, @chunk_size)

objects = []
streamer.get(nesting_level:2) do |object|
Expand All @@ -169,7 +169,7 @@

hash = {obj:@example_hash}
json_file_mock = StringIO.new(JSON.generate(hash))
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, 10)
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, @chunk_size)

objects = []
streamer.get(nesting_level:2, yield_values:false) do |object|
Expand All @@ -184,7 +184,7 @@
it 'should yield value within JSON object' do

json_file_mock = StringIO.new(JSON.generate(@example_hash))
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, 10)
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, @chunk_size)

objects = []
streamer.get(key:@example_key) do |object|
Expand All @@ -203,7 +203,7 @@

hash = {obj1:@example_hash, obj2:@example_hash, obj3:@example_hash}
json_file_mock = StringIO.new(JSON.generate(hash))
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, 10)
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, @chunk_size)

objects = []
streamer.get(key:@example_key) do |object|
Expand All @@ -222,7 +222,7 @@

hash = {'obj1' => @example_hash, @example_key => @example_value}
json_file_mock = StringIO.new(JSON.generate(hash))
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, 10)
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, @chunk_size)

objects = []
streamer.get(key:@example_key) do |object|
Expand All @@ -241,7 +241,7 @@

hash = {'obj1' => @example_hash, @example_key => @example_value, 'obj2' => {@example_key => @example_hash}}
json_file_mock = StringIO.new(JSON.generate(hash))
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, 10)
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, @chunk_size)

objects = []
streamer.get(key:@example_key) do |object|
Expand All @@ -261,7 +261,7 @@

hash = [[@example_hash, @example_hash, @example_hash]]
json_file_mock = StringIO.new(JSON.generate(hash))
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, 10)
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, @chunk_size)

objects = []
streamer.get(nesting_level:2) do |object|
Expand All @@ -280,7 +280,7 @@

hash = [[@example_hash, @example_hash, @example_hash]]
json_file_mock = StringIO.new(JSON.generate(hash))
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, 10)
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, @chunk_size)

objects = []
streamer.get(nesting_level:1) do |object|
Expand All @@ -299,7 +299,7 @@

hash = [@example_value, @example_value]
json_file_mock = StringIO.new(JSON.generate(hash))
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, 10)
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, @chunk_size)

objects = []
streamer.get(nesting_level:0) do |object|
Expand All @@ -316,7 +316,7 @@

hash = [@example_value, @example_value]
json_file_mock = StringIO.new(JSON.generate(hash))
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, 10)
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, @chunk_size)

objects = []
streamer.get(nesting_level:1) do |object|
Expand Down Expand Up @@ -357,7 +357,7 @@

memory_consumption_before_parsing = GetProcessMem.new.mb

streamer = Json::Streamer::JsonStreamer.new(json_file_mock, 10)
streamer = Json::Streamer::JsonStreamer.new(json_file_mock, @chunk_size)
object_count = 0
streamer.get(nesting_level:1) do |object|
expect(object).to eq(@example_hash)
Expand Down

0 comments on commit 1a265cc

Please sign in to comment.