def test_main(self):
self.mox.StubOutWithMock(create_dev_db.getpass, 'getuser')
self.mox.StubOutWithMock(create_dev_db, 'create_dev_db')
create_dev_db.getpass.getuser().AndReturn('foo')
create_dev_db.create_dev_db('foo')
self.mox.ReplayAll()
create_dev_db.main()
self.mox.VerifyAll()
def test_main(self):
self.mox.StubOutWithMock(create_dev_db.getpass, 'getuser')
self.mox.StubOutWithMock(create_dev_db, 'create_dev_db')
create_dev_db.getpass.getuser().AndReturn('foo')
create_dev_db.create_dev_db('foo')
self.mox.ReplayAll()
create_dev_db.main()
self.mox.VerifyAll()
def test_fail_to_create_db(self):
self.mox.StubOutWithMock(create_dev_db.MySQLdb, 'connect')
create_dev_db.DB_PASSWORD = '******'
db = self.mox.CreateMockAnything()
create_dev_db.MySQLdb.connect(user='******',
passwd='bar').AndReturn(db)
cursor = self.mox.CreateMockAnything()
db.cursor().AndReturn(cursor)
cursor.execute('CREATE DATABASE IF NOT EXISTS quark_dev_foo'
' CHARACTER SET utf8'
' COLLATE utf8_unicode_ci')
cursor.close()
db.commit()
ex_cursor = self.mox.CreateMockAnything()
db.cursor().AndReturn(ex_cursor)
ex_cursor.execute(
('SELECT SCHEMA_NAME FROM INFORMATION_SCHEMA.SCHEMATA'
' WHERE SCHEMA_NAME = %s'), ['quark_dev_foo'])
ex_cursor.fetchone().AndReturn(None)
ex_cursor.close()
db.close()
self.mox.ReplayAll()
self.assertFalse(create_dev_db.create_dev_db('foo'))
self.mox.VerifyAll()
def test_fail_to_create_db(self):
self.mox.StubOutWithMock(create_dev_db.MySQLdb, 'connect')
create_dev_db.DB_PASSWORD = '******'
db = self.mox.CreateMockAnything()
create_dev_db.MySQLdb.connect(
user='******', passwd='bar').AndReturn(db)
cursor = self.mox.CreateMockAnything()
db.cursor().AndReturn(cursor)
cursor.execute('CREATE DATABASE IF NOT EXISTS quark_dev_foo'
' CHARACTER SET utf8'
' COLLATE utf8_unicode_ci')
cursor.close()
db.commit()
ex_cursor = self.mox.CreateMockAnything()
db.cursor().AndReturn(ex_cursor)
ex_cursor.execute(
('SELECT SCHEMA_NAME FROM INFORMATION_SCHEMA.SCHEMATA'
' WHERE SCHEMA_NAME = %s'), ['quark_dev_foo'])
ex_cursor.fetchone().AndReturn(None)
ex_cursor.close()
db.close()
self.mox.ReplayAll()
self.assertFalse(create_dev_db.create_dev_db('foo'))
self.mox.VerifyAll()
def test_fail_bad_db_name(self):
self.assertFalse(create_dev_db.create_dev_db('foo!??!'))
def test_fail_bad_db_name(self):
self.assertFalse(create_dev_db.create_dev_db('foo!??!'))
