1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175
|
require_relative 'base_shared'
RSpec.shared_examples :channel_buffered_buffer do
it_behaves_like :channel_buffer
context '#initialize' do
it 'raises an exception if size <= 0' do
expect {
described_class.new(0)
}.to raise_error(ArgumentError)
end
end
context '#capacity' do
it 'returns the maximum capacity of the buffer' do
subject = described_class.new(10)
expect(subject.capacity).to eq 10
end
end
context '#size' do
it 'is 0 when first created' do
expect(subject.size).to eq 0
end
it 'returns the number of items in the buffer' do
fill = subject.capacity / 2
fill.times { subject.put(:foo) }
expect(subject.size).to eq fill
end
it 'is 0 when there are taking threads but no putting threads' do
t = in_thread { subject.take }
t.join(0.1)
expect(subject.size).to eq 0
t.kill # cleanup
end
end
context '#empty?' do
it 'returns true when empty' do
subject = described_class.new(10)
expect(subject).to be_empty
end
end
context '#put' do
it 'enqueues the item when size > 0, not full, and not closed' do
subject.put(:foo)
expect(subject).to_not be_empty
end
it 'returns true when the item is put' do
expect(subject.put(:foo)).to be true
end
end
context '#offer' do
it 'enqueues the item immediately when not full and not closed' do
subject.offer(:foo)
expect(subject.take).to eq :foo
end
end
context '#take' do
it 'returns the first item when not empty' do
subject.put(:foo)
expect(subject.take).to eq :foo
end
it 'blocks until not empty' do
subject # initialize on this thread
bucket = Concurrent::AtomicReference.new(nil)
t = in_thread do
bucket.value = subject.take
end
t.join(0.1)
before = bucket.value
subject.put(42)
t.join(0.1)
after = bucket.value
expect(before).to be nil
expect(after).to eq 42
expect(t.status).to be false
end
it 'returns Concurrent::NULL when closed and empty' do
subject.close
expect(subject.take).to eq Concurrent::NULL
end
end
context '#next' do
it 'blocks until not empty' do
subject # initialize on this thread
bucket = Concurrent::AtomicReference.new([])
t = in_thread do
bucket.value = subject.next
end
t.join(0.1)
before = bucket.value
subject.put(42)
t.join(0.1)
after = bucket.value
expect(before).to eq []
expect(after.first).to eq 42
expect(after.last).to be true
expect(t.status).to be false
end
it 'returns <item>, true when there is only one item and not closed' do
subject.offer(42)
item, more = subject.next
expect(item).to eq 42
expect(more).to be true
end
it 'returns <item>, true when there are multiple items' do
subject.offer(:foo)
subject.offer(:bar)
subject.offer(:baz)
item1, more1 = subject.next
item2, more2 = subject.next
item3, more3 = subject.next
expect(item1).to eq :foo
expect(more1).to be true
expect(item2).to eq :bar
expect(more2).to be true
expect(item3).to eq :baz
expect(more3).to be true
end
it 'returns <item>, true when closed and last item' do
capacity = subject.capacity
expect(capacity).to be >= 1
capacity.times { subject.put(:foo) }
subject.close
capacity.times do
item, more = subject.next
expect(item).to eq :foo
expect(more).to be true
end
end
it 'returns Concurrent::NULL, false when closed and no items remain' do
capacity = subject.capacity
expect(capacity).to be >= 1
capacity.times { subject.put(:foo) }
subject.close
capacity.times { subject.next }
item, more = subject.next
expect(item).to eq Concurrent::NULL
expect(more).to be false
end
end
end
|