def test_full_pii_table(self): schema = Schema('public') table = Table(schema, 'full_pii') table.add_child(Column('name')) table.add_child(Column('location')) table.scan(self.data_generator) self.assertTrue(table.has_pii()) cols = table.get_children() self.assertTrue(cols[0].has_pii()) self.assertTrue(cols[1].has_pii()) self.assertEqual( { 'columns': [{ 'name': 'name', 'pii_types': [PiiTypes.PERSON] }, { 'name': 'location', 'pii_types': [PiiTypes.LOCATION] }], 'has_pii': True, 'name': 'full_pii' }, table.get_dict())
def test_partial_pii_table(self): schema = Schema('public') table = Table(schema, 'partial_pii') table.add_child(Column('a')) table.add_child(Column('b')) table.scan(self.data_generator) self.assertTrue(table.has_pii()) cols = table.get_children() self.assertTrue(cols[0].has_pii()) self.assertFalse(cols[1].has_pii()) self.assertEqual( { 'columns': [{ 'name': 'a', 'pii_types': [PiiTypes.PHONE] }, { 'name': 'b', 'pii_types': [] }], 'has_pii': True, 'name': 'partial_pii' }, table.get_dict())
def test_partial_pii_table(self): schema = Schema("public") table = Table(schema, "partial_pii") table.add_child(Column("a")) table.add_child(Column("b")) table.scan(self.data_generator) self.assertTrue(table.has_pii()) cols = table.get_children() self.assertTrue(cols[0].has_pii()) self.assertFalse(cols[1].has_pii()) self.assertEqual( { "columns": [ { "name": "a", "pii_types": [PiiTypes.PHONE] }, { "name": "b", "pii_types": [] }, ], "has_pii": True, "name": "partial_pii", }, table.get_dict(), )
def test_no_pii_table(self): schema = Schema("public") table = Table(schema, "no_pii") table.add_child(Column("a")) table.add_child(Column("b")) table.scan(self.data_generator) self.assertFalse(table.has_pii()) self.assertEqual( { "columns": [ { "name": "a", "pii_types": [] }, { "name": "b", "pii_types": [] }, ], "has_pii": False, "name": "no_pii", }, table.get_dict(), )
def test_full_pii_table(self): schema = Schema("public") table = Table(schema, "full_pii") table.add_child(Column("name")) table.add_child(Column("location")) table.scan(self.data_generator) self.assertTrue(table.has_pii()) cols = table.get_children() self.assertTrue(cols[0].has_pii()) self.assertTrue(cols[1].has_pii()) self.assertEqual( { "columns": [ { "name": "name", "pii_types": [PiiTypes.PERSON] }, { "name": "location", "pii_types": [PiiTypes.LOCATION] }, ], "has_pii": True, "name": "full_pii", }, table.get_dict(), )
def test_no_pii_table(self): schema = Schema('public') table = Table(schema, 'no_pii') table.add_child(Column('a')) table.add_child(Column('b')) table.scan(self.data_generator) self.assertFalse(table.has_pii()) self.assertEqual( { 'columns': [{ 'name': 'a', 'pii_types': [] }, { 'name': 'b', 'pii_types': [] }], 'has_pii': False, 'name': 'no_pii' }, table.get_dict())