Merge branch 'pulse-3.1.4-prep'

.gitignore

@@ -57,4 +57,7 @@ build_full_environment.sh
 docker-compose.build.yml
 
 # Local environment things
-.condaauto
+.condaauto
+
+# VS Code
+/.vscode

tests/test_traptor_offline.py

@@ -378,6 +378,38 @@ def test_ensure_internal_rule_counters_are_correctly_made_for_rules(self, redis_
         if traptor.traptor_type == 'locations':
             assert len(traptor.rule_counters) == 0
 
+    def test_ensure_internal_rule_counters_are_properly_deleted(self, redis_rules, traptor):
+        """Ensure _delete_rule_counters are properly deleted."""
+        traptor._setup()
+        traptor.redis_conn = redis_rules
+
+        traptor.logger.info = MagicMock()
+        traptor.logger.error = MagicMock()
+
+        traptor.redis_rules = [rule for rule in traptor._get_redis_rules()]
+        traptor.twitter_rules = traptor._make_twitter_rules(traptor.redis_rules)
+
+        if traptor.traptor_type != 'locations':
+            traptor._make_rule_counters()
+            assert len(traptor.rule_counters) == 1
+
+        if traptor.traptor_type == 'track':
+            assert traptor.rule_counters['12347'] is not None
+
+            traptor._delete_rule_counters()
+            assert traptor.logger.error.call_count == 0
+            assert traptor.logger.info.call_count == 3
+
+        if traptor.traptor_type == 'follow':
+            assert traptor.rule_counters['12345'] is not None
+
+            traptor._delete_rule_counters()
+            assert traptor.logger.error.call_count == 0
+            assert traptor.logger.info.call_count == 3
+
+        if traptor.traptor_type == 'locations':
+            assert len(traptor.rule_counters) == 0
+
     def test_ensure_limit_message_counter_is_correctly_created(self, redis_rules, traptor):
         """Ensure _make_limit_message_counter makes the limit counter"""
         traptor._setup()

tests/traptor_integration_tests.py

@@ -32,6 +32,7 @@
 HOST_FOR_TESTING = os.getenv('REDIS_HOST', 'localhost')
 TRAPTOR_TYPE = os.getenv('TRAPTOR_TYPE', 'track')
 TRAPTOR_ID = int(os.getenv('TRAPTOR_ID', 0))
+RULE_ID = 12348
 
 # Create a connection to Redis
 
@@ -51,14 +52,18 @@
     # Collect on tweets with the keyword `python`
     # Change the rule value if you want to collect on a different keyword or hashtag. Go nuts!
     test_track_rule = {
-      "tag": "Traptor.Test",
-      "type": "track",
-      "value": "python",
-      "description": "Test track rule for python"
+        "tag": "Traptor.Test", 
+        "value": "python", 
+        "status": "active", 
+        "description": "Test track rule for python", 
+        "appid": "test-appid", 
+        "date_added": "2017-04-02 16:58:34", 
+        "rule_type": "track", 
+        "rule_id": RULE_ID
     }
 
     try:
-        redis_key = "traptor-{}:{}:{}".format(TRAPTOR_TYPE, TRAPTOR_ID, 123)
+        redis_key = "traptor-{}:{}:{}".format(TRAPTOR_TYPE, TRAPTOR_ID, RULE_ID)
         redis_connection.hmset(redis_key, test_track_rule)
 
         print("Rule added")

traptor/traptor.py

@@ -377,6 +377,21 @@ def _increment_rule_counter(self, tweet):
             dd_monitoring.increment('redis_error',
                                     tags=['error_type:connection_error'])
 
+    def _delete_rule_counters(self):
+        """
+        Delete the existing rule counters.
+        """
+        if len(self.rule_counters) > 0:
+            for counter in self.rule_counters:
+                try:
+                    self.rule_counters[counter].delete_key()
+                except:
+                    self.logger.error("Caught exception while deleting a rule counter", extra={
+                        'error_type': 'RedisConnectionError',
+                        'ex': traceback.format_exc()
+                    })
+            self.logger.info("Rule counters deleted successfully.")
+
     def _make_limit_message_counter(self):
         """
         Make a limit message counter to track the values of incoming limit messages.
@@ -1032,6 +1047,7 @@ def run(self):
 
         # Do all the things
         while True:
+            self._delete_rule_counters()
             self._wait_for_rules()
 
             # Concatenate all of the rule['value'] fields