def testLargeBlobsAreReassembledInCorrectOrder(self):
blob_data = b"0123456789"
blob_id = rdf_objects.BlobID(b"00234567" * 4)
self.blob_store.WriteBlobs({blob_id: blob_data})
result = self.blob_store.ReadBlobs([blob_id])
self.assertEqual({blob_id: blob_data}, result)
def testNotEvenlyDivisibleBlobsAreReassembledCorrectly(self):
blob_data = b"0123456789"
blob_id = rdf_objects.BlobID(b"00234567" * 4)
self.blob_store.WriteBlobs({blob_id: blob_data})
result = self.blob_store.ReadBlobs([blob_id])
self.assertEqual({blob_id: blob_data}, result)
def testFromBytesValidatesLength(self):
with self.assertRaises(ValueError):
rdf_objects.BlobID(b"foobar")
def testDefaultsToNull(self):
string = str(rdf_objects.BlobID())
self.assertEqual(string, "BlobID('{}')".format("0" * 64))
def testFromRDFBytes(self):
foo = rdf_objects.BlobID(rdfvalue.RDFBytes(b"12345678" * 4))
self.assertEqual(foo, b"12345678" * 4)
def testFromBytesValidatesType(self):
with self.assertRaises(TypeError):
rdf_objects.BlobID(42)
def testCheckBlobExistsReturnsTrueForExisting(self):
blob_id = rdf_objects.BlobID(b"01234567" * 4)
blob_data = b"abcdef"
self.blob_store.WriteBlobs({blob_id: blob_data})
self.assertTrue(self.blob_store.CheckBlobExists(blob_id))
def testFromBytes(self):
foo = rdf_objects.BlobID(b"12345678" * 4)
self.assertEqual(foo, b"12345678" * 4)
def testReadingNonExistentBlobsReturnsNone(self):
blob_id = rdf_objects.BlobID(b"01234567" * 4)
result = self.blob_store.ReadBlobs([blob_id])
self.assertEqual(result, {blob_id: None})
def testCheckBlobExistsReturnsFalseForMissing(self):
blob_id = rdf_objects.BlobID(b"11111111" * 4)
self.assertFalse(self.blob_store.CheckBlobExists(blob_id))
def testReadingNonExistentBlobReturnsNone(self):
blob_id = rdf_objects.BlobID(b"01234567" * 4)
result = self.blob_store.ReadBlob(blob_id)
self.assertIsNone(result)
def testOverwritingExistingBlobDoesNotRaise(self):
blob_id = rdf_objects.BlobID(b"01234567" * 4)
blob_data = b"abcdef"
for _ in range(2):
self.blob_store.WriteBlobs({blob_id: blob_data})
def testReadBlobIsMeasured(self):
blob_id = rdf_objects.BlobID(b"01234567" * 4)
self.assertMethodIsMeasured("ReadBlob", blob_id)
def testCheckBlobsExistIsMeasured(self):
blob_id = rdf_objects.BlobID(b"01234567" * 4)
self.assertMethodIsMeasured("CheckBlobsExist", [blob_id])
def testVerifyYaraSignatureReferenceIncorrect(self):
blob_id = rdf_objects.BlobID(os.urandom(32))
self.assertFalse(self.db.VerifyYaraSignatureReference(blob_id))
