def test_get_or_create_db():
server = Server("http://test.test:5984")
httpretty.register_uri(
httpretty.HEAD, "http://test.test:5984/test", status=404, body=""
)
def create_test_db(request, uri, headers):
httpretty.reset()
httpretty.register_uri(
httpretty.HEAD, "http://test.test:5984/test", status=200
)
httpretty.register_uri(
httpretty.PUT, "http://test.test:5984/test", status=500
)
return 201, headers, ""
httpretty.register_uri(
httpretty.PUT, "http://test.test:5984/test", body=create_test_db
)
assert "test" not in server
test_db = server.get_or_create("test")
assert "test" in server
def test_get_or_create_db():
server = Server("http://test.test:5984")
httpretty.register_uri(httpretty.HEAD,
"http://test.test:5984/test",
status=404,
body="")
def create_test_db(request, uri, headers):
httpretty.reset()
httpretty.register_uri(httpretty.HEAD,
"http://test.test:5984/test",
status=200)
httpretty.register_uri(httpretty.PUT,
"http://test.test:5984/test",
status=500)
return 201, headers, ""
httpretty.register_uri(httpretty.PUT,
"http://test.test:5984/test",
body=create_test_db)
assert "test" not in server
test_db = server.get_or_create("test")
assert "test" in server
def init(db_url, api_url):
"""
Initialize the database server. Sets some configuration parameters on the
server, creates the necessary databases for this project, pushes design
documents into those databases, and sets up replication with the cloud
server if one has already been selected.
"""
old_db_url = config["local_server"]["url"]
if old_db_url and old_db_url != db_url:
raise click.ClickException(
"Local database \"{}\" already initialized. Switching local "
"databases is not currently supported".format(old_db_url)
)
db_config = generate_config(api_url)
server = Server(db_url)
# Configure the CouchDB instance itself
config_items = []
for section, values in db_config.items():
for param, value in values.items():
config_items.append((section, param, value))
with click.progressbar(
config_items, label="Applying CouchDB configuration",
length=len(config_items)
) as _config_items:
for section, param, value in _config_items:
url = urljoin(server.resource.url, "_config", section, param)
try:
current_val = server.resource.session.request(
"GET", url
)[2].read().strip()
except ResourceNotFound:
current_val = None
desired_val = '"{}"'.format(value.replace('"', '\\"'))
if current_val != desired_val:
status = server.resource.session.request(
"PUT", url, body=desired_val
)[0]
# Unless there is some delay between requests, CouchDB gets
# sad for some reason
if status != 200:
click.ClickException(
'Failed to set configuration parameter "{}": {}'.format(
param, res.content
)
)
time.sleep(1)
# Create all dbs on the server
with click.progressbar(
all_dbs, label="Creating databases", length=len(all_dbs)
) as _dbs:
for db_name in _dbs:
server.get_or_create(db_name)
# Push design documents
click.echo("Pushing design documents")
design_path = os.path.dirname(_design.__file__)
server.push_design_documents(design_path)
# Set up replication
if config["cloud_server"]["url"]:
click.echo("Setting up replication with cloud server")
utils.replicate_global_dbs(local_url=db_url)
if config["cloud_server"]["farm_name"]:
utils.replicate_per_farm_dbs(local_url=db_url)
config["local_server"]["url"] = db_url