class ThriftTestBuckets(unittest.TestCase):
def setUp(self):
self.db = ThriftDB()
# delete buckets, lame
self.db.delete_bucket('test_bucket')
def tearDown(self):
self.db.delete_bucket('test_bucket')
def test_create_bucket(self):
r = self.db.make_bucket('test_bucket')
self.assertEqual(r.status_code, 201)
def test_already_created_bucket(self):
self.db.make_bucket('test_bucket')
r = self.db.make_bucket('test_bucket')
self.assertEqual(r.status_code, 409)
def test_delete_bucket(self):
self.db.make_bucket('test_bucket')
r = self.db.delete_bucket('test_bucket')
self.assertEqual(r.status_code, 200)
def test_delete_nonexistent_bucket(self):
r = self.db.delete_bucket('test_bucket')
self.assertEqual(r.status_code, 404)
def test_get_bucket(self):
self.db.make_bucket('test_bucket')
r = self.db.get_bucket('test_bucket')
self.assertEqual(r.status_code, 200)
def test_get_nonexistent_bucket(self):
r = self.db.get_bucket('test_bucket')
self.assertEqual(r.status_code, 404)
