import mock
import unittest
-import datetime
from jenkinsapi.node import Node
def testName(self):
with self.assertRaises(AttributeError):
self.n.id()
+ self.assertEquals(self.n.name, 'bobnit')
- def test_online(self):
- return assertEquals(self.n, True)
-
-
+ @mock.patch.object(Node, '_poll')
+ def test_online(self, _poll):
+ _poll.return_value = self.DATA
+ return self.assertEquals(self.n.is_online(), True)
- self.assertEquals(self.n.name, 'bobnit')
+if __name__ == '__main__':
+ unittest.main()