Skip to content

Commit 2c69e21

Browse files
authored
Merge pull request #3167 from mitre/issue-3132
filter 0-score facts from operation source
2 parents 33975f3 + b723aac commit 2c69e21

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

app/objects/c_operation.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -191,6 +191,7 @@ async def all_facts(self):
191191
seeded_facts = []
192192
if self.source:
193193
seeded_facts = await data_svc_handle.get_facts_from_source(self.source.id)
194+
seeded_facts = [f for f in seeded_facts if f.score > 0]
194195
learned_facts = await knowledge_svc_handle.get_facts(criteria=dict(source=self.id))
195196
learned_facts = [f for f in learned_facts if f.score > 0]
196197
return seeded_facts + learned_facts

tests/objects/test_operation.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -190,7 +190,7 @@ def _make_agent(platform='windows', trusted=True, executor_name='psh'):
190190

191191
@pytest.fixture
192192
def op_with_learning_and_seeded(ability, adversary, operation_agent, parse_datestring):
193-
sc = Source(id='3124', name='test', facts=[Fact(trait='domain.user.name', value='bob')])
193+
sc = Source(id='3124', name='test', facts=[Fact(trait='domain.user.name', value='bob'), Fact(trait='domain.user.name', value='jane', score=0)])
194194
op = Operation(id='6789', name='testC', agents=[], adversary=adversary, source=sc, use_learning_parsers=True)
195195
# patch operation to make it 'realistic'
196196
op.start = parse_datestring(OP_START_TIME)

0 commit comments

Comments
 (0)