Skip to content

Commit 3ba0d96

Browse files
committed
fix UT and schema
1 parent 36c4f3d commit 3ba0d96

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

engine/schema/src/main/resources/META-INF/db/schema-41930to41940.sql

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,8 +20,8 @@
2020
--;
2121

2222
-- Add vm_id column to usage_event table for volume usage events
23-
ALTER TABLE `cloud`.`usage_event` ADD COLUMN `vm_id` bigint unsigned NULL COMMENT 'VM ID associated with volume usage events';
24-
ALTER TABLE `cloud_usage`.`usage_event` ADD COLUMN `vm_id` bigint unsigned NULL COMMENT 'VM ID associated with volume usage events';
23+
CALL `cloud`.`IDEMPOTENT_ADD_COLUMN`('cloud.usage_event','vm_id', 'bigint UNSIGNED NULL COMMENT "VM ID associated with volume usage events"');
24+
CALL `cloud_usage`.`IDEMPOTENT_ADD_COLUMN`('cloud_usage.usage_event','vm_id', 'bigint UNSIGNED NULL COMMENT "VM ID associated with volume usage events"');
2525

2626
-- Add vm_id column to cloud_usage.usage_volume table
27-
ALTER TABLE `cloud_usage`.`usage_volume` ADD COLUMN `vm_id` bigint unsigned NULL COMMENT 'VM ID associated with the volume usage';
27+
CALL `cloud_usage`.`IDEMPOTENT_ADD_COLUMN`('cloud_usage.usage_volume','vm_id', 'bigint UNSIGNED NULL COMMENT "VM ID associated with the volume usage"');

server/src/test/java/com/cloud/vm/UserVmManagerImplTest.java

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -449,6 +449,7 @@ public class UserVmManagerImplTest {
449449
private DiskOfferingVO largerDisdkOffering = prepareDiskOffering(10l * GiB_TO_BYTES, 2l, 10L, 20L);
450450
Class<InvalidParameterValueException> expectedInvalidParameterValueException = InvalidParameterValueException.class;
451451
Class<CloudRuntimeException> expectedCloudRuntimeException = CloudRuntimeException.class;
452+
private MockedStatic<UsageEventUtils> usageEventUtilsMocked;
452453

453454
@Before
454455
public void beforeTest() {
@@ -472,11 +473,13 @@ public void beforeTest() {
472473
lenient().doNothing().when(resourceLimitMgr).decrementResourceCount(anyLong(), any(Resource.ResourceType.class), anyLong());
473474

474475
Mockito.when(virtualMachineProfile.getId()).thenReturn(vmId);
476+
usageEventUtilsMocked = Mockito.mockStatic(UsageEventUtils.class);
475477
}
476478

477479
@After
478480
public void afterTest() {
479481
CallContext.unregister();
482+
usageEventUtilsMocked.close();
480483
}
481484

482485
@Test

0 commit comments

Comments
 (0)