1 files changed, 110 insertions, 0 deletions
diff --git a/Experior.Activity/tests/test_sbexecutionengine.py b/Experior.Activity/tests/test_sbexecutionengine.py
new file mode 100755
index 0000000..29582c0
--- /dev/null
+++ b/Experior.Activity/tests/test_sbexecutionengine.py
@@ -0,0 +1,110 @@
+#!/usr/bin/env python
+# encoding: utf-8
+"""
+test_sbexecutionengine.py
+
+This file is part of sugarbot.
+
+sugarbot is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+sugarbot is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with sugarbot.  If not, see <http://www.gnu.org/licenses/>.
+"""
+from sbexecutionengine import *
+
+logging.raiseExceptions = 0
+
+class failException(Exception): pass
+class successException(Exception): pass
+
+class test_sbExecutionEngine:
+		
+	class fakeXmlRpc:
+		def __init__(self):
+			self.assertionValue = 1
+			self.restart 	= False
+			self.kill 		= False
+			self.status		= False
+			
+		def success(self, clientID=None):
+			# raise successException
+			# We can't raise exceptions on success, because of the way
+			# sbExecutionEngine works.  Any exceptions call fail()
+			self.status = True
+			
+		def fail(self, reason=None, clientID=None):
+			raise failException	
+			
+		def getRestartFlag(self):
+			return self.restart
+			
+		def getKillFlag(self):
+			return self.kill
+			
+		def getScript(self, dontCare):
+			return \
+"""
+def sugarbot_main(param):
+	assert param
+"""	
+
+	def __init__(self):
+		self.is_setup = False
+		
+	def setUp(self):
+		assert not self.is_setup
+							
+		self.xml 	= self.fakeXmlRpc()
+		self.ee 	= sbExecutionEngine(None,self.xml)
+				
+		self.is_setup 	= True
+		
+	def tearDown(self):
+		assert self.is_setup
+		
+		self.is_setup 	= False
+			
+	def test_setComplete(self):
+		assert not self.ee.isComplete()
+		self.ee.executionComplete = True
+		assert self.ee.isComplete()
+		
+	def test_killSugarbot(self):
+		try:
+			self.ee.killSugarbot()
+			assert 0
+		except SystemExit:
+			pass
+			
+	def test_executePy_Fail(self):
+		try:
+			self.ee.widgets.__nonzero__ = lambda: False
+			self.ee.executePy()
+			assert 0
+		except failException:
+			pass
+			
+	def test_executePy_Success(self):	
+		# try:
+		self.ee.widgets.__nonzero__ = lambda: True
+		self.ee.executePy()
+
+		assert self.xml.status == True
+
+	def test_executePy_UnexpectedException(self):
+		try:
+			def raiseException():	raise ValueError
+			
+			self.ee.widgets.__nonzero__ = raiseException
+			self.ee.executePy()
+			assert 0
+		except:
+			pass
+\ No newline at end of file