Skip to content

Commit

Permalink
[Bug #19259] Data#with should call initialize method
Browse files Browse the repository at this point in the history
  • Loading branch information
nobu committed Dec 26, 2022
1 parent 9f23789 commit 0b289eb
Show file tree
Hide file tree
Showing 2 changed files with 20 additions and 17 deletions.
23 changes: 6 additions & 17 deletions struct.c
Original file line number Diff line number Diff line change
Expand Up @@ -1831,10 +1831,12 @@ rb_data_initialize_m(int argc, const VALUE *argv, VALUE self)
arg.self = self;
arg.unknown_keywords = Qnil;
rb_hash_foreach(argv[0], struct_hash_set_i, (VALUE)&arg);
// Freeze early before potentially raising, so that we don't leave an
// unfrozen copy on the heap, which could get exposed via ObjectSpace.
OBJ_FREEZE_RAW(self);
if (arg.unknown_keywords != Qnil) {
rb_exc_raise(rb_keyword_error_new("unknown", arg.unknown_keywords));
}
OBJ_FREEZE_RAW(self);
return Qnil;
}

Expand Down Expand Up @@ -1886,22 +1888,9 @@ rb_data_with(int argc, const VALUE *argv, VALUE self)
return self;
}

VALUE copy = rb_obj_alloc(rb_obj_class(self));
rb_struct_init_copy(copy, self);

struct struct_hash_set_arg arg;
arg.self = copy;
arg.unknown_keywords = Qnil;
rb_hash_foreach(kwargs, struct_hash_set_i, (VALUE)&arg);
// Freeze early before potentially raising, so that we don't leave an
// unfrozen copy on the heap, which could get exposed via ObjectSpace.
RB_OBJ_FREEZE_RAW(copy);

if (arg.unknown_keywords != Qnil) {
rb_exc_raise(rb_keyword_error_new("unknown", arg.unknown_keywords));
}

return copy;
VALUE h = rb_struct_to_h(self);
rb_hash_update_by(h, kwargs, NULL);
return rb_class_new_instance_kw(1, &h, rb_obj_class(self), TRUE);
}

/*
Expand Down
14 changes: 14 additions & 0 deletions test/ruby/test_data.rb
Original file line number Diff line number Diff line change
Expand Up @@ -215,6 +215,20 @@ def test_with
assert_raise_with_message(ArgumentError, "wrong number of arguments (given 1, expected 0)") do
source.with({ bar: 2 })
end

oddclass = Data.define(:odd) do
def initialize(odd:)
raise ArgumentError, "Not odd" unless odd.odd?
super(odd: odd)
end
end
assert_raise_with_message(ArgumentError, "Not odd") {
oddclass.new(odd: 0)
}
odd = oddclass.new(odd: 1)
assert_raise_with_message(ArgumentError, "Not odd") {
odd.with(odd: 2)
}
end

def test_memberless
Expand Down

0 comments on commit 0b289eb

Please sign in to comment.