test_filter_rules_ai.py

import os
import ubivar
import warnings
from ubivar.test.helper import (UbivarTestCase)

DUMMY_RULES_AI = {
        "feature": "a_rule_feature",
        "value": "a_rule_value.com",
        "is_active": "false"
        }


class UbivarAPIResourcesTests(UbivarTestCase):

    def test_filter_rules_ai_list(self):
        response = ubivar.FilterRulesAI.list()

        self.assertTrue(hasattr(response.data, "__iter__"))
        self.assertTrue(response.object == "filter_rules_ai")

    def test_filter_rules_ai_update(self):
        response = ubivar.FilterRulesAI.list()
        originalRulesAI = response.data[1]

        response = ubivar.FilterRulesAI.update("1", **DUMMY_RULES_AI)
        filterRulesAI = response.data[1]
        self.assertEqual(filterRulesAI["value"]    , originalRulesAI["value"])
        self.assertEqual(filterRulesAI["feature"]  , originalRulesAI["feature"])
        self.assertEqual(filterRulesAI["is_active"], DUMMY_RULES_AI["is_active"])

        response = ubivar.FilterRulesAI.update("1", is_active="true")
        filterRulesAI = response.data[1]
        self.assertEqual(filterRulesAI["is_active"]   , "true")
        self.assertEqual(response.object, "filter_rules_ai")