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