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()