Ejemplo n.º 1
0
  def test_checkReverseLookup(self, gethostname_mock, gethostbyname_mock, getfqdn_mock):
    gethostname_mock.return_value = "test"
    gethostbyname_mock.side_effect = ["123.123.123.123", "123.123.123.123"]
    getfqdn_mock.return_value = "test.example.com"

    hostInfo = HostInfo()

    self.assertTrue(hostInfo.checkReverseLookup())
    gethostbyname_mock.assert_any_call("test.example.com")
    gethostbyname_mock.assert_any_call("test")
    self.assertEqual(2, gethostbyname_mock.call_count)

    gethostbyname_mock.side_effect = ["123.123.123.123", "231.231.231.231"]

    self.assertFalse(hostInfo.checkReverseLookup())

    gethostbyname_mock.side_effect = ["123.123.123.123", "123.123.123.123"]
    getfqdn_mock.side_effect = socket.error()

    self.assertFalse(hostInfo.checkReverseLookup())
Ejemplo n.º 2
0
    def test_checkReverseLookup(self, gethostname_mock, gethostbyname_mock,
                                getfqdn_mock):
        gethostname_mock.return_value = "test"
        gethostbyname_mock.side_effect = ["123.123.123.123", "123.123.123.123"]
        getfqdn_mock.return_value = "test.example.com"

        hostInfo = HostInfo()

        self.assertTrue(hostInfo.checkReverseLookup())
        gethostbyname_mock.assert_any_call("test.example.com")
        gethostbyname_mock.assert_any_call("test")
        self.assertEqual(2, gethostbyname_mock.call_count)

        gethostbyname_mock.side_effect = ["123.123.123.123", "231.231.231.231"]

        self.assertFalse(hostInfo.checkReverseLookup())

        gethostbyname_mock.side_effect = ["123.123.123.123", "123.123.123.123"]
        getfqdn_mock.side_effect = socket.error()

        self.assertFalse(hostInfo.checkReverseLookup())