def test_update_data(self):
"""Test that update_data calls the expected update method."""
generator = OCPGenerator(self.two_hours_ago, self.now, self.attributes)
test_method1 = Mock(return_value=True)
test_method2 = Mock(return_value=True)
with patch.dict(
generator.ocp_report_generation,
{
"test_report": {
"_generate_hourly_data": test_method1,
"_update_data": test_method2
}
},
):
kwargs = {"report_type": "test_report"}
generator._update_data({}, self.two_hours_ago, self.now, **kwargs)
test_method2.assert_called_with(
{
"interval_start":
self.two_hours_ago.strftime("%Y-%m-%d %H:%M:%S +0000 UTC"),
"interval_end":
self.now.strftime("%Y-%m-%d %H:%M:%S +0000 UTC"),
},
self.two_hours_ago,
self.now,
**kwargs,
)
test_method1.assert_not_called()
def test_update_data(self):
"""Test that update_data calls the expected update method."""
generator = OCPGenerator(self.two_hours_ago, self.now)
test_method1 = Mock(return_value=True)
test_method2 = Mock(return_value=True)
with patch.dict(
generator.ocp_report_generation,
{
"test_report": {
"_generate_hourly_data": test_method1,
"_update_data": test_method2
}
},
):
kwargs = {"report_type": "test_report"}
generator._update_data({}, self.two_hours_ago, self.now, **kwargs)
test_method2.assert_called_with({}, self.two_hours_ago, self.now,
**kwargs)
test_method1.assert_not_called()
