def testDiskVolumeInfoOSXLinux(self): client_mock = action_mocks.UnixVolumeClientMock() session_id = flow_test_lib.TestFlowHelper( compatibility.GetName(filesystem.DiskVolumeInfo), client_mock, client_id=self.client_id, token=self.token, path_list=["/usr/local", "/home"]) results = flow_test_lib.GetFlowResults(self.client_id, session_id) self.assertCountEqual([x.unixvolume.mount_point for x in results], ["/", "/usr"])
def testDiskVolumeInfoOSXLinux(self): client_mock = action_mocks.UnixVolumeClientMock() with test_lib.Instrument(flow.GRRFlow, "SendReply") as send_reply: flow_test_lib.TestFlowHelper(filesystem.DiskVolumeInfo.__name__, client_mock, client_id=self.client_id, token=self.token, path_list=["/usr/local", "/home"]) results = [] for _, reply in send_reply.args: if isinstance(reply, rdf_client.Volume): results.append(reply) self.assertItemsEqual([x.unixvolume.mount_point for x in results], ["/", "/usr"]) self.assertEqual(len(results), 2)