From e3628637873f8f63c5f549374ac9308061e6e0a0 Mon Sep 17 00:00:00 2001
From: Dave Peticolas <dave@krondo.com>
Date: Wed, 17 Oct 2001 23:35:52 +0000
Subject: [PATCH] Test adding new transactions & rolling back changed
 transactions.

git-svn-id: svn+ssh://svn.gnucash.org/repo/gnucash/trunk@5679 57a11ea4-9604-0410-9ed3-97b8803252fd
---
 src/backend/postgres/test/test-db.c      | 110 +++++++++++++++++++----
 src/engine/test-core/test-engine-stuff.c |  13 ++-
 2 files changed, 104 insertions(+), 19 deletions(-)

diff --git a/src/backend/postgres/test/test-db.c b/src/backend/postgres/test/test-db.c
index 38a6b7f3d0..a4766bb973 100644
--- a/src/backend/postgres/test/test-db.c
+++ b/src/backend/postgres/test/test-db.c
@@ -201,6 +201,94 @@ test_access (const char *db_name, const char *mode, gboolean multi_user)
   return TRUE;
 }
 
+static gpointer
+mark_account_commodities (Account *a, gpointer data)
+{
+  GHashTable *hash = data;
+
+  g_hash_table_insert (hash, xaccAccountGetCommodity (a), hash);
+
+  return NULL;
+}
+
+static int
+mark_transaction_commodities (Transaction *t, void *data)
+{
+  GHashTable *hash = data;
+
+  g_hash_table_insert (hash, xaccTransGetCurrency (t), hash);
+
+  return TRUE;
+}
+
+static gboolean
+mark_price_commodities (GNCPrice *p, gpointer data)
+{
+  GHashTable *hash = data;
+
+  g_hash_table_insert (hash, gnc_price_get_commodity (p), hash);
+  g_hash_table_insert (hash, gnc_price_get_currency (p), hash);
+
+  return TRUE;
+}
+
+typedef struct
+{
+  GHashTable *hash;
+  GList *to_delete;
+} CommodityDeleteInfo;
+
+static gboolean
+add_commodity_to_delete (gnc_commodity *com, gpointer data)
+{
+  CommodityDeleteInfo *cdi = data;
+
+  if (!g_hash_table_lookup (cdi->hash, com) &&
+      safe_strcmp (gnc_commodity_get_namespace (com),
+                   GNC_COMMODITY_NS_ISO) != 0)
+    cdi->to_delete = g_list_prepend (cdi->to_delete, com);
+
+  return TRUE;
+}
+
+static void
+remove_unneeded_commodities (GNCSession *session)
+{
+  CommodityDeleteInfo cdi;
+  GNCBook *book;
+  GList *node;
+
+  g_return_if_fail (session);
+
+  cdi.hash = g_hash_table_new (g_direct_hash, g_direct_equal);
+
+  book = gnc_session_get_book (session);
+
+  xaccGroupForEachAccount (gnc_book_get_group (book),
+                           mark_account_commodities,
+                           cdi.hash, TRUE);
+
+  xaccGroupForEachTransaction (gnc_book_get_group (book),
+                               mark_transaction_commodities,
+                               cdi.hash);
+
+  gnc_pricedb_foreach_price (gnc_book_get_pricedb (book),
+                             mark_price_commodities,
+                             cdi.hash, FALSE);
+
+  cdi.to_delete = NULL;
+
+  gnc_commodity_table_foreach_commodity (gnc_book_get_commodity_table (book),
+                                         add_commodity_to_delete, &cdi);
+
+  for (node = cdi.to_delete; node; node = node->next)
+    gnc_commodity_table_remove (gnc_book_get_commodity_table (book),
+                                node->data);
+
+  g_list_free (cdi.to_delete);
+  g_hash_table_destroy (cdi.hash);
+}
+
 static gboolean
 test_updates (GNCSession *session, const char *db_name, const char *mode,
               gboolean multi_user)
@@ -223,22 +311,7 @@ test_updates (GNCSession *session, const char *db_name, const char *mode,
                      db_name, mode))
     return FALSE;
 
-  /* make_random_changes_to_session (session); */
-  {
-    Account *account;
-    Account *new_account;
-    AccountGroup *group;
-    GList *list;
-
-    group = gnc_book_get_group (gnc_session_get_book (session));
-    list = xaccGroupGetSubAccounts (group);
-
-    account = g_list_last (list)->data;
-
-    new_account = get_random_account (session);
-
-    xaccAccountInsertSubAccount (account, new_account);
-  }
+  make_random_changes_to_session (session);
 
   if (!multi_user)
   {
@@ -257,6 +330,9 @@ test_updates (GNCSession *session, const char *db_name, const char *mode,
   if (!load_db_file (session_2, db_name, mode))
     return FALSE;
 
+  remove_unneeded_commodities (session);
+  remove_unneeded_commodities (session_2);
+
   ok = gnc_book_equal (gnc_session_get_book (session),
                        gnc_session_get_book (session_2));
 
@@ -297,7 +373,7 @@ test_mode (const char *db_name, const char *mode,
   session = get_random_session ();
 
   add_random_transactions_to_session (session,
-                                      get_random_int_in_range (1, 20));
+                                      get_random_int_in_range (10, 20));
 
   if (!save_db_file (session, db_name, mode))
     return FALSE;
diff --git a/src/engine/test-core/test-engine-stuff.c b/src/engine/test-core/test-engine-stuff.c
index d9ee212078..a415b86fbf 100644
--- a/src/engine/test-core/test-engine-stuff.c
+++ b/src/engine/test-core/test-engine-stuff.c
@@ -499,7 +499,7 @@ change_trans_helper (GNCSession *session, Transaction *trans, GList *accounts)
 
   make_random_changes_to_transaction (session, trans);
 
-  switch (get_random_int_in_range (0, 6))
+  switch (get_random_int_in_range (0, 3))
   {
     case 0: /* delete some splits, add some more */
       do
@@ -545,7 +545,10 @@ change_trans_helper (GNCSession *session, Transaction *trans, GList *accounts)
       make_random_changes_to_split (split);
   }
 
-  xaccTransCommitEdit (trans);
+  if (get_random_boolean ())
+    xaccTransCommitEdit (trans);
+  else
+    xaccTransRollbackEdit (trans);
 }
 
 static gboolean
@@ -589,6 +592,12 @@ make_random_changes_to_group (GNCSession *session, AccountGroup *group)
     xaccAccountInsertSubAccount (account, new_account);
   }
 
+  g_list_free (accounts);
+  accounts = xaccGroupGetSubAccounts (group);
+
+  /* Add some new transactions */
+  add_random_transactions_to_session (session, get_random_int_in_range (1, 6));
+
   /* Mess with the accounts */
   for (node = accounts; node; node = node->next)
   {