From bda4267fa29bda55c57d0369654c66acda2c08b5 Mon Sep 17 00:00:00 2001
From: Iuri Matias <iuri.matias@gmail.com>
Date: Sat, 13 Jun 2015 09:21:29 -0400
Subject: [PATCH] request needed class

---
 boilerplate/spec/test.js |  4 +++-
 lib/test.js              | 13 ++++++++-----
 2 files changed, 11 insertions(+), 6 deletions(-)

diff --git a/boilerplate/spec/test.js b/boilerplate/spec/test.js
index 80761f81..178bd429 100644
--- a/boilerplate/spec/test.js
+++ b/boilerplate/spec/test.js
@@ -1,4 +1,6 @@
-SimpleStorage = require('embark-framework').Tests();
+var Tests = require('embark-framework').Tests;
+
+SimpleStorage = Tests.request("SimpleStorage");
 
 SimpleStorage.set(100);
 
diff --git a/lib/test.js b/lib/test.js
index 2ee63871..5415c625 100644
--- a/lib/test.js
+++ b/lib/test.js
@@ -29,9 +29,9 @@ TestContractWrapper = (function() {
     py_exec("example_abi")
     py_exec("example_binary = '" + example_binary + "'.decode('hex')")
     py_exec("example_binary")
-    py_exec(className + "_contract = EvmContract(example_abi, example_binary)")
+    py_exec(this.className + "_contract = EvmContract(example_abi, example_binary)")
 
-    this.contractVariable = className + "_contract"
+    this.contractVariable = this.className + "_contract"
   };
 
   TestContractWrapper.prototype.execCmd = function(method, args) {
@@ -58,15 +58,18 @@ TestContract = function(contract, className) {
   return Obj;
 }
 
-SimpleStorage = function() {
+request = function(className) {
   filename = './app/contracts/simple_storage.sol'
   source = fs.readFileSync(filename).toString()
-  className = 'SimpleStorage'
 
   compiled_contracts = web3.eth.compile.solidity(source)
   contract = compiled_contracts[className]
   return TestContract(contract, className)
 }
 
-module.exports = SimpleStorage;
+Test = {
+  request: request
+}
+
+module.exports = Test;