def METHOD_NAME(self):
    rule_ids = ['rule1', 'rule2']
    imsi = 'IMSI12345'
    self._session_rule_version_mapper.save_version(
        imsi, None, rule_ids[0], 1,
    )
    self.assertEqual(
        self._session_rule_version_mapper.get_version(
            imsi, None, rule_ids[0],
        ),
        1,
    )
    self._session_rule_version_mapper.save_version(
        imsi, None, rule_ids[1], 1,
    )
    self.assertEqual(
        self._session_rule_version_mapper.get_version(
            imsi, None, rule_ids[1],
        ),
        1,
    )
    self._session_rule_version_mapper.save_version(
        imsi, None, rule_ids[0], 2,
    )
    self.assertEqual(
        self._session_rule_version_mapper.get_version(
            imsi, None, rule_ids[0],
        ),
        2,
    )
    # Test updating version for all rules of a subscriber
    self._session_rule_version_mapper.remove_all_ue_versions(imsi, None)
    self.assertEqual(
        self._session_rule_version_mapper.get_version(
            imsi, None, rule_ids[0],
        ),
        0,
    )
    self.assertEqual(
        self._session_rule_version_mapper.get_version(
            imsi, None, rule_ids[1],
        ),
        0,
    )