def test_add_del_tenant_queue(self): ret = db_api.get_nwa_tenant_queue(self.ssn, self.tenant1) self.assertFalse(ret) # not found ret = db_api.add_nwa_tenant_queue(self.ssn, self.tenant1) self.assertTrue(ret) # succeed ret = db_api.add_nwa_tenant_queue(self.ssn, self.tenant1) self.assertFalse(ret) # already registered ret = db_api.get_nwa_tenant_queue(self.ssn, self.tenant1) self.assertTrue(ret) # item found ret = db_api.del_nwa_tenant_queue(self.ssn, self.tenant1) self.assertTrue(ret) # delete succeed ret = db_api.del_nwa_tenant_queue(self.ssn, self.tenant1) self.assertFalse(ret) # do nothing ret = db_api.get_nwa_tenant_queue(self.ssn, self.tenant1) self.assertFalse(ret) # not found
def test_get_tenant_queues(self): ret = db_api.get_nwa_tenant_queues(self.ssn) self.assertEqual(0, len(ret)) self.assertTrue(db_api.add_nwa_tenant_queue(self.ssn, self.tenant1)) self.assertTrue(db_api.add_nwa_tenant_queue(self.ssn, self.tenant2)) self.assertTrue(db_api.add_nwa_tenant_queue(self.ssn, self.tenant3)) ret = db_api.get_nwa_tenant_queues(self.ssn) self.assertEqual(3, len(ret)) self.assertTrue(db_api.del_nwa_tenant_queue(self.ssn, self.tenant1)) ret = db_api.get_nwa_tenant_queues(self.ssn) self.assertEqual(2, len(ret)) self.assertTrue(db_api.del_nwa_tenant_queue(self.ssn, self.tenant2)) self.assertTrue(db_api.del_nwa_tenant_queue(self.ssn, self.tenant3)) ret = db_api.get_nwa_tenant_queues(self.ssn) self.assertEqual(0, len(ret))