Esempio n. 1
0
    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"])
Esempio n. 2
0
    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)