def test_updating_works(self):
uri = ("http://localhost", 1, "etag", int(time.time() * 1000),
int(time.time() * 1000), 1)
q = SQLiteMultipleHostUriQueue(":memory:")
q.add_uri(uri)
uri = ("http://localhost", 1, "etag", int(time.time() * 1000),
int(time.time() * 1000), 2)
q.update_uri(uri)
cursor = q._connection.execute("SELECT * FROM queues WHERE queue=1")
uri_res = cursor.fetchone()
(url, queue, etag, mod_date, next_date, prio) = uri
(url_res, queue_res, etag_res, mod_date_res, next_date_res,
prio_res) = uri_res
self.assertEqual(url, url_res)
self.assertEqual(etag, etag_res)
self.assertEqual(mod_date, mod_date_res)
self.assertEqual(prio, prio_res)
self.assertEqual(next_date, next_date_res)
q.close()
def test_adding_works(self):
uri = ("http://localhost", 1, "etag", int(time.time()*1000),
int(time.time() * 1000), 1)
q = SQLiteMultipleHostUriQueue(":memory:")
q.add_uri(uri)
self.assertEqual(1, q.qsize())
cursor = q._connection.execute("SELECT * FROM queues WHERE queue=1")
uri_res = cursor.fetchone()
(url, queue, etag, mod_date, next_date, prio) = uri
(url_res, queue_res, etag_res, mod_date_res, next_date_res, prio_res) = uri_res
self.assertEqual(url, url_res)
self.assertEqual(queue, queue_res)
self.assertEqual(etag, etag_res)
self.assertEqual(mod_date, mod_date_res)
self.assertEqual(prio, prio_res)
self.assertEqual(next_date, next_date_res)
q.close()
