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))