def get_vocab(self, vocab):
"""Return defined vocabularies."""
try:
v = Setting.get_or_create(name='vocabs')
return v.get_vocab(vocab)
except RuntimeException as exception:
return exception, 400
def get_killchain(self, killchain):
"""Return defined killchains."""
try:
v = Setting.get_or_create(name='killchains')
return v.get_killchain(killchain)
except RuntimeException as exception:
return exception, 400
def test_create_vocab_setting():
"""Tests that Vocab settings can be created."""
vocab = Setting.get_or_create(name='my_vocab', type='vocab')
vocab_ = Vocabs.get(vocab.id)
assert vocab_ is not None
assert vocab.id == vocab_.id
assert vocab_.name == 'my_vocab'
def test_create_killchain_setting():
"""Tests that killchain settings can be fetched"""
kc = Setting.get_or_create(name='my_killchain', type='killchain')
assert isinstance(kc, KillChains)
kc_ = KillChains.get(kc.id)
assert kc_ is not None
assert kc.id == kc_.id
assert kc_.name == 'my_killchain'
def set_values_for_vocab(self, vocab):
"""Set a vocabulary."""
values = request.json['values']
try:
v = Setting.get_or_create(name=vocab, type='vocab')
v.set_vocab(values)
return v.get_vocab()
except RuntimeException as exception:
return exception, 400
def set_killchain_pahses(self, killchain):
"""Set phases for a killchain."""
phases = request.json['phases']
try:
kc = Setting.get_or_create(name=killchain)
kc.set_phases(phases)
return kc.get_killchain()
except RuntimeException as exception:
return exception, 400
def add_phase_to_killchain(self, killchain):
"""Set phases for a killchain."""
phase = request.json['phase']
try:
kc = Setting.get_or_create(name=killchain)
kc.add_phase_to_killchain(phase)
return kc.get_killchain()
except RuntimeException as exception:
return exception, 400