def test_which_message_oneof(self):
payload = pb_util.build_payload("", 0, "", pb_util.mauka_pb2.MEASUREMENTS, [], 0, 0)
self.assertEqual(pb_util.which_message_oneof(payload), pb_util.PAYLOAD)
heartbeat = pb_util.build_heartbeat("", 0, 0, "", "")
self.assertEqual(pb_util.which_message_oneof(heartbeat), pb_util.HEARTBEAT)
makai_event = pb_util.build_makai_event("", 0)
self.assertEqual(pb_util.which_message_oneof(makai_event), pb_util.MAKAI_EVENT)
measurement = pb_util.build_measurement("", "", 0, 0, 0, 0)
self.assertEqual(pb_util.which_message_oneof(measurement), pb_util.MEASUREMENT)
makai_trigger = pb_util.build_makai_trigger("", 0, 0, "", 0, "")
self.assertEqual(pb_util.which_message_oneof(makai_trigger), pb_util.MAKAI_TRIGGER)
laha_ttl = pb_util.build_ttl("", "", 0)
self.assertEqual(pb_util.which_message_oneof(laha_ttl), pb_util.LAHA)
laha_gc_trigger = pb_util.build_gc_trigger("", [])
self.assertEqual(pb_util.which_message_oneof(laha_gc_trigger), pb_util.LAHA)
laha_gc_update = pb_util.build_gc_update("", pb_util.mauka_pb2.MEASUREMENTS, 0)
self.assertEqual(pb_util.which_message_oneof(laha_gc_update), pb_util.LAHA)
laha_gc_stat = pb_util.build_gc_stat("", pb_util.mauka_pb2.MEASUREMENTS, 0)
self.assertEqual(pb_util.which_message_oneof(laha_gc_stat), pb_util.LAHA)
trigger_request = pb_util.build_trigger_request("", 0, 0, [], 0)
self.assertEqual(pb_util.which_message_oneof(trigger_request), pb_util.TRIGGER_REQUEST)
triggered_event = pb_util.build_triggered_event("", [], 0, "", 0, 0)
self.assertEqual(pb_util.which_message_oneof(triggered_event), pb_util.TRIGGERED_EVENT)
threshold_optimization_request = pb_util.build_threshold_optimization_request("")
self.assertEqual(pb_util.which_message_oneof(threshold_optimization_request), pb_util.THRESHOLD_OPTIMIZATION_REQUEST)
box_optimization_request = pb_util.build_box_optimization_request("", [], 0)
self.assertEqual(pb_util.which_message_oneof(box_optimization_request), pb_util.BOX_OPTIMIZATION_REQUEST)
def test_makai_config_default_override(self):
threshold_req = pb_util.build_threshold_optimization_request("test",
box_id="0",
ref_f=61).threshold_optimization_request
self.makai_config.modify_thresholds(threshold_req)
self.assertEqual(self.makai_config.as_dict(), {
"_id": bson.ObjectId("5cd2152d49de19d1ea2b7a25"),
"triggering": {
"default_ref_f": 60,
"default_ref_v": 120,
"default_threshold_percent_f_low": 0.5,
"default_threshold_percent_f_high": 0.5,
"default_threshold_percent_v_low": 2.5,
"default_threshold_percent_v_high": 2.5,
"default_threshold_percent_thd_high": 3,
"triggering_overrides": [
{
"box_id": "0",
"ref_f": 61.0,
"ref_v": 120,
"threshold_percent_f_low": 1,
"threshold_percent_f_high": 1,
"threshold_percent_v_low": 5,
"threshold_percent_v_high": 5,
"threshold_percent_thd_high": 5
},
{
"box_id": "1003",
"ref_f": 60,
"ref_v": 120,
"threshold_percent_f_low": 0.5,
"threshold_percent_f_high": 0.5,
"threshold_percent_v_low": 2.5,
"threshold_percent_v_high": 2.5,
"threshold_percent_thd_high": 3
},
{
"box_id": "2001",
"ref_f": 60,
"ref_v": 120,
"threshold_percent_f_low": 0.5,
"threshold_percent_f_high": 0.5,
"threshold_percent_v_low": 2.5,
"threshold_percent_v_high": 2.5,
"threshold_percent_thd_high": 3
}
]
}
}
)
def test_trigger_override_none(self):
override = threshold_optimization_plugin._default_override(self.makai_config, "5000")
threshold_req = pb_util.build_threshold_optimization_request("test").threshold_optimization_request
override.modify_thresholds(threshold_req)
self.assertEqual(override.as_dict(),
{
"box_id": "5000",
"ref_f": 60,
"ref_v": 120,
"threshold_percent_f_low": 0.5,
"threshold_percent_f_high": 0.5,
"threshold_percent_v_low": 2.5,
"threshold_percent_v_high": 2.5,
"threshold_percent_thd_high": 3
})
def test_build_threshold_optimization_request_defaults(self):
mauka_message = pb_util.build_threshold_optimization_request("test")
self.assertTrue(isinstance(mauka_message, pb_util.mauka_pb2.MaukaMessage))
self.assertTrue(pb_util.is_threshold_optimization_request(mauka_message))
self.assertEqual(mauka_message.source, "test")
self.assertEqual(mauka_message.threshold_optimization_request.default_ref_f, 0.0)
self.assertEqual(mauka_message.threshold_optimization_request.default_ref_v, 0.0)
self.assertEqual(mauka_message.threshold_optimization_request.default_threshold_percent_f_low, 0.0)
self.assertEqual(mauka_message.threshold_optimization_request.default_threshold_percent_f_high, 0.0)
self.assertEqual(mauka_message.threshold_optimization_request.default_threshold_percent_v_low, 0.0)
self.assertEqual(mauka_message.threshold_optimization_request.default_threshold_percent_v_high, 0.0)
self.assertEqual(mauka_message.threshold_optimization_request.default_threshold_percent_thd_high, 0.0)
self.assertEqual(mauka_message.threshold_optimization_request.default_threshold_percent_thd_high, 0.0)
self.assertEqual(mauka_message.threshold_optimization_request.box_id, "")
self.assertEqual(mauka_message.threshold_optimization_request.ref_f, 0.0)
self.assertEqual(mauka_message.threshold_optimization_request.ref_v, 0.0)
self.assertEqual(mauka_message.threshold_optimization_request.threshold_percent_f_low, 0.0)
self.assertEqual(mauka_message.threshold_optimization_request.threshold_percent_f_high, 0.0)
self.assertEqual(mauka_message.threshold_optimization_request.threshold_percent_v_low, 0.0)
self.assertEqual(mauka_message.threshold_optimization_request.threshold_percent_v_high, 0.0)
self.assertEqual(mauka_message.threshold_optimization_request.threshold_percent_thd_high, 0.0)
self.assertEqual(mauka_message.threshold_optimization_request.threshold_percent_thd_high, 0.0)
def test_build_threshold_optimization_request(self):
mauka_message = pb_util.build_threshold_optimization_request("test",
1, 2, 3, 4, 5, 6, 7,
"1000",
8, 9, 10, 11, 12, 13, 14)
self.assertTrue(isinstance(mauka_message, pb_util.mauka_pb2.MaukaMessage))
self.assertTrue(pb_util.is_threshold_optimization_request(mauka_message))
self.assertEqual(mauka_message.source, "test")
self.assertEqual(mauka_message.threshold_optimization_request.default_ref_f, 1.0)
self.assertEqual(mauka_message.threshold_optimization_request.default_ref_v, 2.0)
self.assertEqual(mauka_message.threshold_optimization_request.default_threshold_percent_f_low, 3.0)
self.assertEqual(mauka_message.threshold_optimization_request.default_threshold_percent_f_high, 4.0)
self.assertEqual(mauka_message.threshold_optimization_request.default_threshold_percent_v_low, 5.0)
self.assertEqual(mauka_message.threshold_optimization_request.default_threshold_percent_v_high, 6.0)
self.assertEqual(mauka_message.threshold_optimization_request.default_threshold_percent_thd_high, 7.0)
self.assertEqual(mauka_message.threshold_optimization_request.box_id, "1000")
self.assertEqual(mauka_message.threshold_optimization_request.ref_f, 8.0)
self.assertEqual(mauka_message.threshold_optimization_request.ref_v, 9.0)
self.assertEqual(mauka_message.threshold_optimization_request.threshold_percent_f_low, 10.0)
self.assertEqual(mauka_message.threshold_optimization_request.threshold_percent_f_high, 11.0)
self.assertEqual(mauka_message.threshold_optimization_request.threshold_percent_v_low, 12.0)
self.assertEqual(mauka_message.threshold_optimization_request.threshold_percent_v_high, 13.0)
self.assertEqual(mauka_message.threshold_optimization_request.threshold_percent_thd_high, 14.0)
def test_makai_config_none(self):
threshold_req = pb_util.build_threshold_optimization_request("test").threshold_optimization_request
self.makai_config.modify_thresholds(threshold_req)
self.assertEqual(self.makai_config.as_dict(), MAKAI_CONFIG)
def test_is_threshold_optimization_request(self):
mauka_message = pb_util.build_threshold_optimization_request("")
other_message = pb_util.build_makai_event("", 0)
self.assertTrue(pb_util.is_threshold_optimization_request(mauka_message))
self.assertFalse(pb_util.is_threshold_optimization_request(other_message))