def test_create_delete_methods(self):
mock_resp = mock.Mock()
mock_resp.json = mock.Mock()
mock_resp.json.return_value = {}
mock_resp.text = "mock response"
mock_resp.status_code = 201
mock_del = mock.Mock()
mock_del.status_code = 200
mock_get = mock.Mock()
mock_get.status_code = 404
self.mock_instance.put.return_value = mock_resp
self.mock_instance.delete.return_value = mock_del
self.mock_instance.get.return_value = mock_get
# instantiate and connect
c = Cloudant(self.username, self.password)
c.connect()
self.failUnless(self.mock_session.called)
# create db call
c.create_database("unittest")
self.mock_instance.get.assert_has_calls(
mock.call('https://steve.cloudant.com/unittest')
)
self.mock_instance.put.assert_has_calls(
mock.call('https://steve.cloudant.com/unittest')
)
# delete db call
mock_get.reset_mocks()
mock_get.status_code = 200
c.delete_database("unittest")
self.mock_instance.get.assert_has_calls(
mock.call('https://steve.cloudant.com/unittest')
)
self.mock_instance.delete.assert_has_calls(
mock.call('https://steve.cloudant.com/unittest')
)
# create existing db fails
mock_get.reset_mocks()
mock_get.status_code = 200
self.assertRaises(CloudantException, c.create_database, "unittest")
# delete non-existing db fails
mock_get.reset_mocks()
mock_get.status_code = 404
self.assertRaises(CloudantException, c.delete_database, "unittest")
def setup_db(username, password, url):
dbname = "spark_data"
client = Cloudant(username, password, url=url)
client.connect()
# Perform client tasks...
session = client.session()
print 'Username: {0}'.format(session['userCtx']['name'])
databases = client.all_dbs()
db = client.create_database(dbname)
print 'Databases: {0}'.format(client.all_dbs())
return db
