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))
def dictfetchone(self):
row = _2cursor.fetchone(self)
if row:
return self.__build_dict(row)
else:
return row
def dictfetchone(self):
row = _2cursor.fetchone(self)
if row:
return self.__build_dict(row)
else:
return row