def test_change_colorbar_config(self):
view_colorbar_range_mock = PropertyMock()
view_colorbar_log_mock = PropertyMock()
type(self.view).colorbar_range = view_colorbar_range_mock
type(self.view).colorbar_log = view_colorbar_log_mock
model_colorbar_range_mock = PropertyMock(return_value=(1, 5))
model_colorbar_log_mock = PropertyMock(return_value=False)
type(self.model).colorbar_range = model_colorbar_range_mock
type(self.model).colorbar_log = model_colorbar_log_mock
# passed model -> view
self.presenter = SlicePlotOptionsPresenter(self.view, self.model)
model_colorbar_range_mock.assert_called_with()
model_colorbar_log_mock.assert_called_with()
view_colorbar_range_mock.assert_called_once_with((1, 5))
view_colorbar_log_mock.assert_called_once_with(False)
# passed view -> model
view_colorbar_range_mock.return_value = (2, 10)
view_colorbar_log_mock.return_value = True
self.presenter._set_c_range()
self.presenter._set_colorbar_log()
self.presenter.get_new_config()
self.model.change_axis_scale.assert_called_once_with((2, 10), True)
def test_data_values_should_use_same_style_for_submission_dates(self):
ws = Mock(spec=Worksheet)
date_1 = ExcelDate(datetime.datetime(2012, 04, 12, 13, 56, 40),
'submission_date')
date_2 = ExcelDate(datetime.datetime(2012, 05, 29, 13, 56, 40),
'submission_date')
row = [date_1, date_2]
width = utils.WIDTH_ONE_CHAR * (len(str(date_1.date)) +
utils.BUFFER_WIDTH)
mock_width_property = PropertyMock(return_value=width)
mock_column = Mock(spec=Column)
ws.col.return_value = mock_column
type(mock_column).width = mock_width_property
utils.write_row_to_worksheet(ws, row, 1)
mock_width_property.assert_called_with(width)
ws.write.assert_any_call(
1,
0,
date_1.date,
style=utils.EXCEL_DATE_STYLE.get('submission_date'))
ws.write.assert_any_call(
1,
1,
date_2.date,
style=utils.EXCEL_DATE_STYLE.get('submission_date'))
ws.col.assert_any_with(0)
ws.col.assert_any_with(1)
def test_data_values_should_use_same_style_for_mm_yyyy_format(self):
ws = Mock(spec=Worksheet)
date_1 = ExcelDate(datetime.datetime(2012, 04, 24), 'mm.yyyy')
date_2 = ExcelDate(datetime.datetime(2012, 05, 24), 'mm.yyyy')
row = [date_1, date_2]
width = workbook_utils.WIDTH_ONE_CHAR * (len(str(date_1.date)) +
workbook_utils.BUFFER_WIDTH)
mock_width = PropertyMock(return_value=width)
mock_column = Mock(spec=Column)
ws.col.return_value = mock_column
type(mock_column).width = mock_width
write_row_to_worksheet(ws, row, 1)
mock_width.assert_called_with(width)
ws.write.assert_any_call(
1,
0,
date_1.date,
style=workbook_utils.EXCEL_DATE_STYLE.get('mm.yyyy'))
ws.write.assert_any_call(
1,
1,
date_2.date,
style=workbook_utils.EXCEL_DATE_STYLE.get('mm.yyyy'))
ws.col.assert_any_with(0)
ws.col.assert_any_with(1)
def test_get_keyring_success(self):
key = "key"
tsigsecret = "secret"
conf_m = self.create_patch("yandex_designate.backend.dynamicdns.cfg.CONF")
read_tsigkey_m = self.create_patch("yandex_designate.backend.dynamicdns.read_tsigkey")
parse_tsigkeyring_m = self.create_patch("dns.tsigkeyring.from_text")
keyfile = Mock()
config = Mock()
keyring_m = Mock()
key_m = PropertyMock(return_value=key)
keyfile_m = PropertyMock(return_value=keyfile)
type(config).dyndns_key = key_m
type(config).dyndns_key_file = keyfile_m
conf_m.__getitem__.return_value = config
read_tsigkey_m.return_value = ("md5", tsigsecret)
parse_tsigkeyring_m.return_value = keyring_m
self.assertEqual(get_keyring(), keyring_m)
key_m.assert_called_with()
keyfile_m.assert_called_with()
read_tsigkey_m.assert_called_with(keyfile, key)
parse_tsigkeyring_m.assert_called_with({key: tsigsecret})
def test_change_xrange(self):
view_x_range_mock = PropertyMock()
type(self.view).x_range = view_x_range_mock
model_x_range_mock = PropertyMock(return_value=(1, 5))
type(self.model).x_range = model_x_range_mock
# passed model -> view
self.presenter1 = SlicePlotOptionsPresenter(self.view, self.model)
model_x_range_mock.assert_called_with()
view_x_range_mock.assert_called_once_with((1, 5))
# passed view -> model through slice presenter
model_x_range_mock.reset_mock()
view_x_range_mock.return_value = (2, 10)
self.presenter1._xy_config_modified('x_range')
self.presenter1.get_new_config()
model_x_range_mock.assert_called_once_with((2, 10))
# passed view -> model through cut presenter
self.presenter2 = CutPlotOptionsPresenter(self.view, self.model)
self.presenter2._xy_config_modified('x_range')
self.presenter2.get_new_config()
self.model.change_axis_scale.assert_called_once()
def test_is_element_displayed_not_displayed_splinter(page, splinter, splinter_strategy):
locator = (splinter_strategy, str(random.random()))
from mock import PropertyMock
visible_mock = PropertyMock(return_value=False)
page.driver.configure_mock(**{'find_by_{0}.return_value.first.visible'.format(splinter_strategy): visible_mock})
type(getattr(page.driver, 'find_by_{0}'.format(splinter_strategy)).return_value.first).visible = visible_mock
assert not page.is_element_displayed(*locator)
getattr(page.driver, 'find_by_{0}'.format(splinter_strategy)).assert_called_once_with(locator[1])
visible_mock.assert_called_with()
def test_get_config_xml(self):
expected_config_xml = "<Sample Config XML/>"
p = PropertyMock(return_value=expected_config_xml)
mock_data_io = MagicMock()
type(mock_data_io).config_xml = p
j = vJob (mock_data_io, None)
self.assertEqual(j.config_xml, expected_config_xml)
p.assert_called_with()
def test_is_supported(self):
input_mock = PropertyMock(return_value="coffee")
class IsSupportedTestCompiler(BaseCompiler):
input_extension = input_mock
compiler = IsSupportedTestCompiler()
self.assertEqual(compiler.is_supported("dummy.coffee"), True)
self.assertEqual(compiler.is_supported("dummy.js"), False)
input_mock.assert_called_with()
def test_is_supported(self):
input_mock = PropertyMock(return_value="coffee")
class IsSupportedTestCompiler(BaseCompiler):
input_extension = input_mock
compiler = IsSupportedTestCompiler()
self.assertEqual(compiler.is_supported("dummy.coffee"), True)
self.assertEqual(compiler.is_supported("dummy.js"), False)
input_mock.assert_called_with()
def test_get_output_filename(self):
input_mock = PropertyMock(return_value="coffee")
output_mock = PropertyMock(return_value="js")
class GetOutputFilenameTestCompiler(BaseCompiler):
input_extension = input_mock
output_extension = output_mock
compiler = GetOutputFilenameTestCompiler()
self.assertEqual(compiler.get_output_filename("dummy.coffee"),
"dummy.js")
self.assertEqual(compiler.get_output_filename("dummy.coffee.coffee"),
"dummy.coffee.js")
input_mock.assert_called_with()
output_mock.assert_called_with()
def test_only_test_class_gets_set_if_not_test_identifier(
self, mock_metric, get_context):
result = Mock()
test_method = PropertyMock()
test_class = PropertyMock()
type(result).test_method = test_method
type(result).test_class = test_class
mock_metric.return_value = result
logger = BoundedMetricsLogger(event=self.event)
logger.context = self.context
logger.publisher = self.publisher
logger.context.identifier = 'BestClass'
logger.add('limitless_metric', 123, unit='skies')
test_class.assert_called_with('BestClass')
test_method.assert_not_called()
def test_test_method_and_test_class_get_set_if_test_method_identifier(
self, mock_metric, get_context):
result = Mock()
test_method = PropertyMock()
test_class = PropertyMock()
type(result).test_method = test_method
type(result).test_class = test_class
test_method = PropertyMock()
type(result).test_method = test_method
mock_metric.return_value = result
logger = BoundedMetricsLogger(event=self.event)
logger.context = self.context
logger.publisher = self.publisher
logger.context.identifier = 'AwesomeClass.incredible_test'
logger.add('limitless_metric', 123, unit='skies')
test_class.assert_called_with('AwesomeClass')
test_method.assert_called_with('incredible_test')
def test_get_output_filename(self):
input_mock = PropertyMock(return_value="coffee")
output_mock = PropertyMock(return_value="js")
class GetOutputFilenameTestCompiler(BaseCompiler):
input_extension = input_mock
output_extension = output_mock
compiler = GetOutputFilenameTestCompiler()
self.assertEqual(
compiler.get_output_filename("dummy.coffee"),
"dummy.js"
)
self.assertEqual(
compiler.get_output_filename("dummy.coffee.coffee"),
"dummy.coffee.js"
)
input_mock.assert_called_with()
output_mock.assert_called_with()
def test_get_keyring_fail_on_parse(self):
key = "key"
tsigsecret = "secret"
conf_m = self.create_patch("yandex_designate.backend.dynamicdns.cfg.CONF")
read_tsigkey_m = self.create_patch("yandex_designate.backend.dynamicdns.read_tsigkey")
parse_tsigkeyring_m = self.create_patch("dns.tsigkeyring.from_text")
keyfile = Mock()
config = Mock()
key_m = PropertyMock(return_value=key)
keyfile_m = PropertyMock(return_value=keyfile)
type(config).dyndns_key = key_m
type(config).dyndns_key_file = keyfile_m
conf_m.__getitem__.return_value = config
read_tsigkey_m.return_value = ("md5", tsigsecret)
parse_tsigkeyring_m.side_effect = binascii.Error
self.assertRaises(TsigKeyError, get_keyring)
key_m.assert_called_with()
keyfile_m.assert_called_with()
read_tsigkey_m.assert_called_with(keyfile, key)
parse_tsigkeyring_m.assert_called_with({key: tsigsecret})
def test_is_element_displayed_splinter(self, page, splinter, splinter_strategy):
locator = (splinter_strategy, str(random.random()))
from mock import PropertyMock
visible_mock = PropertyMock(return_value=True)
page.driver.configure_mock(
**{
"find_by_{0}.return_value.first.visible".format(
splinter_strategy
): visible_mock
}
)
type(
getattr(
page.driver, "find_by_{0}".format(splinter_strategy)
).return_value.first
).visible = visible_mock
assert Region(page).is_element_displayed(*locator)
getattr(
page.driver, "find_by_{0}".format(splinter_strategy)
).assert_called_once_with(locator[1])
visible_mock.assert_called_with()
