memory_view.c: Check availability in rb_memory_view_get

1 files changed, 13 insertions, 0 deletions

diff --git a/test/ruby/test_memory_view.rb b/test/ruby/test_memory_view.rb
index 668d738974..2432f713d1 100644
--- a/test/ruby/test_memory_view.rb
+++ b/test/ruby/test_memory_view.rb
@@ -197,6 +197,13 @@ class TestMemoryView < Test::Unit::TestCase
     assert_equal(expected_result, members)
   end
 
+  def test_rb_memory_view_available_p
+    es = MemoryViewTestUtils::ExportableString.new("ruby")
+    assert_equal(true, MemoryViewTestUtils.available?(es))
+    es = MemoryViewTestUtils::ExportableString.new(nil)
+    assert_equal(false, MemoryViewTestUtils.available?(es))
+  end
+
   def test_ref_count_with_exported_object
     es = MemoryViewTestUtils::ExportableString.new("ruby")
     assert_equal(1, MemoryViewTestUtils.ref_count_while_exporting(es, 1))
@@ -223,6 +230,12 @@ class TestMemoryView < Test::Unit::TestCase
                  memory_view_info)
   end
 
+  def test_rb_memory_view_get_with_memory_view_unavailable_object
+    es = MemoryViewTestUtils::ExportableString.new(nil)
+    memory_view_info = MemoryViewTestUtils.get_memory_view_info(es)
+    assert_nil(memory_view_info)
+  end
+
   def test_rb_memory_view_fill_contiguous_strides
     row_major_strides = MemoryViewTestUtils.fill_contiguous_strides(3, 8, [2, 3, 4], true)
     assert_equal([96, 32, 8],