X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/b3f9cef0402c67cf06f825b283d95ef5819ab93f..debf08b0415cf0f9d35338cfb280bde8628619eb:/services/nodemanager/tests/test_nodelist.py?ds=sidebyside diff --git a/services/nodemanager/tests/test_nodelist.py b/services/nodemanager/tests/test_nodelist.py deleted file mode 100644 index 5346e7ab7b..0000000000 --- a/services/nodemanager/tests/test_nodelist.py +++ /dev/null @@ -1,57 +0,0 @@ -#!/usr/bin/env python - -from __future__ import absolute_import, print_function - -import unittest - -import arvnodeman.nodelist as nodelist -from . import testutil - -class ArvadosNodeListMonitorActorTestCase(testutil.RemotePollLoopActorTestMixin, - unittest.TestCase): - TEST_CLASS = nodelist.ArvadosNodeListMonitorActor - - def build_monitor(self, side_effect, *args, **kwargs): - super(ArvadosNodeListMonitorActorTestCase, self).build_monitor( - *args, **kwargs) - self.client.nodes().list().execute.side_effect = side_effect - - def test_uuid_is_subscription_key(self): - node = testutil.arvados_node_mock() - self.build_monitor([{'items': [node]}]) - self.monitor.subscribe_to(node['uuid'], - self.subscriber).get(self.TIMEOUT) - self.stop_proxy(self.monitor) - self.subscriber.assert_called_with(node) - - -class CloudNodeListMonitorActorTestCase(testutil.RemotePollLoopActorTestMixin, - unittest.TestCase): - TEST_CLASS = nodelist.CloudNodeListMonitorActor - - class MockNode(object): - def __init__(self, count): - self.id = str(count) - self.name = 'test{}.example.com'.format(count) - self.private_ips = ['10.0.0.{}'.format(count)] - self.public_ips = [] - self.size = None - self.state = 0 - - - def build_monitor(self, side_effect, *args, **kwargs): - super(CloudNodeListMonitorActorTestCase, self).build_monitor( - *args, **kwargs) - self.client.list_nodes.side_effect = side_effect - - def test_id_is_subscription_key(self): - node = self.MockNode(1) - self.build_monitor([[node]]) - self.monitor.subscribe_to('1', self.subscriber).get(self.TIMEOUT) - self.stop_proxy(self.monitor) - self.subscriber.assert_called_with(node) - - -if __name__ == '__main__': - unittest.main() -