def testNoBans(self): bit9_binary = test_utils.CreateBit9Binary() file_catalog = bit9_test_utils.CreateFileCatalog( id=bit9_binary.file_catalog_id, sha256=bit9_binary.key.id(), certificate_state=bit9_constants.APPROVAL_STATE.APPROVED, file_state=bit9_constants.APPROVAL_STATE.APPROVED, publisher_state=bit9_constants.APPROVAL_STATE.APPROVED) self.assertEntityCount(base_models.Note, 0) sync._PersistBanNote(file_catalog).wait() self.assertEntityCount(base_models.Note, 0)
def testNewBan(self): bit9_binary = test_utils.CreateBit9Binary() # Create FileCatalog with at least one that's BANNED. file_catalog = bit9_test_utils.CreateFileCatalog( id=bit9_binary.file_catalog_id, sha256=bit9_binary.key.id(), certificate_state=bit9_constants.APPROVAL_STATE.BANNED, file_state=bit9_constants.APPROVAL_STATE.APPROVED, publisher_state=bit9_constants.APPROVAL_STATE.APPROVED) self.assertEntityCount(base_db.Note, 0) sync._PersistBanNote(file_catalog).wait() self.assertEntityCount(base_db.Note, 1)