Github user jburwell commented on a diff in the pull request:
https://github.com/apache/cloudstack/pull/1624#discussion_r74313885
--- Diff: server/test/com/cloud/user/AccountManagerImplTest.java ---
@@ -231,6 +253,73 @@ public void cleanup() {
CallContext.unregister();
}
+ public UsageEventUtils setupUsageUtils() {
+ _usageEventDao = new MockUsageEventDao();
+ UsageEventUtils utils = new UsageEventUtils();
+
+ Map<String, String> usageUtilsFields = new HashMap<String,
String>();
+ usageUtilsFields.put("usageEventDao", "_usageEventDao");
+ usageUtilsFields.put("accountDao", "_accountDao");
+ usageUtilsFields.put("dcDao", "_dcDao");
+ usageUtilsFields.put("configDao", "_configDao");
+
+ for (String fieldName : usageUtilsFields.keySet()) {
+ try {
+ Field f =
UsageEventUtils.class.getDeclaredField(fieldName);
+ f.setAccessible(true);
+ //Remember the old fields for cleanup later (see
cleanupUsageUtils)
+ Field staticField =
UsageEventUtils.class.getDeclaredField("s_" + fieldName);
+ staticField.setAccessible(true);
+ oldFields.put(f.getName(), staticField.get(null));
+ f.set(utils,
this.getClass().getDeclaredField(usageUtilsFields.get(fieldName)).get(this));
+ } catch (IllegalArgumentException | IllegalAccessException
+ | NoSuchFieldException | SecurityException e) {
+ // TODO Auto-generated catch block
+ e.printStackTrace();
+ }
+ }
+
+ Method method;
+ try {
+ method = UsageEventUtils.class.getDeclaredMethod("init");
+ method.setAccessible(true);
+ method.invoke(utils);
+ } catch (NoSuchMethodException | SecurityException |
IllegalAccessException |
+ IllegalArgumentException | InvocationTargetException
e) {
+ // TODO Auto-generated catch block
+ e.printStackTrace();
+ }
+
+ return utils;
+ }
+
+ public void cleanupUsageUtils() {
+ UsageEventUtils utils = new UsageEventUtils();
+
+ for (String fieldName : oldFields.keySet()) {
+ try {
+ Field f =
UsageEventUtils.class.getDeclaredField(fieldName);
+ f.setAccessible(true);
+ f.set(utils, oldFields.get(fieldName));
+ } catch (IllegalArgumentException | IllegalAccessException
+ | NoSuchFieldException | SecurityException e) {
+ // TODO Auto-generated catch block
--- End diff --
Is there a TODO to be resolved?
---
If your project is set up for it, you can reply to this email and have your
reply appear on GitHub as well. If your project does not have this feature
enabled and wishes so, or if the feature is enabled but not working, please
contact infrastructure at infrastruct...@apache.org or file a JIRA ticket
with INFRA.
---
