Skip to content

Commit fa8d6e8

Browse files
committed
Added tests for custom and default RedisConsumers
1 parent 0d93f41 commit fa8d6e8

2 files changed

Lines changed: 11 additions & 3 deletions

File tree

.travis.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,4 +19,4 @@ install:
1919
script:
2020
nosetests --with-coverage --cover-package=snowplow_tracker
2121
after_success:
22-
coveralls
22+
coveralls

snowplow_tracker/test/integration/test_integration.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -149,14 +149,22 @@ def test_integration_standard_nv_pairs(self):
149149
for key in expected_fields:
150150
self.assertEquals(from_querystring(key, querystrings[-1]), expected_fields[key])
151151

152-
def test_integration_redis(self):
152+
def test_integration_redis_default(self):
153153
r = redis.StrictRedis()
154-
t = tracker.Tracker(consumer.RedisConsumer(rdb=r), default_subject)
154+
t = tracker.Tracker(consumer.RedisConsumer(), default_subject)
155155
t.track_page_view("http://www.example.com")
156156
event_string = r.rpop("snowplow")
157157
event_dict = json.loads(event_string.decode("utf-8"))
158158
self.assertEquals(event_dict["e"], "pv")
159159

160+
def test_integration_redis_custom(self):
161+
r = redis.StrictRedis(db=1)
162+
t = tracker.Tracker(consumer.RedisConsumer(rdb=r, key="custom_key"), default_subject)
163+
t.track_page_view("http://www.example.com")
164+
event_string = r.rpop("custom_key")
165+
event_dict = json.loads(event_string.decode("utf-8"))
166+
self.assertEquals(event_dict["e"], "pv")
167+
160168
def test_integration_success_callback(self):
161169
callback_queue = []
162170
callback_consumer = consumer.Consumer("localhost", on_success=lambda x: callback_queue.append(x))

0 commit comments

Comments
 (0)