Add more CKA unit tests