diff --git a/test/space_test.rb b/test/space_test.rb index 24f53de8..551b4fdf 100644 --- a/test/space_test.rb +++ b/test/space_test.rb @@ -7,9 +7,9 @@ def test_database_size def test_relation_sizes relation_sizes = database.relation_sizes - assert relation_sizes.find { |r| r[:relation] == "users" } - assert relation_sizes.find { |r| r[:relation] == "users_pkey" } - assert relation_sizes.find { |r| r[:relation] == "all_users" } + assert relation_sizes.find { |r| r[:relation] == "users" && r[:type] == "table" } + assert relation_sizes.find { |r| r[:relation] == "users_pkey" && r[:type] == "index" } + assert relation_sizes.find { |r| r[:relation] == "all_users" && r[:type] == "matview" } end def test_table_sizes