def ProcessMessages(self, msgs=None, token=None):
log_urn = aff4.CurrentAuditLog()
self.EnsureLogIsIndexed(log_urn, token=token)
with data_store.DB.GetMutationPool() as pool:
for msg in msgs:
AuditEventCollection.StaticAdd(log_urn,
msg,
mutation_pool=pool)
def testFlowExecution(self):
client_mock = action_mocks.ListDirectoryClientMock()
client_id = self.SetupClient(0)
rollover = aff4.AUDIT_ROLLOVER_TIME.seconds
# Set time to epoch + 20 intervals
with test_lib.FakeTime(20 * rollover):
flow_test_lib.TestFlowHelper(
filesystem.ListDirectory.__name__,
client_mock,
client_id=client_id,
pathspec=rdf_paths.PathSpec(
path=os.path.join(self.base_path,
"test_img.dd/test directory"),
pathtype=rdf_paths.PathSpec.PathType.OS),
token=self.token)
flow_test_lib.TestFlowHelper(
filesystem.ListDirectory.__name__,
client_mock,
client_id=client_id,
pathspec=rdf_paths.PathSpec(
path=os.path.join(self.base_path, "test_img.dd/glob_test"),
pathtype=rdf_paths.PathSpec.PathType.OS),
token=self.token)
parentdir = aff4.FACTORY.Open("aff4:/audit/logs",
aff4.AFF4Volume,
mode="r",
token=self.token)
logs = list(parentdir.ListChildren())
self.assertEqual(len(logs), 1)
log = aff4.CurrentAuditLog()
stored_events = audit.AuditEventCollection(log)
self.assertEqual(len(stored_events), 2)
for event in stored_events:
self.assertEqual(event.action,
rdf_events.AuditEvent.Action.RUN_FLOW)
self.assertEqual(event.flow_name,
filesystem.ListDirectory.__name__)
self.assertEqual(event.user, self.token.username)
# Set time to epoch + 22 intervals
with test_lib.FakeTime(22 * rollover):
flow_test_lib.TestFlowHelper(
filesystem.ListDirectory.__name__,
client_mock,
client_id=client_id,
pathspec=rdf_paths.PathSpec(
path=os.path.join(self.base_path,
"test_img.dd/test directory"),
pathtype=rdf_paths.PathSpec.PathType.OS),
token=self.token)
parentdir = aff4.FACTORY.Open("aff4:/audit/logs",
aff4.AFF4Volume,
mode="r",
token=self.token)
# Now we should have two collections
logs = list(parentdir.ListChildren())
self.assertEqual(len(logs), 2)
# One with two events
stored_events = audit.AuditEventCollection(logs[0])
self.assertEqual(len(stored_events), 2)
# The other with one
stored_events = audit.AuditEventCollection(logs[1])
self.assertEqual(len(stored_events), 1)
