Improved test for relation_sizes method

ankane · Feb 20, 2024 · c0187ce · c0187ce
1 parent 53ba89d
commit c0187ce
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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