@@ -83,11 +83,15 @@ def test_credit_card(self):
8383 def test_device (self ):
8484 id = 'b643d445-18b2-4b9d-bad4-c9c4366e402a'
8585 last_seen = '2016-06-08T14:16:38Z'
86- device = Device ({'confidence' : 99 , 'id' : id , 'last_seen' : last_seen })
86+ session_age = 3600
87+ session_id = 'foobar'
88+ device = Device ({'confidence' : 99 , 'id' : id , 'last_seen' : last_seen , 'session_age' : session_age , 'session_id' : session_id })
8789
8890 self .assertEqual (99 , device .confidence )
8991 self .assertEqual (id , device .id )
9092 self .assertEqual (last_seen , device .last_seen )
93+ self .assertEqual (session_age , device .session_age )
94+ self .assertEqual (session_id , device .session_id )
9195
9296 def test_disposition (self ):
9397 disposition = Disposition ({'action' : 'accept' , 'reason' : 'default' })
@@ -96,8 +100,10 @@ def test_disposition(self):
96100 self .assertEqual ('default' , disposition .reason )
97101
98102 def test_email (self ):
99- email = Email ({'is_free' : True , 'is_high_risk' : False })
103+ first_seen = '2016-01-01' ;
104+ email = Email ({'first_seen' : first_seen , 'is_free' : True , 'is_high_risk' : False })
100105
106+ self .assertEqual (first_seen , email .first_seen )
101107 self .assertEqual (True , email .is_free )
102108 self .assertEqual (False , email .is_high_risk )
103109
0 commit comments