def test_get_boot2docker_vm_size_less_8_y(self, fake_virtual_memory, fake_get_raw_input): total_mock = Mock() total_mock.total = 6 * 2 ** 30 fake_virtual_memory.return_value = total_mock fake_get_raw_input.return_value = "y" self.assertEqual(_get_boot2docker_vm_size(), 2048) fake_get_raw_input.assert_has_calls( [ call( "Your system seems to have 6144 megabytes of memory. We would like to allocate 2048 to your vm. Is that ok? (y/n) " ) ] )
def test_get_boot2docker_vm_size_less_8_n(self, fake_virtual_memory, fake_get_raw_input): total_mock = Mock() total_mock.total = 6 * 2 ** 30 fake_virtual_memory.return_value = total_mock fake_get_raw_input.side_effect = ["n", 1] self.assertEqual(_get_boot2docker_vm_size(), 1) fake_get_raw_input.assert_has_calls( [ call( "Your system seems to have 6144 megabytes of memory. We would like to allocate 2048 to your vm. Is that ok? (y/n) " ), call("Please input the number of megabytes to allocate to the vm: "), ] )