00001 <?php
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036
00037 class AccountReadPermissionsOptimizationBaseTest extends ZurmoBaseTest
00038 {
00039 protected function rebuildAndTestThatTheMungeDoesntChange()
00040 {
00041 $beforeRows = ZurmoRedBean::getAll('select munge_id, securableitem_id, count from account_read order by munge_id, securableitem_id, count');
00042 ReadPermissionsOptimizationUtil::rebuild();
00043 $afterRows = ZurmoRedBean::getAll('select munge_id, securableitem_id, count from account_read order by munge_id, securableitem_id, count');
00044
00045 if ($beforeRows != $afterRows)
00046 {
00047 echo "Before and after rebuild munge doesn't match.\n";
00048 echo "--------\n";
00049 foreach ($beforeRows as $row)
00050 {
00051 echo join(', ', array_values($row)) . "\n";
00052 }
00053 echo "--------\n";
00054 foreach ($afterRows as $row)
00055 {
00056 echo join(', ', array_values($row)) . "\n";
00057 }
00058 echo "--------\n";
00059 }
00060
00061 $this->assertEquals(count($beforeRows), count($afterRows));
00062 $this->assertEquals($beforeRows, $afterRows);
00063 }
00064
00065 protected function nukeExistingAccounts()
00066 {
00067 while (true)
00068 {
00069 $accounts = Account::getSubset(null, 0, 50);
00070 if (count($accounts) == 0)
00071 {
00072 break;
00073 }
00074 foreach ($accounts as $account)
00075 {
00076 $account->delete();
00077 unset($account);
00078 }
00079 unset($accounts);
00080 }
00081 $this->assertEquals(0, Account::getCount());
00082 }
00083
00084 protected function createAccounts($count, $testRebuildAfterCreateAccount = false, $firstAccount = 1, $step = 1)
00085 {
00086 $this->assertTrue(is_bool($testRebuildAfterCreateAccount));
00087 $this->assertTrue($firstAccount <= $count);
00088 $this->assertTrue($step < $count);
00089 $betty = User::getByUsername('betty');
00090 $benny = User::getByUsername('benny');
00091 $salesStaff = Group::getByName('Sales Staff');
00092 $countThatBennyCanRead = 0;
00093 $accountIdsThatBennyCanRead = array();
00094 $startTime = microtime(true);
00095 for ($i = 0; $i < $count; $i++)
00096 {
00097 $bennyCanRead = false;
00098 $account = self::createRandomAccount($i);
00099 $securableItemId = $i + 1;
00100 if ($i % 10 == 0)
00101 {
00102 $account->owner = $betty;
00103 $bennyCanRead = true;
00104 }
00105 if ($i % 8 == 0)
00106 {
00107 $account->addPermissions($benny, Permission::READ);
00108 $bennyCanRead = true;
00109 }
00110 if ($i % 12 == 0)
00111 {
00112 $account->addPermissions($salesStaff, Permission::READ);
00113 $bennyCanRead = true;
00114 }
00115
00116 $this->assertTrue($account->save());
00117 if ($bennyCanRead)
00118 {
00119 $countThatBennyCanRead++;
00120 if ($countThatBennyCanRead <= 10)
00121 {
00122 $accountIdsThatBennyCanRead[] = $account->id;
00123 }
00124 }
00125
00126 if ($i >= $firstAccount - 1 &&
00127 ($i - ($firstAccount - 1)) % $step == 0 &&
00128 $testRebuildAfterCreateAccount)
00129 {
00130 $startTime = microtime(true);
00131 ReadPermissionsOptimizationUtil::rebuild(true, true);
00132 $endTime = microtime(true);
00133 if ($this->isDebug())
00134 {
00135 echo 'Rebuilt the munge in php in ' . round($endTime - $startTime, 1) . ' seconds, ' . self::getAccountMungeRowCount() . " rows.\n";
00136 }
00137 $phpRows = self::getAccountMungeRows();
00138
00139
00140 $startTime = microtime(true);
00141 ReadPermissionsOptimizationUtil::rebuild();
00142 $endTime = microtime(true);
00143 if ($this->isDebug())
00144 {
00145 echo 'Rebuilt the munge ' . (SECURITY_OPTIMIZED ? 'optimized' : 'in php') . ' in ' . round($endTime - $startTime, 1) . ' seconds, ' . self::getAccountMungeRowCount() . " rows.\n";
00146 }
00147 $otherRows = self::getAccountMungeRows();
00148 if ($phpRows != $otherRows)
00149 {
00150 echo 'PHP & optimized munges don\'t match after account ' . ($i + 1) . "\n";
00151 echo "--------\n";
00152 foreach ($phpRows as $row)
00153 {
00154 echo join(', ', array_values($row)) . "\n";
00155 }
00156 echo "--------\n";
00157 foreach ($otherRows as $row)
00158 {
00159 echo join(', ', array_values($row)) . "\n";
00160 }
00161 echo "--------\n";
00162 }
00163 $this->assertEquals(count($phpRows), count($otherRows));
00164 $this->assertEquals($phpRows, $otherRows);
00165 }
00166 }
00167 $endTime = microtime(true);
00168
00169 return array($endTime - $startTime, $countThatBennyCanRead, $accountIdsThatBennyCanRead);
00170 }
00171
00172 protected static function createRandomAccount($i)
00173 {
00174 $account = new Account();
00175 $account->name = "Account#$i";
00176 $account->officePhone = rand(10000000, 90000000);
00177 $account->officeFax = rand(10000000, 90000000);
00178 $account->employees = rand(1, 100);
00179 $account->website = "http://www.account$i.com";
00180 $account->annualRevenue = rand(10000, 10000000);
00181 $account->description = "An account for some company called Account#$i.";
00182 $account->primaryEmail->emailAddress = "info@account$i.com";
00183 $account->primaryEmail->optOut = false;
00184 $account->primaryEmail->isInvalid = false;
00185 $account->billingAddress->street1 = "$i Some St";
00186 $account->billingAddress->city = 'City';
00187 $account->billingAddress->state = 'State';
00188 $account->billingAddress->postalCode = rand(1000, 9999);
00189 return $account;
00190 }
00191
00192 protected static function accountMungeDoesntChangeWhenRebuilt()
00193 {
00194
00195
00196 RedBeanModelsCache::forgetAll();
00197 RedBeansCache::forgetAll();
00198 $beforeRows = self::getAccountMungeRows();
00199 ReadPermissionsOptimizationUtil::rebuild();
00200 $afterRows = self::getAccountMungeRows();
00201
00202 if ($beforeRows != $afterRows)
00203 {
00204 echo "Before and after rebuild munge doesn't match.\n";
00205 self::echoRows($beforeRows);
00206 self::echoRows($afterRows);
00207 }
00208
00209 return $beforeRows == $afterRows;
00210 }
00211
00212 public static function getAccountMungeRows(SecurableItem $securableItem = null)
00213 {
00214 if ($securableItem === null)
00215 {
00216 $rows = ZurmoRedBean::getAll('select name, munge_id, count
00217 from account_read, ownedsecurableitem, account
00218 where account_read.securableitem_id = ownedsecurableitem.securableitem_id and
00219 ownedsecurableitem.id = account.ownedsecurableitem_id
00220 order by name, munge_id, account_read.securableitem_id, count');
00221 }
00222 else
00223 {
00224 $securableItemId = $securableItem->getClassId('SecurableItem');
00225 $rows = ZurmoRedBean::getAll("select munge_id, count
00226 from account_read
00227 where securableitem_id = $securableItemId
00228 order by munge_id, count");
00229 }
00230 $rowsWithValues = array();
00231 foreach ($rows as $row)
00232 {
00233 $row = array_values($row);
00234 array_walk($row, array('self', 'stripFullStops'));
00235 $rowsWithValues[] = $row;
00236 }
00237 return $rowsWithValues;
00238 }
00239
00240 protected static function stripFullStops(&$value, $index)
00241 {
00242
00243
00244
00245
00246
00247 $value = str_replace('.', '', $value);
00248 }
00249
00250 protected static function echoAccountMungeRows(SecurableItem $securableItem = null)
00251 {
00252 self::echoRows(self::getAccountMungeRows($securableItem));
00253 }
00254
00255 protected static function echoRows(array $rows)
00256 {
00257 echo "--------\n";
00258 foreach ($rows as $row)
00259 {
00260 echo join(', ', array_values($row)) . "\n";
00261 }
00262 echo "--------\n";
00263 }
00264
00265 protected static function getAccountMungeRowCount()
00266 {
00267 return intval(ZurmoRedBean::getCell('select count(*) from account_read'));
00268 }
00269 }
00270 ?>
