def test_finding_numeric_points(self, uow_mock):
numeric_point_handler = NumericPointHandler()
numeric_point_handler._get_numeric_points()
uow_mock.return_value.component_points.get_points_by_type.assert_called_with("NP")
assert numeric_point_handler.numeric_points == uow_mock.return_value.component_points.get_points_by_type.return_value
def test_run(self, compile_equipment_point_records_mock, evaluate_points_mock, get_numeric_points_mock, uow_mock):
calculated_point_handler = NumericPointHandler()
calculated_point_handler.run()
get_numeric_points_mock.assert_called_with()
evaluate_points_mock.assert_called_with()
compile_equipment_point_records_mock.assert_called_with()
def test_evaluate_points(self, evaluate_point_mock, uow_mock):
numeric_point_handler = NumericPointHandler()
numeric_point_handler.numeric_points = [Mock(), Mock(), Mock()]
numeric_point_handler._evaluate_points()
evaluate_point_mock.assert_has_calls([call(numeric_point_handler.numeric_points[0]),
call(numeric_point_handler.numeric_points[1]),
call(numeric_point_handler.numeric_points[2])])
def test_evaluate_point(self, evaluate_equipment_point_mock, uow_mock):
point_mock = MagicMock()
equipment_points = [Mock(), Mock(), Mock()]
uow_mock.return_value.equipment.get_equipment_points_by_component_point_id.return_value = equipment_points
numeric_point_handler = NumericPointHandler()
numeric_point_handler._evaluate_point(point_mock)
uow_mock.return_value.equipment.get_equipment_points_by_component_point_id.assert_called_with(point_mock["id"])
evaluate_equipment_point_mock.assert_has_calls([call(equipment_points[0], point_mock),
call(equipment_points[1], point_mock),
call(equipment_points[2], point_mock)])