def test_batch_size(self): def assert_batch(instances, batch_size): self.group.instances.count.return_value = instances self.assertEquals(self.group.batch_size, batch_size) with patch_settings('STRETCH_BATCH_SIZE', 5): assert_batch(1, 1) assert_batch(2, 1) assert_batch(3, 2) assert_batch(4, 2) assert_batch(5, 3) assert_batch(20, 5)
def test_node_get_image(): node = Node(name='node') node.system = Mock() node.system.pk = 1 eq_(node.get_image(local=True), 'stretch_agent/sys1/node') location = UrlLocation('public_url', private='private_url') with patch_settings('STRETCH_REGISTRY', location): eq_(node.get_image(local=False, private=True), 'private_url/sys1/node') eq_(node.get_image(local=False, private=False), 'public_url/sys1/node')
def test_batch_size(self): def assert_batch(instances, batch_size): self.group.instances.count.return_value = instances self.assertEquals(self.group.batch_size, batch_size) with patch_settings("STRETCH_BATCH_SIZE", 5): assert_batch(1, 1) assert_batch(2, 1) assert_batch(3, 2) assert_batch(4, 2) assert_batch(5, 3) assert_batch(20, 5)
def setUp(self): patcher = patch('stretch.agent.client.requests') self.requests = patcher.start() self.addCleanup(patcher.stop) patcher = patch_settings('STRETCH_AGENT_CERT', '/cert.pem') patcher.start() self.addCleanup(patcher.stop) self.host = mock_attr(address='127.0.0.1') self.host.environment = mock_attr(name='env', pk=2) self.host.environment.app_paths = {'node': '/path'} self.client = AgentClient(self.host)
def setUp(self): patcher = patch("stretch.agent.client.requests") self.requests = patcher.start() self.addCleanup(patcher.stop) patcher = patch_settings("STRETCH_AGENT_CERT", "/cert.pem") patcher.start() self.addCleanup(patcher.stop) self.host = mock_attr(address="127.0.0.1") self.host.environment = mock_attr(name="env", pk=2) self.host.environment.app_paths = {"node": "/path"} self.client = AgentClient(self.host)