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: "),
]
)
