def test_GetProbes(self, mock_get_probe_data): TestProbing.test_GetProbes(self) probes = self.get_probes_object() metadata = Mock() probes.GetProbes(metadata) mock_get_probe_data.assert_called_with(metadata)
def test_ReceiveData(self, mock_ReceiveDataItem, mock_write_data): TestProbing.test_ReceiveData(self) # we use a simple (read: bogus) datalist here to make this # easy to test datalist = ["a", "b", "c"] probes = self.get_probes_object() client = Mock() client.hostname = "foo.example.com" probes.ReceiveData(client, datalist) self.assertItemsEqual(mock_ReceiveDataItem.call_args_list, [call(client, "a"), call(client, "b"), call(client, "c")]) mock_write_data.assert_called_with(client)