def testDupeCerts(self):
# Create some cert entities, and a matching protobuf signing chain.
bit9_certs = test_utils.CreateBit9Certificates(3)
thumbprints = [c.key.id() for c in bit9_certs]
signing_chain = [
bit9_test_utils.CreateCertificate(thumbprint=t) for t in thumbprints]
bit9_test_utils.LinkSigningChain(*signing_chain)
self.assertEntityCount(bit9_models.Bit9Certificate, 3)
sync._PersistBit9Certificates(signing_chain).wait()
self.assertEntityCount(bit9_models.Bit9Certificate, 3)
def testNewCerts(self):
# Create some certs, and an unrelated signing chain.
test_utils.CreateBit9Certificates(3)
signing_chain = [
bit9_test_utils.CreateCertificate(thumbprint=test_utils.RandomSHA1())
for _ in xrange(4)]
bit9_test_utils.LinkSigningChain(*signing_chain)
self.assertEntityCount(bit9_db.Bit9Certificate, 3)
sync._PersistBit9Certificates(signing_chain).wait()
self.assertEntityCount(bit9_db.Bit9Certificate, 7)
def testNewCerts(self):
# Create some certs, and an unrelated signing chain.
test_utils.CreateBit9Certificates(3)
signing_chain = [
bit9_test_utils.CreateCertificate(thumbprint=test_utils.RandomSHA1())
for _ in xrange(4)]
bit9_test_utils.LinkSigningChain(*signing_chain)
self.assertEntityCount(bit9_models.Bit9Certificate, 3)
sync._PersistBit9Certificates(signing_chain).wait()
self.assertEntityCount(bit9_models.Bit9Certificate, 7)
self.assertBigQueryInsertions(
[constants.BIGQUERY_TABLE.CERTIFICATE] * len(signing_chain))
def testNoSigningChain(self): self.assertEntityCount(bit9_models.Bit9Certificate, 0) sync._PersistBit9Certificates([]).wait() self.assertEntityCount(bit9_models.Bit9Certificate, 0)