def main(argv=sys.argv):
conf = argv[1]
sqlFile = "%s/.d2c_test/main_test_stub.sqlite" % os.path.expanduser('~')
if os.path.exists(sqlFile):
print "Deleting existing DB"
os.unlink(sqlFile)
mockBoto = mock(boto)
def mock_connect_ec2(*args, **kwargs):
return DummyConn()
mockBoto.connect_ec2 = mock_connect_ec2
mockRemoteFactory = mock(RemoteShellExecutorFactory)
when(mockRemoteFactory).executor(any(), any(),
any(), any(), any()).thenReturn(mock(RemoteShellExecutor))
when(mockRemoteFactory).executor(any(), any(),
any()).thenReturn(mock(RemoteShellExecutor))
mockExecFactory = mock(ShellExecutorFactory)
when(mockExecFactory).executor(any(), any()).thenReturn(mock(ShellExecutor))
dao = DAO(sqlFile, mockBoto, mockRemoteFactory, mockExecFactory)
test_initor.init_db(dao, conf)
mockAMIFactory = mock(AMIToolsFactory)
mockAMITools = mock(AMITools)
when(mockAMIFactory).getAMITools(any()).thenReturn(mockAMITools)
when(mockAMITools).getArch(any()).thenReturn("x86_64")
when(mockAMITools).registerAMI(any(), any(), any()).thenReturn("foobarami")
print dao.getAMIs()
app = Application(dao, mockAMIFactory)
app.MainLoop()
