def load_user(self, username):
dt = DatabaseTemplate(self.dataSource)
users = dt.query(self.users_by_username_query, (username, ),
self.UsersByUsernameMapping())
if len(users) == 0:
raise UsernameNotFoundException("User not found")
user = users[
0] # First item in list, first column of tuple, containing no GrantedAuthority[]
dbAuths = dt.query(self.auth_by_username_query, (user.username, ),
self.AuthoritiesByUsernameMapping(self.role_prefix))
self.add_custom_authorities(user.username, dbAuths)
if len(dbAuths) == 0:
raise UsernameNotFoundException("User has no GrantedAuthority")
auths = [dbAuth for dbAuth in dbAuths]
return_username = user.username
if not self.username_based_pk:
return_username = username
self.logger.debug("Just fetched %s from the database" % user)
return User(return_username, user.password, user.enabled, True, True,
True, auths)
def testQueryingOracleWithValidlyFormattedArguments(self):
cursor = self.mock()
cursor.expects(once()).method("execute")
cursor.expects(once()).method("fetchall").will(return_value([("workDir", "reloadDir", "archiveDir", "oid1")]))
conn = self.mock()
conn.expects(once()).method("cursor").will(return_value(cursor))
conn.expects(once()).method("close")
sys.modules["cx_Oracle"] = self.mock()
sys.modules["cx_Oracle"].expects(once()).method("connect").will(return_value(conn))
connection_factory = factory.cxoraConnectionFactory(username="******", password="******", hostname="localhost", db="mock")
dt = DatabaseTemplate(connection_factory)
dt.query("""
SELECT
impcarrcfg.paystat_work_dir,
impcarrcfg.paystat_reload_dir,
impcarrcfg.paystat_archive_dir,
impcarrcfg.oid
FROM impcarrcfg, carr, lklabelsys
WHERE (lklabelsys.oid = impcarrcfg.lklabelsys_oid)
and (carr.oid = impcarrcfg.carr_oid )
and (carr.oid = :carr_oid and lklabelsys.oid = :lklabelsys_oid)
""",
{'carr_oid':5, 'lklabelsys_oid':5},
testSupportClasses.ImpFilePropsRowMapper())
del(sys.modules["cx_Oracle"])
def testQueryingOracleWithValidlyFormattedArguments(self):
cursor = self.mock()
cursor.expects(once()).method("execute")
cursor.expects(once()).method("fetchall").will(return_value([("workDir", "reloadDir", "archiveDir", "oid1")]))
conn = self.mock()
conn.expects(once()).method("cursor").will(return_value(cursor))
sys.modules["cx_Oracle"] = self.mock()
sys.modules["cx_Oracle"].expects(once()).method("connect").will(return_value(conn))
connection_factory = factory.cxoraConnectionFactory(username="******", password="******", hostname="localhost", db="mock")
dt = DatabaseTemplate(connection_factory)
dt.query("""
SELECT
impcarrcfg.paystat_work_dir,
impcarrcfg.paystat_reload_dir,
impcarrcfg.paystat_archive_dir,
impcarrcfg.oid
FROM impcarrcfg, carr, lklabelsys
WHERE (lklabelsys.oid = impcarrcfg.lklabelsys_oid)
and (carr.oid = impcarrcfg.carr_oid )
and (carr.oid = :carr_oid and lklabelsys.oid = :lklabelsys_oid)
""",
{'carr_oid':5, 'lklabelsys_oid':5},
testSupportClasses.ImpFilePropsRowMapper())
del(sys.modules["cx_Oracle"])
def load_user(self, username):
dt = DatabaseTemplate(self.dataSource)
users = dt.query(self.users_by_username_query, (username,), self.UsersByUsernameMapping())
if len(users) == 0:
raise UsernameNotFoundException("User not found")
user = users[0] # First item in list, first column of tuple, containing no GrantedAuthority[]
dbAuths = dt.query(
self.auth_by_username_query, (user.username,), self.AuthoritiesByUsernameMapping(self.role_prefix)
)
self.add_custom_authorities(user.username, dbAuths)
if len(dbAuths) == 0:
raise UsernameNotFoundException("User has no GrantedAuthority")
auths = [dbAuth for dbAuth in dbAuths]
return_username = user.username
if not self.username_based_pk:
return_username = username
self.logger.debug("Just fetched %s from the database" % user)
return User(return_username, user.password, user.enabled, True, True, True, auths)
def testIoCGeneralQueryWithDictionaryRowMapper(self):
appContext = ApplicationContext(XMLConfig("support/databaseTestSqliteApplicationContext.xml"))
factory = appContext.get_object("connection_factory")
databaseTemplate = DatabaseTemplate(factory)
databaseTemplate.execute("DROP TABLE IF EXISTS animal")
databaseTemplate.execute("""
CREATE TABLE animal (
id serial PRIMARY KEY,
name VARCHAR(11),
category VARCHAR(20),
population integer
)
""")
factory.commit()
databaseTemplate.execute("DELETE FROM animal")
factory.commit()
self.assertEquals(len(databaseTemplate.query_for_list("SELECT * FROM animal")), 0)
databaseTemplate.execute("INSERT INTO animal (name, category, population) VALUES ('snake', 'reptile', 1)")
databaseTemplate.execute("INSERT INTO animal (name, category, population) VALUES ('racoon', 'mammal', 0)")
databaseTemplate.execute ("INSERT INTO animal (name, category, population) VALUES ('black mamba', 'kill_bill_viper', 1)")
databaseTemplate.execute ("INSERT INTO animal (name, category, population) VALUES ('cottonmouth', 'kill_bill_viper', 1)")
factory.commit()
self.assertEquals(len(databaseTemplate.query_for_list("SELECT * FROM animal")), 4)
results = databaseTemplate.query("select * from animal", rowhandler=DictionaryRowMapper())
def testIoCGeneralQueryWithDictionaryRowMapper(self):
appContext = ApplicationContext(XMLConfig("support/databaseTestSqliteApplicationContext.xml"))
factory = appContext.get_object("connection_factory")
databaseTemplate = DatabaseTemplate(factory)
databaseTemplate.execute("DROP TABLE IF EXISTS animal")
databaseTemplate.execute("""
CREATE TABLE animal (
id serial PRIMARY KEY,
name VARCHAR(11),
category VARCHAR(20),
population integer
)
""")
factory.commit()
databaseTemplate.execute("DELETE FROM animal")
factory.commit()
self.assertEquals(len(databaseTemplate.query_for_list("SELECT * FROM animal")), 0)
databaseTemplate.execute("INSERT INTO animal (name, category, population) VALUES ('snake', 'reptile', 1)")
databaseTemplate.execute("INSERT INTO animal (name, category, population) VALUES ('racoon', 'mammal', 0)")
databaseTemplate.execute ("INSERT INTO animal (name, category, population) VALUES ('black mamba', 'kill_bill_viper', 1)")
databaseTemplate.execute ("INSERT INTO animal (name, category, population) VALUES ('cottonmouth', 'kill_bill_viper', 1)")
factory.commit()
self.assertEquals(len(databaseTemplate.query_for_list("SELECT * FROM animal")), 4)
results = databaseTemplate.query("select * from animal", rowhandler=DictionaryRowMapper())
def testIoCGeneralQuery(self):
appContext = ApplicationContext(XMLConfig("support/databaseTestApplicationContext.xml"))
mockConnectionFactory = appContext.get_object("mockConnectionFactory")
mockConnectionFactory.stubConnection.mockCursor = self.mock
self.mock.expects(once()).method("execute")
self.mock.expects(once()).method("fetchall").will(return_value([("me", "myphone")]))
databaseTemplate = DatabaseTemplate(connection_factory = mockConnectionFactory)
results = databaseTemplate.query("select * from foobar", rowhandler=testSupportClasses.SampleRowMapper())
def testIoCGeneralQuery(self):
appContext = ApplicationContext(XMLConfig("support/databaseTestApplicationContext.xml"))
mockConnectionFactory = appContext.get_object("mockConnectionFactory")
mockConnectionFactory.stubConnection.mockCursor = self.mock
self.mock.expects(once()).method("execute")
self.mock.expects(once()).method("fetchall").will(return_value([("me", "myphone")]))
databaseTemplate = DatabaseTemplate(connection_factory = mockConnectionFactory)
results = databaseTemplate.query("select * from foobar", rowhandler=testSupportClasses.SampleRowMapper())
def testIoCGeneralQuery(self):
appContext = ApplicationContext(XMLConfig("support/databaseTestSQLServerApplicationContext.xml"))
factory = appContext.get_object("connection_factory")
databaseTemplate = DatabaseTemplate(factory)
results = databaseTemplate.query("select * from animal", rowhandler=testSupportClasses.SampleRowMapper())
def testIoCGeneralQueryWithDictionaryRowMapper(self):
appContext = ApplicationContext(XMLConfig("support/databaseTestMySQLApplicationContext.xml"))
factory = appContext.get_object("connection_factory")
databaseTemplate = DatabaseTemplate(factory)
results = databaseTemplate.query("select * from animal", rowhandler=DictionaryRowMapper())
class AbstractDatabaseTemplateTestCase(unittest.TestCase):
def __init__(self, methodName='runTest'):
unittest.TestCase.__init__(self, methodName)
self.factory = None
self.createdTables = False
def setUp(self):
if not self.createdTables:
self.createTables()
self.databaseTemplate = DatabaseTemplate(self.factory)
self.databaseTemplate.execute("DELETE FROM animal")
self.factory.commit()
self.assertEquals(len(self.databaseTemplate.query_for_list("SELECT * FROM animal")), 0)
self.databaseTemplate.execute("INSERT INTO animal (name, category, population) VALUES ('snake', 'reptile', 1)")
self.databaseTemplate.execute("INSERT INTO animal (name, category, population) VALUES ('racoon', 'mammal', 0)")
self.databaseTemplate.execute ("INSERT INTO animal (name, category, population) VALUES ('black mamba', 'kill_bill_viper', 1)")
self.databaseTemplate.execute ("INSERT INTO animal (name, category, population) VALUES ('cottonmouth', 'kill_bill_viper', 1)")
self.factory.commit()
self.assertEquals(len(self.databaseTemplate.query_for_list("SELECT * FROM animal")), 4)
def tearDown(self):
self.factory.rollback()
def testProgrammaticallyInstantiatingAnAbstractDatabaseTemplate(self):
emptyTemplate = DatabaseTemplate()
self.assertRaises(AttributeError, emptyTemplate.query, "sql query shouldn't work", None)
def testProgrammaticHandlingInvalidRowHandler(self):
self.assertRaises(AttributeError, self.databaseTemplate.query, "select * from animal", rowhandler=testSupportClasses.InvalidCallbackHandler())
def testProgrammaticHandlingImproperRowHandler(self):
self.assertRaises(TypeError, self.databaseTemplate.query, "select * from animal", rowhandler=testSupportClasses.ImproperCallbackHandler())
def testProgrammaticHandlingValidDuckTypedRowHandler(self):
results = self.databaseTemplate.query("select * from animal", rowhandler=testSupportClasses.ValidHandler())
def testProgrammaticStaticQuery(self):
self.assertRaises(ArgumentMustBeNamed, self.databaseTemplate.query, "select * from animal", testSupportClasses.AnimalRowMapper())
animals = self.databaseTemplate.query("select name, category from animal", rowhandler=testSupportClasses.AnimalRowMapper())
self.assertEquals(animals[0].name, "snake")
self.assertEquals(animals[0].category, "reptile")
self.assertEquals(animals[1].name, "racoon")
self.assertEquals(animals[1].category, "mammal")
def testProgrammaticStaticQueryWithSimpleRowMapper(self):
animals = self.databaseTemplate.query("select name, category from animal", rowhandler=SimpleRowMapper(testSupportClasses.Animal))
self.assertEquals(animals[0].name, "snake")
self.assertEquals(animals[0].category, "reptile")
self.assertEquals(animals[1].name, "racoon")
self.assertEquals(animals[1].category, "mammal")
def testProgrammaticStaticQueryWithDictionaryRowMapper(self):
animals = self.databaseTemplate.query("select name, category from animal", rowhandler=DictionaryRowMapper())
self.assertEquals(animals[0]["name"], "snake")
self.assertEquals(animals[0]["category"], "reptile")
self.assertEquals(animals[1]["name"], "racoon")
self.assertEquals(animals[1]["category"], "mammal")
def testProgrammaticQueryWithBoundArguments(self):
animals = self.databaseTemplate.query("select name, category from animal where name = %s", ("snake",), testSupportClasses.AnimalRowMapper())
self.assertEquals(animals[0].name, "snake")
self.assertEquals(animals[0].category, "reptile")
animals = self.databaseTemplate.query("select name, category from animal where name = ?", ("snake",), testSupportClasses.AnimalRowMapper())
self.assertEquals(animals[0].name, "snake")
self.assertEquals(animals[0].category, "reptile")
def testProgrammaticQueryWithBoundArgumentsWithSimpleRowMapper(self):
animals = self.databaseTemplate.query("select name, category from animal where name = %s", ("snake",), SimpleRowMapper(testSupportClasses.Animal))
self.assertEquals(animals[0].name, "snake")
self.assertEquals(animals[0].category, "reptile")
animals = self.databaseTemplate.query("select name, category from animal where name = ?", ("snake",), SimpleRowMapper(testSupportClasses.Animal))
self.assertEquals(animals[0].name, "snake")
self.assertEquals(animals[0].category, "reptile")
def testProgrammaticQueryWithBoundArgumentsWithDictionaryRowMapper(self):
animals = self.databaseTemplate.query("select name, category from animal where name = %s", ("snake",), DictionaryRowMapper())
self.assertEquals(animals[0]["name"], "snake")
self.assertEquals(animals[0]["category"], "reptile")
animals = self.databaseTemplate.query("select name, category from animal where name = ?", ("snake",), DictionaryRowMapper())
self.assertEquals(animals[0]["name"], "snake")
self.assertEquals(animals[0]["category"], "reptile")
def testProgrammaticStaticQueryForList(self):
animals = self.databaseTemplate.query_for_list("select name, category from animal")
self.assertEquals(animals[0][0], "snake")
self.assertEquals(animals[0][1], "reptile")
self.assertEquals(animals[1][0], "racoon")
self.assertEquals(animals[1][1], "mammal")
def testProgrammaticQueryForListWithBoundArguments(self):
animals = self.databaseTemplate.query_for_list("select name, category from animal where name = %s", ("snake",))
self.assertEquals(animals[0][0], "snake")
self.assertEquals(animals[0][1], "reptile")
animals = self.databaseTemplate.query_for_list("select name, category from animal where name = ?", ("snake",))
self.assertEquals(animals[0][0], "snake")
self.assertEquals(animals[0][1], "reptile")
def testProgrammaticQueryForListWithBoundArgumentsNotProperlyTuplized(self):
self.assertRaises(InvalidArgumentType, self.databaseTemplate.query_for_list, "select * from animal where name = %s", "snake")
self.assertRaises(InvalidArgumentType, self.databaseTemplate.query_for_list, "select * from animal where name = ?", "snake")
def testProgrammaticStaticQueryForInt(self):
count = self.databaseTemplate.query_for_int("select population from animal where name = 'snake'")
self.assertEquals(count, 1)
def testProgrammaticQueryForIntWithBoundArguments(self):
count = self.databaseTemplate.query_for_int("select population from animal where name = %s", ("snake",))
self.assertEquals(count, 1)
count = self.databaseTemplate.query_for_int("select population from animal where name = ?", ("snake",))
self.assertEquals(count, 1)
def testProgrammaticStaticQueryForLong(self):
count = self.databaseTemplate.query_for_object("select count(*) from animal", required_type=self.factory.count_type())
self.assertEquals(count, 4)
def testProgrammaticQueryForLongWithBoundVariables(self):
count = self.databaseTemplate.query_for_object("select count(*) from animal where name = %s", ("snake",), self.factory.count_type())
self.assertEquals(count, 1)
count = self.databaseTemplate.query_for_object("select count(*) from animal where name = ?", ("snake",), self.factory.count_type())
self.assertEquals(count, 1)
def testProgrammaticStaticQueryForObject(self):
self.assertRaises(ArgumentMustBeNamed, self.databaseTemplate.query_for_object, "select name from animal where category = 'reptile'", types.StringType)
name = self.databaseTemplate.query_for_object("select name from animal where category = 'reptile'", required_type=types.StringType)
self.assertEquals(name, "snake")
def testProgrammaticQueryForObjectWithBoundVariables(self):
name = self.databaseTemplate.query_for_object("select name from animal where category = %s", ("reptile",), types.StringType)
self.assertEquals(name, "snake")
name = self.databaseTemplate.query_for_object("select name from animal where category = ?", ("reptile",), types.StringType)
self.assertEquals(name, "snake")
def testProgrammaticStaticUpdate(self):
rows = self.databaseTemplate.update("UPDATE animal SET name = 'python' WHERE name = 'snake'")
self.assertEquals(rows, 1)
name = self.databaseTemplate.query_for_object("SELECT name FROM animal WHERE category = 'reptile'", required_type=types.StringType)
self.assertEquals(name, "python")
def testProgrammaticUpdateWithBoundVariables(self):
rows = self.databaseTemplate.update("UPDATE animal SET name = ? WHERE category = ?", ("python", "reptile"))
self.assertEquals(rows, 1)
name = self.databaseTemplate.query_for_object("SELECT name FROM animal WHERE category = 'reptile'", required_type=types.StringType)
self.assertEquals(name, "python")
rows = self.databaseTemplate.update("UPDATE animal SET name = ? WHERE category = %s", ("coily", "reptile"))
self.assertEquals(rows, 1)
name = self.databaseTemplate.query_for_object("SELECT name FROM animal WHERE category = 'reptile'", required_type=types.StringType)
self.assertEquals(name, "coily")
def testProgrammaticStaticInsert(self):
self.databaseTemplate.execute("DELETE FROM animal")
rows = self.databaseTemplate.execute("INSERT INTO animal (name, category, population) VALUES ('black mamba', 'kill_bill_viper', 1)")
self.assertEquals(rows, 1)
name = self.databaseTemplate.query_for_object("SELECT name FROM animal WHERE category = 'kill_bill_viper'", required_type=types.StringType)
self.assertEquals(name, "black mamba")
def testProgrammaticInsertWithBoundVariables(self):
self.databaseTemplate.execute("DELETE FROM animal")
rows = self.databaseTemplate.execute("INSERT INTO animal (name, category, population) VALUES (?, ?, ?)", ('black mamba', 'kill_bill_viper', 1))
self.assertEquals(rows, 1)
name = self.databaseTemplate.query_for_object("SELECT name FROM animal WHERE category = 'kill_bill_viper'", required_type=types.StringType)
self.assertEquals(name, "black mamba")
rows = self.databaseTemplate.execute("INSERT INTO animal (name, category, population) VALUES (%s, %s, %s)", ('cottonmouth', 'kill_bill_viper', 1))
self.assertEquals(rows, 1)
name = self.databaseTemplate.query_for_object("select name from animal where name = 'cottonmouth'", required_type=types.StringType)
self.assertEquals(name, "cottonmouth")
class DatabaseTemplateMockTestCase(MockTestCase):
"""Testing the DatabaseTemplate utilizes stubbing and mocking, in order to isolate from different
vendor implementations. This reduces the overhead in making changes to core functionality."""
def setUp(self):
self.mock = self.mock()
connection_factory = testSupportClasses.StubDBFactory()
connection_factory.stubConnection.mockCursor = self.mock
self.databaseTemplate = DatabaseTemplate(connection_factory)
def testProgrammaticallyInstantiatingAnAbstractDatabaseTemplate(self):
emptyTemplate = DatabaseTemplate()
self.assertRaises(AttributeError, emptyTemplate.query, "sql query shouldn't work", None)
def testProgrammaticHandlingInvalidRowHandler(self):
self.mock.expects(once()).method("execute")
self.mock.expects(once()).method("fetchall").will(return_value([("me", "myphone")]))
self.assertRaises(AttributeError, self.databaseTemplate.query, "select * from foobar", rowhandler=testSupportClasses.InvalidCallbackHandler())
def testProgrammaticHandlingImproperRowHandler(self):
self.mock.expects(once()).method("execute")
self.mock.expects(once()).method("fetchall").will(return_value([("me", "myphone")]))
self.assertRaises(TypeError, self.databaseTemplate.query, "select * from foobar", rowhandler=testSupportClasses.ImproperCallbackHandler())
def testProgrammaticHandlingValidDuckTypedRowHandler(self):
self.mock.expects(once()).method("execute")
self.mock.expects(once()).method("fetchall").will(return_value([("me", "myphone")]))
results = self.databaseTemplate.query("select * from foobar", rowhandler=testSupportClasses.ValidHandler())
def testIoCGeneralQuery(self):
appContext = ApplicationContext(XMLConfig("support/databaseTestApplicationContext.xml"))
mockConnectionFactory = appContext.get_object("mockConnectionFactory")
mockConnectionFactory.stubConnection.mockCursor = self.mock
self.mock.expects(once()).method("execute")
self.mock.expects(once()).method("fetchall").will(return_value([("me", "myphone")]))
databaseTemplate = DatabaseTemplate(connection_factory = mockConnectionFactory)
results = databaseTemplate.query("select * from foobar", rowhandler=testSupportClasses.SampleRowMapper())
def testProgrammaticStaticQuery(self):
self.assertRaises(ArgumentMustBeNamed, self.databaseTemplate.query, "select * from animal", testSupportClasses.AnimalRowMapper())
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("fetchall").will(return_value([('snake', 'reptile', 1), ('racoon', 'mammal', 1)])).id("#2").after("#1")
animals = self.databaseTemplate.query("select * from animal", rowhandler=testSupportClasses.AnimalRowMapper())
self.assertEquals(animals[0].name, "snake")
self.assertEquals(animals[0].category, "reptile")
self.assertEquals(animals[1].name, "racoon")
self.assertEquals(animals[1].category, "mammal")
def testProgrammaticQueryWithBoundArguments(self):
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("fetchall").will(return_value([('snake', 'reptile', 1)])).id("#2").after("#1")
self.mock.expects(once()).method("execute").id("#3").after("#2")
self.mock.expects(once()).method("fetchall").will(return_value([('snake', 'reptile', 1)])).id("#4").after("#3")
animals = self.databaseTemplate.query("select * from animal where name = %s", ("snake",), testSupportClasses.AnimalRowMapper())
self.assertEquals(animals[0].name, "snake")
self.assertEquals(animals[0].category, "reptile")
animals = self.databaseTemplate.query("select * from animal where name = ?", ("snake",), testSupportClasses.AnimalRowMapper())
self.assertEquals(animals[0].name, "snake")
self.assertEquals(animals[0].category, "reptile")
def testProgrammaticStaticQueryForList(self):
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("fetchall").will(return_value([('snake', 'reptile', 1), ('racoon', 'mammal', 1)])).id("#2").after("#1")
animals = self.databaseTemplate.query_for_list("select * from animal")
self.assertEquals(animals[0][0], "snake")
self.assertEquals(animals[0][1], "reptile")
self.assertEquals(animals[1][0], "racoon")
self.assertEquals(animals[1][1], "mammal")
def testProgrammaticQueryForListWithBoundArguments(self):
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("fetchall").will(return_value([('snake', 'reptile', 1)])).id("#2").after("#1")
self.mock.expects(once()).method("execute").id("#3").after("#2")
self.mock.expects(once()).method("fetchall").will(return_value([('snake', 'reptile', 1)])).id("#4").after("#3")
animals = self.databaseTemplate.query_for_list("select * from animal where name = %s", ("snake",))
self.assertEquals(animals[0][0], "snake")
self.assertEquals(animals[0][1], "reptile")
animals = self.databaseTemplate.query_for_list("select * from animal where name = ?", ("snake",))
self.assertEquals(animals[0][0], "snake")
self.assertEquals(animals[0][1], "reptile")
def testProgrammaticQueryForListWithBoundArgumentsNotProperlyTuplized(self):
self.assertRaises(InvalidArgumentType, self.databaseTemplate.query_for_list, "select * from animal where name = %s", "snake")
self.assertRaises(InvalidArgumentType, self.databaseTemplate.query_for_list, "select * from animal where name = ?", "snake")
def testProgrammaticStaticQueryForInt(self):
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("fetchall").will(return_value([(1,)])).id("#2").after("#1")
count = self.databaseTemplate.query_for_int("select population from animal where name = 'snake'")
self.assertEquals(count, 1)
def testProgrammaticQueryForIntWithBoundArguments(self):
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("fetchall").will(return_value([(1,)])).id("#2").after("#1")
self.mock.expects(once()).method("execute").id("#3").after("#2")
self.mock.expects(once()).method("fetchall").will(return_value([(1,)])).id("#4").after("#3")
count = self.databaseTemplate.query_for_int("select population from animal where name = %s", ("snake",))
self.assertEquals(count, 1)
count = self.databaseTemplate.query_for_int("select population from animal where name = ?", ("snake",))
self.assertEquals(count, 1)
def testProgrammaticStaticQueryForLong(self):
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("fetchall").will(return_value([(4,)])).id("#2").after("#1")
count = self.databaseTemplate.query_for_object("select count(*) from animal", required_type=types.IntType)
self.assertEquals(count, 4)
def testProgrammaticQueryForLongWithBoundVariables(self):
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("fetchall").will(return_value([(1,)])).id("#2").after("#1")
self.mock.expects(once()).method("execute").id("#3").after("#2")
self.mock.expects(once()).method("fetchall").will(return_value([(1,)])).id("#4").after("#3")
count = self.databaseTemplate.query_for_object("select count(*) from animal where name = %s", ("snake",), types.IntType)
self.assertEquals(count, 1)
count = self.databaseTemplate.query_for_object("select count(*) from animal where name = ?", ("snake",), types.IntType)
self.assertEquals(count, 1)
def testProgrammaticStaticQueryForObject(self):
self.assertRaises(ArgumentMustBeNamed, self.databaseTemplate.query_for_object, "select name from animal where category = 'reptile'", types.StringType)
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("fetchall").will(return_value([("snake",)])).id("#2").after("#1")
name = self.databaseTemplate.query_for_object("select name from animal where category = 'reptile'", required_type=types.StringType)
self.assertEquals(name, "snake")
def testProgrammaticQueryForObjectWithBoundVariables(self):
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("fetchall").will(return_value([("snake",)])).id("#2").after("#1")
self.mock.expects(once()).method("execute").id("#3").after("#2")
self.mock.expects(once()).method("fetchall").will(return_value([("snake",)])).id("#4").after("#3")
name = self.databaseTemplate.query_for_object("select name from animal where category = %s", ("reptile",), types.StringType)
self.assertEquals(name, "snake")
name = self.databaseTemplate.query_for_object("select name from animal where category = ?", ("reptile",), types.StringType)
self.assertEquals(name, "snake")
def testProgrammaticStaticUpdate(self):
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("execute").id("#2").after("#1")
self.mock.expects(once()).method("fetchall").will(return_value([("python",)])).id("#3").after("#2")
self.mock.rowcount = 1
rows = self.databaseTemplate.update("UPDATE animal SET name = 'python' WHERE name = 'snake'")
self.assertEquals(rows, 1)
name = self.databaseTemplate.query_for_object("SELECT name FROM animal WHERE category = 'reptile'", required_type=types.StringType)
self.assertEquals(name, "python")
def testProgrammaticUpdateWithBoundVariables(self):
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("execute").id("#2").after("#1")
self.mock.expects(once()).method("fetchall").will(return_value([("python",)])).id("#3").after("#2")
self.mock.expects(once()).method("execute").id("#4").after("#3")
self.mock.expects(once()).method("execute").id("#5").after("#4")
self.mock.expects(once()).method("fetchall").will(return_value([("coily",)])).id("#6").after("#5")
self.mock.rowcount = 1
rows = self.databaseTemplate.update("UPDATE animal SET name = ? WHERE category = ?", ("python", "reptile"))
self.assertEquals(rows, 1)
name = self.databaseTemplate.query_for_object("SELECT name FROM animal WHERE category = 'reptile'", required_type=types.StringType)
self.assertEquals(name, "python")
rows = self.databaseTemplate.update("UPDATE animal SET name = ? WHERE category = %s", ("coily", "reptile"))
self.assertEquals(rows, 1)
name = self.databaseTemplate.query_for_object("SELECT name FROM animal WHERE category = 'reptile'", required_type=types.StringType)
self.assertEquals(name, "coily")
def testProgrammaticStaticInsert(self):
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("execute").id("#2").after("#1")
self.mock.expects(once()).method("fetchall").will(return_value([("black mamba",)])).id("#3").after("#2")
self.mock.rowcount = 1
rows = self.databaseTemplate.execute ("INSERT INTO animal (name, category, population) VALUES ('black mamba', 'kill_bill_viper', 1)")
self.assertEquals(rows, 1)
name = self.databaseTemplate.query_for_object("SELECT name FROM animal WHERE category = 'kill_bill_viper'", required_type=types.StringType)
self.assertEquals(name, "black mamba")
def testProgrammaticInsertWithBoundVariables(self):
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("execute").id("#2").after("#1")
self.mock.expects(once()).method("fetchall").will(return_value([("black mamba",)])).id("#3").after("#2")
self.mock.expects(once()).method("execute").id("#4").after("#3")
self.mock.expects(once()).method("execute").id("#5").after("#4")
self.mock.expects(once()).method("fetchall").will(return_value([("cottonmouth",)])).id("#6").after("#5")
self.mock.rowcount = 1
rows = self.databaseTemplate.execute ("INSERT INTO animal (name, category, population) VALUES (?, ?, ?)", ('black mamba', 'kill_bill_viper', 1))
self.assertEquals(rows, 1)
name = self.databaseTemplate.query_for_object("SELECT name FROM animal WHERE category = 'kill_bill_viper'", required_type=types.StringType)
self.assertEquals(name, "black mamba")
rows = self.databaseTemplate.execute("INSERT INTO animal (name, category, population) VALUES (%s, %s, %s)", ('cottonmouth', 'kill_bill_viper', 1))
self.assertEquals(rows, 1)
name = self.databaseTemplate.query_for_object("select name from animal where name = 'cottonmouth'", required_type=types.StringType)
self.assertEquals(name, "cottonmouth")
def testIoCGeneralQuery(self):
appContext = ApplicationContext(XMLConfig("support/databaseTestSQLServerApplicationContext.xml"))
factory = appContext.get_object("connection_factory")
databaseTemplate = DatabaseTemplate(factory)
results = databaseTemplate.query("select * from animal", rowhandler=testSupportClasses.SampleRowMapper())
def testIoCGeneralQueryWithDictionaryRowMapper(self):
appContext = ApplicationContext(XMLConfig("support/databaseTestMySQLApplicationContext.xml"))
factory = appContext.get_object("connection_factory")
databaseTemplate = DatabaseTemplate(factory)
results = databaseTemplate.query("select * from animal", rowhandler=DictionaryRowMapper())
class AbstractDatabaseTemplateTestCase(unittest.TestCase):
def __init__(self, methodName='runTest'):
unittest.TestCase.__init__(self, methodName)
self.factory = None
self.createdTables = False
def setUp(self):
if not self.createdTables:
self.createTables()
self.databaseTemplate = DatabaseTemplate(self.factory)
self.databaseTemplate.execute("DELETE FROM animal")
self.factory.commit()
self.assertEquals(len(self.databaseTemplate.query_for_list("SELECT * FROM animal")), 0)
self.databaseTemplate.execute("INSERT INTO animal (name, category, population) VALUES ('snake', 'reptile', 1)")
self.databaseTemplate.execute("INSERT INTO animal (name, category, population) VALUES ('racoon', 'mammal', 0)")
self.databaseTemplate.execute ("INSERT INTO animal (name, category, population) VALUES ('black mamba', 'kill_bill_viper', 1)")
self.databaseTemplate.execute ("INSERT INTO animal (name, category, population) VALUES ('cottonmouth', 'kill_bill_viper', 1)")
self.factory.commit()
self.assertEquals(len(self.databaseTemplate.query_for_list("SELECT * FROM animal")), 4)
def tearDown(self):
self.factory.rollback()
def testProgrammaticallyInstantiatingAnAbstractDatabaseTemplate(self):
emptyTemplate = DatabaseTemplate()
self.assertRaises(AttributeError, emptyTemplate.query, "sql query shouldn't work", None)
def testProgrammaticHandlingInvalidRowHandler(self):
self.assertRaises(AttributeError, self.databaseTemplate.query, "select * from animal", rowhandler=testSupportClasses.InvalidCallbackHandler())
def testProgrammaticHandlingImproperRowHandler(self):
self.assertRaises(TypeError, self.databaseTemplate.query, "select * from animal", rowhandler=testSupportClasses.ImproperCallbackHandler())
def testProgrammaticHandlingValidDuckTypedRowHandler(self):
results = self.databaseTemplate.query("select * from animal", rowhandler=testSupportClasses.ValidHandler())
def testProgrammaticStaticQuery(self):
self.assertRaises(ArgumentMustBeNamed, self.databaseTemplate.query, "select * from animal", testSupportClasses.AnimalRowMapper())
animals = self.databaseTemplate.query("select name, category from animal", rowhandler=testSupportClasses.AnimalRowMapper())
self.assertEquals(animals[0].name, "snake")
self.assertEquals(animals[0].category, "reptile")
self.assertEquals(animals[1].name, "racoon")
self.assertEquals(animals[1].category, "mammal")
def testProgrammaticStaticQueryWithSimpleRowMapper(self):
animals = self.databaseTemplate.query("select name, category from animal", rowhandler=SimpleRowMapper(testSupportClasses.Animal))
self.assertEquals(animals[0].name, "snake")
self.assertEquals(animals[0].category, "reptile")
self.assertEquals(animals[1].name, "racoon")
self.assertEquals(animals[1].category, "mammal")
def testProgrammaticStaticQueryWithDictionaryRowMapper(self):
animals = self.databaseTemplate.query("select name, category from animal", rowhandler=DictionaryRowMapper())
self.assertEquals(animals[0]["name"], "snake")
self.assertEquals(animals[0]["category"], "reptile")
self.assertEquals(animals[1]["name"], "racoon")
self.assertEquals(animals[1]["category"], "mammal")
def testProgrammaticQueryWithBoundArguments(self):
animals = self.databaseTemplate.query("select name, category from animal where name = %s", ("snake",), testSupportClasses.AnimalRowMapper())
self.assertEquals(animals[0].name, "snake")
self.assertEquals(animals[0].category, "reptile")
animals = self.databaseTemplate.query("select name, category from animal where name = ?", ("snake",), testSupportClasses.AnimalRowMapper())
self.assertEquals(animals[0].name, "snake")
self.assertEquals(animals[0].category, "reptile")
def testProgrammaticQueryWithBoundArgumentsWithSimpleRowMapper(self):
animals = self.databaseTemplate.query("select name, category from animal where name = %s", ("snake",), SimpleRowMapper(testSupportClasses.Animal))
self.assertEquals(animals[0].name, "snake")
self.assertEquals(animals[0].category, "reptile")
animals = self.databaseTemplate.query("select name, category from animal where name = ?", ("snake",), SimpleRowMapper(testSupportClasses.Animal))
self.assertEquals(animals[0].name, "snake")
self.assertEquals(animals[0].category, "reptile")
def testProgrammaticQueryWithBoundArgumentsWithDictionaryRowMapper(self):
animals = self.databaseTemplate.query("select name, category from animal where name = %s", ("snake",), DictionaryRowMapper())
self.assertEquals(animals[0]["name"], "snake")
self.assertEquals(animals[0]["category"], "reptile")
animals = self.databaseTemplate.query("select name, category from animal where name = ?", ("snake",), DictionaryRowMapper())
self.assertEquals(animals[0]["name"], "snake")
self.assertEquals(animals[0]["category"], "reptile")
def testProgrammaticStaticQueryForList(self):
animals = self.databaseTemplate.query_for_list("select name, category from animal")
self.assertEquals(animals[0][0], "snake")
self.assertEquals(animals[0][1], "reptile")
self.assertEquals(animals[1][0], "racoon")
self.assertEquals(animals[1][1], "mammal")
def testProgrammaticQueryForListWithBoundArguments(self):
animals = self.databaseTemplate.query_for_list("select name, category from animal where name = %s", ("snake",))
self.assertEquals(animals[0][0], "snake")
self.assertEquals(animals[0][1], "reptile")
animals = self.databaseTemplate.query_for_list("select name, category from animal where name = ?", ("snake",))
self.assertEquals(animals[0][0], "snake")
self.assertEquals(animals[0][1], "reptile")
def testProgrammaticQueryForListWithBoundArgumentsNotProperlyTuplized(self):
self.assertRaises(InvalidArgumentType, self.databaseTemplate.query_for_list, "select * from animal where name = %s", "snake")
self.assertRaises(InvalidArgumentType, self.databaseTemplate.query_for_list, "select * from animal where name = ?", "snake")
def testProgrammaticStaticQueryForInt(self):
count = self.databaseTemplate.query_for_int("select population from animal where name = 'snake'")
self.assertEquals(count, 1)
def testProgrammaticQueryForIntWithBoundArguments(self):
count = self.databaseTemplate.query_for_int("select population from animal where name = %s", ("snake",))
self.assertEquals(count, 1)
count = self.databaseTemplate.query_for_int("select population from animal where name = ?", ("snake",))
self.assertEquals(count, 1)
def testProgrammaticStaticQueryForLong(self):
count = self.databaseTemplate.query_for_object("select count(*) from animal", required_type=self.factory.count_type())
self.assertEquals(count, 4)
def testProgrammaticQueryForLongWithBoundVariables(self):
count = self.databaseTemplate.query_for_object("select count(*) from animal where name = %s", ("snake",), self.factory.count_type())
self.assertEquals(count, 1)
count = self.databaseTemplate.query_for_object("select count(*) from animal where name = ?", ("snake",), self.factory.count_type())
self.assertEquals(count, 1)
def testProgrammaticStaticQueryForObject(self):
self.assertRaises(ArgumentMustBeNamed, self.databaseTemplate.query_for_object, "select name from animal where category = 'reptile'", types.StringType)
name = self.databaseTemplate.query_for_object("select name from animal where category = 'reptile'", required_type=types.StringType)
self.assertEquals(name, "snake")
def testProgrammaticQueryForObjectWithBoundVariables(self):
name = self.databaseTemplate.query_for_object("select name from animal where category = %s", ("reptile",), types.StringType)
self.assertEquals(name, "snake")
name = self.databaseTemplate.query_for_object("select name from animal where category = ?", ("reptile",), types.StringType)
self.assertEquals(name, "snake")
def testProgrammaticStaticUpdate(self):
rows = self.databaseTemplate.update("UPDATE animal SET name = 'python' WHERE name = 'snake'")
self.assertEquals(rows, 1)
name = self.databaseTemplate.query_for_object("SELECT name FROM animal WHERE category = 'reptile'", required_type=types.StringType)
self.assertEquals(name, "python")
def testProgrammaticUpdateWithBoundVariables(self):
rows = self.databaseTemplate.update("UPDATE animal SET name = ? WHERE category = ?", ("python", "reptile"))
self.assertEquals(rows, 1)
name = self.databaseTemplate.query_for_object("SELECT name FROM animal WHERE category = 'reptile'", required_type=types.StringType)
self.assertEquals(name, "python")
rows = self.databaseTemplate.update("UPDATE animal SET name = ? WHERE category = %s", ("coily", "reptile"))
self.assertEquals(rows, 1)
name = self.databaseTemplate.query_for_object("SELECT name FROM animal WHERE category = 'reptile'", required_type=types.StringType)
self.assertEquals(name, "coily")
def testProgrammaticStaticInsert(self):
self.databaseTemplate.execute("DELETE FROM animal")
rows = self.databaseTemplate.execute("INSERT INTO animal (name, category, population) VALUES ('black mamba', 'kill_bill_viper', 1)")
self.assertEquals(rows, 1)
name = self.databaseTemplate.query_for_object("SELECT name FROM animal WHERE category = 'kill_bill_viper'", required_type=types.StringType)
self.assertEquals(name, "black mamba")
def testProgrammaticStaticInsertWithInsertApi(self):
self.databaseTemplate.execute("DELETE FROM animal")
id = self.databaseTemplate.insert_and_return_id("INSERT INTO animal (name, category, population) VALUES ('black mamba', 'kill_bill_viper', 1)")
self.assertEquals(id, 1)
name = self.databaseTemplate.query_for_object("SELECT name FROM animal WHERE category = 'kill_bill_viper'", required_type=types.StringType)
self.assertEquals(name, "black mamba")
def testProgrammaticInsertWithBoundVariables(self):
self.databaseTemplate.execute("DELETE FROM animal")
rows = self.databaseTemplate.execute("INSERT INTO animal (name, category, population) VALUES (?, ?, ?)", ('black mamba', 'kill_bill_viper', 1))
self.assertEquals(rows, 1)
name = self.databaseTemplate.query_for_object("SELECT name FROM animal WHERE category = 'kill_bill_viper'", required_type=types.StringType)
self.assertEquals(name, "black mamba")
rows = self.databaseTemplate.execute("INSERT INTO animal (name, category, population) VALUES (%s, %s, %s)", ('cottonmouth', 'kill_bill_viper', 1))
self.assertEquals(rows, 1)
name = self.databaseTemplate.query_for_object("select name from animal where name = 'cottonmouth'", required_type=types.StringType)
self.assertEquals(name, "cottonmouth")
def testProgrammaticInsertWithBoundVariablesWithInsertApi(self):
self.databaseTemplate.execute("DELETE FROM animal")
id = self.databaseTemplate.insert_and_return_id("INSERT INTO animal (name, category, population) VALUES (?, ?, ?)", ('black mamba', 'kill_bill_viper', 1))
self.assertEquals(id, 1)
name = self.databaseTemplate.query_for_object("SELECT name FROM animal WHERE category = 'kill_bill_viper'", required_type=types.StringType)
self.assertEquals(name, "black mamba")
id = self.databaseTemplate.insert_and_return_id("INSERT INTO animal (name, category, population) VALUES (%s, %s, %s)", ('cottonmouth', 'kill_bill_viper', 1))
self.assertEquals(id, 2)
name = self.databaseTemplate.query_for_object("select name from animal where name = 'cottonmouth'", required_type=types.StringType)
self.assertEquals(name, "cottonmouth")
class DatabaseTemplateMockTestCase(MockTestCase):
"""Testing the DatabaseTemplate utilizes stubbing and mocking, in order to isolate from different
vendor implementations. This reduces the overhead in making changes to core functionality."""
def setUp(self):
self.mock = self.mock()
connection_factory = testSupportClasses.StubDBFactory()
connection_factory.stubConnection.mockCursor = self.mock
self.databaseTemplate = DatabaseTemplate(connection_factory)
def testProgrammaticallyInstantiatingAnAbstractDatabaseTemplate(self):
emptyTemplate = DatabaseTemplate()
self.assertRaises(AttributeError, emptyTemplate.query, "sql query shouldn't work", None)
def testProgrammaticHandlingInvalidRowHandler(self):
self.mock.expects(once()).method("execute")
self.mock.expects(once()).method("fetchall").will(return_value([("me", "myphone")]))
self.assertRaises(AttributeError, self.databaseTemplate.query, "select * from foobar", rowhandler=testSupportClasses.InvalidCallbackHandler())
def testProgrammaticHandlingImproperRowHandler(self):
self.mock.expects(once()).method("execute")
self.mock.expects(once()).method("fetchall").will(return_value([("me", "myphone")]))
self.assertRaises(TypeError, self.databaseTemplate.query, "select * from foobar", rowhandler=testSupportClasses.ImproperCallbackHandler())
def testProgrammaticHandlingValidDuckTypedRowHandler(self):
self.mock.expects(once()).method("execute")
self.mock.expects(once()).method("fetchall").will(return_value([("me", "myphone")]))
results = self.databaseTemplate.query("select * from foobar", rowhandler=testSupportClasses.ValidHandler())
def testIoCGeneralQuery(self):
appContext = ApplicationContext(XMLConfig("support/databaseTestApplicationContext.xml"))
mockConnectionFactory = appContext.get_object("mockConnectionFactory")
mockConnectionFactory.stubConnection.mockCursor = self.mock
self.mock.expects(once()).method("execute")
self.mock.expects(once()).method("fetchall").will(return_value([("me", "myphone")]))
databaseTemplate = DatabaseTemplate(connection_factory = mockConnectionFactory)
results = databaseTemplate.query("select * from foobar", rowhandler=testSupportClasses.SampleRowMapper())
def testProgrammaticStaticQuery(self):
self.assertRaises(ArgumentMustBeNamed, self.databaseTemplate.query, "select * from animal", testSupportClasses.AnimalRowMapper())
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("fetchall").will(return_value([('snake', 'reptile', 1), ('racoon', 'mammal', 1)])).id("#2").after("#1")
animals = self.databaseTemplate.query("select * from animal", rowhandler=testSupportClasses.AnimalRowMapper())
self.assertEquals(animals[0].name, "snake")
self.assertEquals(animals[0].category, "reptile")
self.assertEquals(animals[1].name, "racoon")
self.assertEquals(animals[1].category, "mammal")
def testProgrammaticQueryWithBoundArguments(self):
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("fetchall").will(return_value([('snake', 'reptile', 1)])).id("#2").after("#1")
self.mock.expects(once()).method("execute").id("#3").after("#2")
self.mock.expects(once()).method("fetchall").will(return_value([('snake', 'reptile', 1)])).id("#4").after("#3")
animals = self.databaseTemplate.query("select * from animal where name = %s", ("snake",), testSupportClasses.AnimalRowMapper())
self.assertEquals(animals[0].name, "snake")
self.assertEquals(animals[0].category, "reptile")
animals = self.databaseTemplate.query("select * from animal where name = ?", ("snake",), testSupportClasses.AnimalRowMapper())
self.assertEquals(animals[0].name, "snake")
self.assertEquals(animals[0].category, "reptile")
def testProgrammaticStaticQueryForList(self):
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("fetchall").will(return_value([('snake', 'reptile', 1), ('racoon', 'mammal', 1)])).id("#2").after("#1")
animals = self.databaseTemplate.query_for_list("select * from animal")
self.assertEquals(animals[0][0], "snake")
self.assertEquals(animals[0][1], "reptile")
self.assertEquals(animals[1][0], "racoon")
self.assertEquals(animals[1][1], "mammal")
def testProgrammaticQueryForListWithBoundArguments(self):
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("fetchall").will(return_value([('snake', 'reptile', 1)])).id("#2").after("#1")
self.mock.expects(once()).method("execute").id("#3").after("#2")
self.mock.expects(once()).method("fetchall").will(return_value([('snake', 'reptile', 1)])).id("#4").after("#3")
animals = self.databaseTemplate.query_for_list("select * from animal where name = %s", ("snake",))
self.assertEquals(animals[0][0], "snake")
self.assertEquals(animals[0][1], "reptile")
animals = self.databaseTemplate.query_for_list("select * from animal where name = ?", ("snake",))
self.assertEquals(animals[0][0], "snake")
self.assertEquals(animals[0][1], "reptile")
def testProgrammaticQueryForListWithBoundArgumentsNotProperlyTuplized(self):
self.assertRaises(InvalidArgumentType, self.databaseTemplate.query_for_list, "select * from animal where name = %s", "snake")
self.assertRaises(InvalidArgumentType, self.databaseTemplate.query_for_list, "select * from animal where name = ?", "snake")
def testProgrammaticStaticQueryForInt(self):
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("fetchall").will(return_value([(1,)])).id("#2").after("#1")
count = self.databaseTemplate.query_for_int("select population from animal where name = 'snake'")
self.assertEquals(count, 1)
def testProgrammaticQueryForIntWithBoundArguments(self):
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("fetchall").will(return_value([(1,)])).id("#2").after("#1")
self.mock.expects(once()).method("execute").id("#3").after("#2")
self.mock.expects(once()).method("fetchall").will(return_value([(1,)])).id("#4").after("#3")
count = self.databaseTemplate.query_for_int("select population from animal where name = %s", ("snake",))
self.assertEquals(count, 1)
count = self.databaseTemplate.query_for_int("select population from animal where name = ?", ("snake",))
self.assertEquals(count, 1)
def testProgrammaticStaticQueryForLong(self):
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("fetchall").will(return_value([(4,)])).id("#2").after("#1")
count = self.databaseTemplate.query_for_object("select count(*) from animal", required_type=types.IntType)
self.assertEquals(count, 4)
def testProgrammaticQueryForLongWithBoundVariables(self):
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("fetchall").will(return_value([(1,)])).id("#2").after("#1")
self.mock.expects(once()).method("execute").id("#3").after("#2")
self.mock.expects(once()).method("fetchall").will(return_value([(1,)])).id("#4").after("#3")
count = self.databaseTemplate.query_for_object("select count(*) from animal where name = %s", ("snake",), types.IntType)
self.assertEquals(count, 1)
count = self.databaseTemplate.query_for_object("select count(*) from animal where name = ?", ("snake",), types.IntType)
self.assertEquals(count, 1)
def testProgrammaticStaticQueryForObject(self):
self.assertRaises(ArgumentMustBeNamed, self.databaseTemplate.query_for_object, "select name from animal where category = 'reptile'", types.StringType)
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("fetchall").will(return_value([("snake",)])).id("#2").after("#1")
name = self.databaseTemplate.query_for_object("select name from animal where category = 'reptile'", required_type=types.StringType)
self.assertEquals(name, "snake")
def testProgrammaticQueryForObjectWithBoundVariables(self):
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("fetchall").will(return_value([("snake",)])).id("#2").after("#1")
self.mock.expects(once()).method("execute").id("#3").after("#2")
self.mock.expects(once()).method("fetchall").will(return_value([("snake",)])).id("#4").after("#3")
name = self.databaseTemplate.query_for_object("select name from animal where category = %s", ("reptile",), types.StringType)
self.assertEquals(name, "snake")
name = self.databaseTemplate.query_for_object("select name from animal where category = ?", ("reptile",), types.StringType)
self.assertEquals(name, "snake")
def testProgrammaticStaticUpdate(self):
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("execute").id("#2").after("#1")
self.mock.expects(once()).method("fetchall").will(return_value([("python",)])).id("#3").after("#2")
self.mock.rowcount = 1
rows = self.databaseTemplate.update("UPDATE animal SET name = 'python' WHERE name = 'snake'")
self.assertEquals(rows, 1)
name = self.databaseTemplate.query_for_object("SELECT name FROM animal WHERE category = 'reptile'", required_type=types.StringType)
self.assertEquals(name, "python")
def testProgrammaticUpdateWithBoundVariables(self):
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("execute").id("#2").after("#1")
self.mock.expects(once()).method("fetchall").will(return_value([("python",)])).id("#3").after("#2")
self.mock.expects(once()).method("execute").id("#4").after("#3")
self.mock.expects(once()).method("execute").id("#5").after("#4")
self.mock.expects(once()).method("fetchall").will(return_value([("coily",)])).id("#6").after("#5")
self.mock.rowcount = 1
rows = self.databaseTemplate.update("UPDATE animal SET name = ? WHERE category = ?", ("python", "reptile"))
self.assertEquals(rows, 1)
name = self.databaseTemplate.query_for_object("SELECT name FROM animal WHERE category = 'reptile'", required_type=types.StringType)
self.assertEquals(name, "python")
rows = self.databaseTemplate.update("UPDATE animal SET name = ? WHERE category = %s", ("coily", "reptile"))
self.assertEquals(rows, 1)
name = self.databaseTemplate.query_for_object("SELECT name FROM animal WHERE category = 'reptile'", required_type=types.StringType)
self.assertEquals(name, "coily")
def testProgrammaticStaticInsert(self):
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("execute").id("#2").after("#1")
self.mock.expects(once()).method("fetchall").will(return_value([("black mamba",)])).id("#3").after("#2")
self.mock.rowcount = 1
rows = self.databaseTemplate.execute ("INSERT INTO animal (name, category, population) VALUES ('black mamba', 'kill_bill_viper', 1)")
self.assertEquals(rows, 1)
name = self.databaseTemplate.query_for_object("SELECT name FROM animal WHERE category = 'kill_bill_viper'", required_type=types.StringType)
self.assertEquals(name, "black mamba")
def testProgrammaticStaticInsertWithInsertApi(self):
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("execute").id("#2").after("#1")
self.mock.expects(once()).method("fetchall").will(return_value([("black mamba",)])).id("#3").after("#2")
self.mock.lastrowid = 42
id = self.databaseTemplate.insert_and_return_id("INSERT INTO animal (name, category, population) VALUES ('black mamba', 'kill_bill_viper', 1)")
self.assertEquals(id, 42)
name = self.databaseTemplate.query_for_object("SELECT name FROM animal WHERE category = 'kill_bill_viper'", required_type=types.StringType)
self.assertEquals(name, "black mamba")
def testProgrammaticInsertWithBoundVariables(self):
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("execute").id("#2").after("#1")
self.mock.expects(once()).method("fetchall").will(return_value([("black mamba",)])).id("#3").after("#2")
self.mock.expects(once()).method("execute").id("#4").after("#3")
self.mock.expects(once()).method("execute").id("#5").after("#4")
self.mock.expects(once()).method("fetchall").will(return_value([("cottonmouth",)])).id("#6").after("#5")
self.mock.rowcount = 1
rows = self.databaseTemplate.execute ("INSERT INTO animal (name, category, population) VALUES (?, ?, ?)", ('black mamba', 'kill_bill_viper', 1))
self.assertEquals(rows, 1)
name = self.databaseTemplate.query_for_object("SELECT name FROM animal WHERE category = 'kill_bill_viper'", required_type=types.StringType)
self.assertEquals(name, "black mamba")
rows = self.databaseTemplate.execute("INSERT INTO animal (name, category, population) VALUES (%s, %s, %s)", ('cottonmouth', 'kill_bill_viper', 1))
self.assertEquals(rows, 1)
name = self.databaseTemplate.query_for_object("select name from animal where name = 'cottonmouth'", required_type=types.StringType)
self.assertEquals(name, "cottonmouth")
def testProgrammaticInsertWithBoundVariablesWithInsertApi(self):
self.mock.expects(once()).method("execute").id("#1")
self.mock.expects(once()).method("execute").id("#2").after("#1")
self.mock.expects(once()).method("fetchall").will(return_value([("black mamba",)])).id("#3").after("#2")
self.mock.expects(once()).method("execute").id("#4").after("#3")
self.mock.expects(once()).method("execute").id("#5").after("#4")
self.mock.expects(once()).method("fetchall").will(return_value([("cottonmouth",)])).id("#6").after("#5")
self.mock.lastrowid = 42
id = self.databaseTemplate.insert_and_return_id ("INSERT INTO animal (name, category, population) VALUES (?, ?, ?)", ('black mamba', 'kill_bill_viper', 1))
self.assertEquals(id, 42)
name = self.databaseTemplate.query_for_object("SELECT name FROM animal WHERE category = 'kill_bill_viper'", required_type=types.StringType)
self.assertEquals(name, "black mamba")
id = self.databaseTemplate.insert_and_return_id("INSERT INTO animal (name, category, population) VALUES (%s, %s, %s)", ('cottonmouth', 'kill_bill_viper', 1))
self.assertEquals(id, 42)
name = self.databaseTemplate.query_for_object("select name from animal where name = 'cottonmouth'", required_type=types.StringType)
self.assertEquals(name, "cottonmouth")
