Improved test for relation_sizes method

ankane · Feb 20, 2024 · 53ba89d · 53ba89d
1 parent 20d9ffc
commit 53ba89d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/test/space_test.rb b/test/space_test.rb
@@ -6,7 +6,10 @@ def test_database_size
   end
 
   def test_relation_sizes
-    assert database.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" }
   end
 
   def test_table_sizes