Added more classes (source for anything 'sourced', etc):

[hub.git] / application / hub / main / producer / cruncher / work_units / class_CruncherTestUnitProducer.php

diff --git a/application/hub/main/producer/cruncher/work_units/class_CruncherTestUnitProducer.php b/application/hub/main/producer/cruncher/work_units/class_CruncherTestUnitProducer.php
index efd3109ce59683d3b60370478bd22c53cdd56cd8..02d7052ef89745cdf27542a49ad5ab375c86033f 100644 (file)
--- a/application/hub/main/producer/cruncher/work_units/class_CruncherTestUnitProducer.php
+++ b/application/hub/main/producer/cruncher/work_units/class_CruncherTestUnitProducer.php
@@ -41,6 +41,12 @@ class CruncherTestUnitProducer extends BaseUnitProducer implements UnitProducer,
                // Get new instance
                $producerInstance = new CruncherTestUnitProducer();
 
+               // Test units have a 'test' source
+               $sourceInstance = UnitSourceFactory::createUnitSourceInstance('test');
+
+               // So set it in the producer
+               $producerInstance->setSourceInstance($sourceInstance);
+
                // Return the prepared instance
                return $producerInstance;
        }
@@ -79,7 +85,7 @@ class CruncherTestUnitProducer extends BaseUnitProducer implements UnitProducer,
                $searchInstance = ObjectFactory::createObjectByConfiguredName('search_criteria_class');
 
                // Add criteria for looking up already created and available test units
-               $searchInstance->addCriteria(CruncherUnitDatabaseWrapper::DB_COLUMN_UNIT_TYPE, BaseUnitProducer::UNIT_TYPE_TEST_UNIT);
+               $searchInstance->addCriteria(CruncherUnitDatabaseWrapper::DB_COLUMN_UNIT_TYPE  , BaseUnitProducer::UNIT_TYPE_TEST_UNIT);
                $searchInstance->addCriteria(CruncherUnitDatabaseWrapper::DB_COLUMN_UNIT_STATUS, BaseUnitProducer::UNIT_STATUS_AVAILABLE);
                $searchInstance->setLimit($this->getConfigInstance()->getConfigEntry('cruncher_test_unit_max_count'));
 
@@ -92,21 +98,8 @@ class CruncherTestUnitProducer extends BaseUnitProducer implements UnitProducer,
                        // @TODO Unfinished work here
                        $this->debugInstance();
                } else {
-                       // No entries found, so generate some. First we need a very secret message (a random string) ...
-                       $rngInstance = ObjectFactory::createObjectByConfiguredName('rng_class');
-                       $secretMessage = base64_encode($rngInstance->randomString($this->getConfigInstance()->getConfigEntry('cruncher_random_secret_message_length')));
-
-                       // Second, we need a secret key
-                       $secretKey = $rngInstance->randomString($this->getConfigInstance()->getConfigEntry('cruncher_random_secret_key_length'));
-
-                       // Now encrypt the message with our key and a good (strong) cipher
-                       $cryptoInstance = ObjectFactory::createObjectByConfiguredName('crypto_class');
-                       $encryptedMessage = base64_encode($cryptoInstance->encryptString($secretMessage, $secretKey));
-
-                       // Make sure the secret key and message are removed from memory
-                       unset($secretKey);
-                       unset($secretMessage);
-                       die($encryptedMessage);
+                       // Get an encrypted, random message from our source
+                       $encryptedMessage = $this->getSourceInstance()->generateMessageFromSource();
                }
        }
 }