|
@@ 86-103 (lines=18) @@
|
| 83 |
|
_rule = NaxsiRuleSets.query.filter(NaxsiRuleSets.name == random_name).first() |
| 84 |
|
self.assertEqual(_rule, None) |
| 85 |
|
|
| 86 |
|
def test_select(self): |
| 87 |
|
current_sid = NaxsiRules.query.order_by(NaxsiRules.sid.desc()).first() |
| 88 |
|
current_sid = 1337 if current_sid is None else current_sid.sid + 1 |
| 89 |
|
|
| 90 |
|
db.session.add(NaxsiRules(u'POUET', 'str:test', u'BODY', u'$SQL:8', current_sid, u'WEB_APPS', |
| 91 |
|
u'f hqewifueiwf hueiwhf uiewh fiewh fhw', '1', True, 1457101045)) |
| 92 |
|
db.session.commit() |
| 93 |
|
|
| 94 |
|
_ruleset = NaxsiRules.query.first().ruleset |
| 95 |
|
rv = self.app.get('/rulesets/select/%s' % _ruleset) |
| 96 |
|
self.assertEqual(rv.status_code, 200) |
| 97 |
|
self.assertIn(_ruleset, str(rv.data)) |
| 98 |
|
|
| 99 |
|
random_name = ''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(16)) |
| 100 |
|
rv = self.app.get('/rulesets/select/%s' % random_name) |
| 101 |
|
self.assertEqual(rv.status_code, 200) |
| 102 |
|
|
| 103 |
|
db.session.delete(NaxsiRules.query.filter(current_sid == NaxsiRules.sid).first()) |
| 104 |
|
|
|
@@ 71-84 (lines=14) @@
|
| 68 |
|
db.session.delete(_rule) |
| 69 |
|
db.session.commit() |
| 70 |
|
|
| 71 |
|
def test_del(self): |
| 72 |
|
random_name = ''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(16)) |
| 73 |
|
db.session.add(NaxsiRuleSets(random_name, "naxsi-ruleset: %s" % random_name, int(time()))) |
| 74 |
|
db.session.commit() |
| 75 |
|
_rid = NaxsiRuleSets.query.filter(NaxsiRuleSets.name == random_name).first().id |
| 76 |
|
|
| 77 |
|
_nonexistent_rid = _rid + 1 |
| 78 |
|
rv = self.app.post('/rulesets/del/%d' % _nonexistent_rid) |
| 79 |
|
self.assertEqual(rv.status_code, 302) |
| 80 |
|
|
| 81 |
|
rv = self.app.post('/rulesets/del/%d' % _rid) |
| 82 |
|
self.assertEqual(rv.status_code, 302) |
| 83 |
|
_rule = NaxsiRuleSets.query.filter(NaxsiRuleSets.name == random_name).first() |
| 84 |
|
self.assertEqual(_rule, None) |
| 85 |
|
|
| 86 |
|
def test_select(self): |
| 87 |
|
current_sid = NaxsiRules.query.order_by(NaxsiRules.sid.desc()).first() |
nbs-system /
spike
