mirror of
https://github.com/peeringdb/peeringdb.git
synced 2024-05-11 05:55:09 +00:00
ba6f9b6432
* use new peeringdb client (1.0.0) for pdb_load_data sync (#599) * drop django-mobi for lack of py3/dj2 support (#492) remove django-forms-bootstrap for lack of py3/dj2 support (#492) * black formatted * django2.2 and py3 upgrade (#492) * drop ixlans (#21) ui and api changes * drop local_asn (#168) * org search (#193) * phone number validation (#50) * implement help text tooltips (#228) * Mark own ASN as transit-free (#394) * py3 fix for `pdb_migrate_ixlans` command when writing migration report * pdb_migrate_ixlans: properly handle py3 Runtime error if ixlan dict changes during iteration * set rest DEFAULT_SCHEMA_CLASS to coreapi to fix swagger apidocs fix migration 0027 missing from facsimile manifest * fix swagger doc strings * fix tests that were broken from api doc fixes * fix UniqueFieldValidator for netixlan ipaddress validation that broke during django/drf upgrade * fix org merge tool layout issues * travis config * update pipfile and lock * black formatting * update travis dist * beta mode banner (#411) * add beta banner template (#411) * automatically scheduled sync may not always be on, add a flag that lets us reflect that state in the beta banner message clean up beta banner implementation (#411) * add tests for beta banner (#411)
84 lines
2.5 KiB
Python
84 lines
2.5 KiB
Python
import py.test
|
|
import pytest
|
|
import json
|
|
|
|
from rest_framework.test import APIClient
|
|
from django.test import Client, TestCase
|
|
|
|
from peeringdb_server import maintenance, settings
|
|
from peeringdb_server.models import REFTAG_MAP, User
|
|
import peeringdb_server.views as views
|
|
|
|
from .util import ClientCase
|
|
|
|
|
|
class TestMaintenanceMode(ClientCase):
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
super(TestMaintenanceMode, cls).setUpTestData()
|
|
cls.superuser = User.objects.create_user(
|
|
"su", "su@localhost", "su", is_superuser=True
|
|
)
|
|
cls.org = REFTAG_MAP["org"].objects.create(name="Test Org", status="ok")
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def init_lockfile(self, tmpdir):
|
|
settings.MAINTENANCE_MODE_LOCKFILE = str(tmpdir.join("maintenance.lock"))
|
|
|
|
def test_signup(self):
|
|
"""
|
|
user signup should be blocked during maintenance
|
|
"""
|
|
|
|
maintenance.on()
|
|
|
|
client = Client()
|
|
resp = client.post("/register", data={})
|
|
assert resp.status_code == 503
|
|
assert "maintenance mode" in resp.content.decode()
|
|
|
|
maintenance.off()
|
|
|
|
def test_api(self):
|
|
"""
|
|
test that maintenance mode on blocks all write ops to the rest api
|
|
"""
|
|
|
|
# set maintenance on
|
|
maintenance.on()
|
|
|
|
# init api client
|
|
self.client = APIClient()
|
|
self.client.force_authenticate(self.superuser)
|
|
|
|
err_str = "in maintenance mode"
|
|
|
|
# GET requests should work as expected
|
|
r = self.client.get("/api/org/1", format="json")
|
|
content = json.loads(r.content)
|
|
assert r.status_code == 200
|
|
|
|
# POST should be blocked
|
|
r = self.client.post(
|
|
"/api/net", {"org_id": 1, "name": "Test net", "asn": 9000000}, format="json"
|
|
)
|
|
content = json.loads(r.content)
|
|
assert r.status_code == 503
|
|
assert err_str in content["meta"]["error"]
|
|
net = {"id": 1}
|
|
|
|
# PUT should be blocked
|
|
r = self.client.put("/api/net/{}".format(net["id"]), net, format="json")
|
|
content = json.loads(r.content)
|
|
assert r.status_code == 503
|
|
assert err_str in content["meta"]["error"]
|
|
|
|
# DELETE should be blocked
|
|
r = self.client.delete("/api/net/{}".format(net["id"]), {}, format="json")
|
|
content = json.loads(r.content)
|
|
assert r.status_code == 503
|
|
assert err_str in content["meta"]["error"]
|
|
|
|
# set maintenance mode off
|
|
maintenance.off()
|