Merge pull request #565 from ayobi/master

scrub interested users

Author: cassidysymons

microsetta_private_api/admin/admin_impl.py

@@ -836,6 +836,7 @@ def delete_account(account_id, token_info):
         src_repo = SourceRepo(t)
         samp_repo = SampleRepo(t)
         sar_repo = SurveyAnswersRepo(t)
+        interested_users_repo = InterestedUserRepo(t)
 
         acct = acct_repo.get_account(account_id)
         if acct is None:
@@ -873,6 +874,8 @@ def delete_account(account_id, token_info):
 
         acct_repo.scrub(account_id)
 
+        interested_users_repo.scrub(acct.email)
+
         t.commit()
 
     return None, 204

microsetta_private_api/repo/interested_user_repo.py

@@ -66,6 +66,9 @@ def update_interested_user(self, interested_user):
                 "state = %s, "
                 "postal_code = %s, "
                 "country = %s, "
+                "latitude = %s, "
+                "longitude = %s, "
+                "confirm_consent = %s, "
                 "address_checked = %s, "
                 "address_valid = %s, "
                 "residential_address = %s, "
@@ -82,6 +85,9 @@ def update_interested_user(self, interested_user):
                  interested_user.state,
                  interested_user.postal_code,
                  interested_user.country,
+                 interested_user.latitude,
+                 interested_user.longitude,
+                 interested_user.confirm_consent,
                  interested_user.address_checked,
                  interested_user.address_valid,
                  interested_user.residential_address,
@@ -207,3 +213,41 @@ def verify_address(self, interested_user_id):
                         (interested_user_id,)
                     )
                     return False
+
+    def get_interested_user_by_just_email(self, email):
+        with self._transaction.dict_cursor() as cur:
+            cur.execute(
+                "SELECT * FROM campaign.interested_users "
+                "WHERE lower(email) = lower(%s)",
+                (email,)
+            )
+            rs = cur.fetchall()
+            return [__class__._row_to_interested_user(r) for r in rs]
+
+    def scrub(self, interested_user_email):
+        interested_users = self. \
+            get_interested_user_by_just_email(interested_user_email)
+
+        for interested_user in interested_users:
+            interested_user.first_name = "scrubbed"
+            interested_user.last_name = "scrubbed"
+            interested_user.email = "scrubbed"
+            interested_user.phone = "scrubbed"
+            interested_user.address_1 = "scrubbed"
+            interested_user.address_2 = "scrubbed"
+            interested_user.address_3 = "scrubbed"
+            interested_user.city = "scrubbed"
+            interested_user.state = "scrubbed"
+            interested_user.postal_code = "scrubbed"
+            interested_user.country = "scrubbed"
+            interested_user.latitude = None
+            interested_user.longitude = None
+            interested_user.confirm_consent = False
+            interested_user.ip_address = "scrubbed"
+            interested_user.address_checked = False
+            interested_user.address_valid = False
+            interested_user.residential_address = False
+
+            if not self.update_interested_user(interested_user):
+                raise RepoException("Error scrubbing interested user: "
+                                    + interested_user.interested_user_id)

microsetta_private_api/repo/tests/test_interested_user_repo.py

@@ -324,6 +324,50 @@ def test_opt_out_interested_user(self):
             # Verify that the user is now opted out
             self.assertFalse(i_u.confirm_consent)
 
+    def test_scrub_interested_users_success(self):
+        # Create a test interested user
+        dummy_user = {
+            "campaign_id": self.test_campaign_id,
+            "first_name": "Test",
+            "last_name": "McTesterson",
+            "email": "[email protected]"
+        }
+        interested_user = InterestedUser.from_dict(dummy_user)
+        with Transaction() as t:
+            interested_user_repo = InterestedUserRepo(t)
+            iuid = interested_user_repo.insert_interested_user(interested_user)
+
+            # Scrub the interested user with the right email
+            interested_user_repo.scrub("[email protected]")
+
+            # Verify that the interested user is scrubbed
+            obs = interested_user_repo.get_interested_user_by_id(iuid)
+            self.assertEqual(obs.first_name, 'scrubbed')
+            self.assertEqual(obs.last_name, 'scrubbed')
+            self.assertEqual(obs.email, 'scrubbed')
+
+    def test_scrub_interested_users_failure(self):
+        # Create a test interested user
+        dummy_user = {
+            "campaign_id": self.test_campaign_id,
+            "first_name": "Test",
+            "last_name": "McTesterson",
+            "email": "[email protected]"
+        }
+        interested_user = InterestedUser.from_dict(dummy_user)
+        with Transaction() as t:
+            interested_user_repo = InterestedUserRepo(t)
+            iuid = interested_user_repo.insert_interested_user(interested_user)
+
+            # Scrub the interested user with the wrong email
+            interested_user_repo.scrub("[email protected]")
+
+            # Verify that the interested user isn't scrubbed
+            obs = interested_user_repo.get_interested_user_by_id(iuid)
+            self.assertEqual(obs.first_name, 'Test')
+            self.assertEqual(obs.last_name, 'McTesterson')
+            self.assertEqual(obs.email, '[email protected]')
+
 
 if __name__ == '__main__':
     unittest.main()