def execute(self, sql, args=None):
self.logger.info(self.mogrify(sql, args))
try:
cursor.execute(self, sql, args)
except Exception as exc:
self.logger.error("%s: %s" % (exc.__class__.__name__, exc))
raise
def execute(self, sql, args=None):
self.logger.info(self.mogrify(sql, args))
try:
cursor.execute(self, sql, args)
except Exception as exc:
self.logger.error("%s: %s" % (exc.__class__.__name__, exc))
raise
def test_get_and_set(self):
queue = self.make_one("test")
queue.put({'foo': 'bar'})
job = queue.get()
self.assertEqual(job.data, {'foo': 'bar'})
job.data = {'foo': 'boo'}
with queue._transaction() as cursor:
cursor.execute("SELECT data FROM %s WHERE id = %s",
(queue.table, job.id))
data = cursor.fetchone()[0]
self.assertEqual(job.data, data)
self.assertIn('size=%d' % len(dumps(data)), repr(job))
def test_get_and_set(self):
queue = self.make_one("test")
queue.put({'foo': 'bar'})
task = queue.get()
self.assertEqual(task.data, {'foo': 'bar'})
task.data = {'foo': 'boo'}
with queue._transaction() as cursor:
cursor.execute(
"SELECT data FROM %s WHERE id = %s",
(queue.table, task.id)
)
data = cursor.fetchone()[0]
self.assertEqual(task.data, data)
self.assertIn('size=%d' % len(dumps(data)), repr(task))
def test_context_manager_get_and_set(self):
queue = self.make_one("test")
queue.put({'foo': 'bar'})
with queue:
task = queue.get()
self.assertEqual(task.data, {'foo': 'bar'})
task.data = {'foo': 'boo'}
with queue as cursor:
cursor.execute("SELECT data FROM %s WHERE id = %s",
(queue.table, task.id))
data = cursor.fetchone()[0]
self.assertEqual(task.data, data)
self.assertIn('size=%d' % len(dumps(data)), repr(task))
def test_context_manager_get_and_set(self):
queue = self.make_one("test")
queue.put({'foo': 'bar'})
with queue:
job = queue.get()
self.assertEqual(job.data, {'foo': 'bar'})
job.data = {'foo': 'boo'}
with queue as cursor:
cursor.execute(
"SELECT data FROM %s WHERE id = %s",
(queue.table, job.id)
)
data = cursor.fetchone()[0]
self.assertEqual(job.data, data)
self.assertIn('size=%d' % len(dumps(data)), repr(job))
