summaryrefslogtreecommitdiff
path: root/spec/ruby/core/io/buffer/transfer_spec.rb
blob: cb8c843ff247508b3dd5121d8496fa586dda6e04 (plain)
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
require_relative '../../../spec_helper'

describe "IO::Buffer#transfer" do
  after :each do
    @buffer&.free
    @buffer = nil
  end

  context "with a buffer created with .new" do
    it "transfers internal memory to a new buffer, nullifying the original" do
      buffer = IO::Buffer.new(4)
      info = buffer.to_s
      @buffer = buffer.transfer
      @buffer.to_s.should == info
      buffer.null?.should be_true
    end

    it "transfers mapped memory to a new buffer, nullifying the original" do
      buffer = IO::Buffer.new(4, IO::Buffer::MAPPED)
      info = buffer.to_s
      @buffer = buffer.transfer
      @buffer.to_s.should == info
      buffer.null?.should be_true
    end
  end

  context "with a file-backed buffer created with .map" do
    it "transfers mapped memory to a new buffer, nullifying the original" do
      File.open(__FILE__, "r") do |file|
        buffer = IO::Buffer.map(file, nil, 0, IO::Buffer::READONLY)
        info = buffer.to_s
        @buffer = buffer.transfer
        @buffer.to_s.should == info
        buffer.null?.should be_true
      end
    end
  end

  context "with a String-backed buffer created with .for" do
    context "without a block" do
      it "transfers memory to a new buffer, nullifying the original" do
        buffer = IO::Buffer.for("test")
        info = buffer.to_s
        @buffer = buffer.transfer
        @buffer.to_s.should == info
        buffer.null?.should be_true
      end
    end

    context "with a block" do
      it "transfers memory to a new buffer, breaking the transaction by nullifying the original" do
        IO::Buffer.for(+"test") do |buffer|
          info = buffer.to_s
          @buffer = buffer.transfer
          @buffer.to_s.should == info
          buffer.null?.should be_true
        end
        @buffer.null?.should be_false
      end
    end
  end

  ruby_version_is "3.3" do
    context "with a String-backed buffer created with .string" do
      it "transfers memory to a new buffer, breaking the transaction by nullifying the original" do
        IO::Buffer.string(4) do |buffer|
          info = buffer.to_s
          @buffer = buffer.transfer
          @buffer.to_s.should == info
          buffer.null?.should be_true
        end
        @buffer.null?.should be_false
      end
    end
  end

  it "allows multiple transfers" do
    buffer_1 = IO::Buffer.new(4)
    buffer_2 = buffer_1.transfer
    @buffer = buffer_2.transfer
    buffer_1.null?.should be_true
    buffer_2.null?.should be_true
    @buffer.null?.should be_false
  end

  it "is disallowed while locked, raising IO::Buffer::LockedError" do
    @buffer = IO::Buffer.new(4)
    @buffer.locked do
      -> { @buffer.transfer }.should raise_error(IO::Buffer::LockedError, "Cannot transfer ownership of locked buffer!")
    end
  end

  context "with a slice of a buffer" do
    it "transfers source to a new slice, not touching the buffer" do
      @buffer = IO::Buffer.new(4)
      slice = @buffer.slice(0, 2)
      @buffer.set_string("test")

      new_slice = slice.transfer
      slice.null?.should be_true
      new_slice.null?.should be_false
      @buffer.null?.should be_false

      new_slice.set_string("ea")
      @buffer.get_string.should == "east"
    end

    it "nullifies buffer, invalidating the slice" do
      buffer = IO::Buffer.new(4)
      slice = buffer.slice(0, 2)
      @buffer = buffer.transfer

      slice.null?.should be_false
      slice.valid?.should be_false
      -> { slice.get_string }.should raise_error(IO::Buffer::InvalidatedError, "Buffer has been invalidated!")
    end
  end
end