def test_get_is_connected_badge_class_returns_correct_class(
self, garden_factory, value, klass):
formatter = GardenFormatter(garden_factory.build(is_connected=value))
ret_val = formatter.get_is_connected_badge_class()
assert ret_val == klass
def test_get_is_connected_element_return_value_contains_is_connected_display_and_badge_classes(
self, garden_factory):
formatter = GardenFormatter(garden_factory.build())
ret1 = 'ret1'
ret2 = 'ret2'
formatter.get_is_connected_display = Mock(return_value=ret1)
formatter.get_is_connected_badge_class = Mock(return_value=ret2)
ret_val = formatter.get_is_connected_element()
assert formatter.get_is_connected_display.return_value in ret_val
assert formatter.get_is_connected_badge_class.return_value in ret_val