/*
 *
 *    Copyright (c) 2013-2017 Nest Labs, Inc.
 *    All rights reserved.
 *
 *    Licensed under the Apache License, Version 2.0 (the "License");
 *    you may not use this file except in compliance with the License.
 *    You may obtain a copy of the License at
 *
 *        http://www.apache.org/licenses/LICENSE-2.0
 *
 *    Unless required by applicable law or agreed to in writing, software
 *    distributed under the License is distributed on an "AS IS" BASIS,
 *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *    See the License for the specific language governing permissions and
 *    limitations under the License.
 */

import com.nestlabs.weave.security.HKDF;

public class HKDFTest extends SimpleTest {

    public static void main(String[] args)
    {
        HKDFTest testObj = new HKDFTest();
        
        int res = testObj.runTests(new String[] {
            "testHKDF_RFC5869_TestCase1",
            "testHKDF_RFC5869_TestCase2",
            "testHKDF_RFC5869_TestCase3",
            "testHKDF_RFC5869_TestCase4",
            "testHKDF_RFC5869_TestCase5",
            "testHKDF_RFC5869_TestCase6",
        });
        
        System.exit(res);
    }
    
    public void testHKDF_RFC5869_TestCase1() throws Exception {
        
        final byte[] ikm  = new byte[] { 
            (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, 
            (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b,
            (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b };
        final byte[] salt = new byte[] { 
            (byte)0x00, (byte)0x01, (byte)0x02, (byte)0x03, (byte)0x04, (byte)0x05, (byte)0x06, (byte)0x07, 
            (byte)0x08, (byte)0x09, (byte)0x0a, (byte)0x0b, (byte)0x0c };
        final byte[] info = new byte[] {
            (byte)0xf0, (byte)0xf1, (byte)0xf2, (byte)0xf3, (byte)0xf4, (byte)0xf5, (byte)0xf6, (byte)0xf7, 
            (byte)0xf8, (byte)0xf9 };
        final byte[] prk  = new byte[] {
            (byte)0x07, (byte)0x77, (byte)0x09, (byte)0x36, (byte)0x2c, (byte)0x2e, (byte)0x32, (byte)0xdf,
            (byte)0x0d, (byte)0xdc, (byte)0x3f, (byte)0x0d, (byte)0xc4, (byte)0x7b, (byte)0xba, (byte)0x63, 
            (byte)0x90, (byte)0xb6, (byte)0xc7, (byte)0x3b, (byte)0xb5, (byte)0x0f, (byte)0x9c, (byte)0x31,
            (byte)0x22, (byte)0xec, (byte)0x84, (byte)0x4a, (byte)0xd7, (byte)0xc2, (byte)0xb3, (byte)0xe5 };
        final byte[] okm  = new byte[] { 
            (byte)0x3c, (byte)0xb2, (byte)0x5f, (byte)0x25, (byte)0xfa, (byte)0xac, (byte)0xd5, (byte)0x7a,
            (byte)0x90, (byte)0x43, (byte)0x4f, (byte)0x64, (byte)0xd0, (byte)0x36, (byte)0x2f, (byte)0x2a,
            (byte)0x2d, (byte)0x2d, (byte)0x0a, (byte)0x90, (byte)0xcf, (byte)0x1a, (byte)0x5a, (byte)0x4c,
            (byte)0x5d, (byte)0xb0, (byte)0x2d, (byte)0x56, (byte)0xec, (byte)0xc4, (byte)0xc5, (byte)0xbf,
            (byte)0x34, (byte)0x00, (byte)0x72, (byte)0x08, (byte)0xd5, (byte)0xb8, (byte)0x87, (byte)0x18,
            (byte)0x58, (byte)0x65 };

        testHKDF("HKDFSHA256", ikm, salt, info, prk, okm);
    }

    public void testHKDF_RFC5869_TestCase2() throws Exception {
        
        final byte[] ikm  = new byte[] { 
            (byte)0x00, (byte)0x01, (byte)0x02, (byte)0x03, (byte)0x04, (byte)0x05, (byte)0x06, (byte)0x07,
            (byte)0x08, (byte)0x09, (byte)0x0a, (byte)0x0b, (byte)0x0c, (byte)0x0d, (byte)0x0e, (byte)0x0f,
            (byte)0x10, (byte)0x11, (byte)0x12, (byte)0x13, (byte)0x14, (byte)0x15, (byte)0x16, (byte)0x17,
            (byte)0x18, (byte)0x19, (byte)0x1a, (byte)0x1b, (byte)0x1c, (byte)0x1d, (byte)0x1e, (byte)0x1f,
            (byte)0x20, (byte)0x21, (byte)0x22, (byte)0x23, (byte)0x24, (byte)0x25, (byte)0x26, (byte)0x27,
            (byte)0x28, (byte)0x29, (byte)0x2a, (byte)0x2b, (byte)0x2c, (byte)0x2d, (byte)0x2e, (byte)0x2f,
            (byte)0x30, (byte)0x31, (byte)0x32, (byte)0x33, (byte)0x34, (byte)0x35, (byte)0x36, (byte)0x37,
            (byte)0x38, (byte)0x39, (byte)0x3a, (byte)0x3b, (byte)0x3c, (byte)0x3d, (byte)0x3e, (byte)0x3f,
            (byte)0x40, (byte)0x41, (byte)0x42, (byte)0x43, (byte)0x44, (byte)0x45, (byte)0x46, (byte)0x47,
            (byte)0x48, (byte)0x49, (byte)0x4a, (byte)0x4b, (byte)0x4c, (byte)0x4d, (byte)0x4e, (byte)0x4f
        };
        final byte[] salt = new byte[] { 
            (byte)0x60, (byte)0x61, (byte)0x62, (byte)0x63, (byte)0x64, (byte)0x65, (byte)0x66, (byte)0x67,
            (byte)0x68, (byte)0x69, (byte)0x6a, (byte)0x6b, (byte)0x6c, (byte)0x6d, (byte)0x6e, (byte)0x6f,
            (byte)0x70, (byte)0x71, (byte)0x72, (byte)0x73, (byte)0x74, (byte)0x75, (byte)0x76, (byte)0x77,
            (byte)0x78, (byte)0x79, (byte)0x7a, (byte)0x7b, (byte)0x7c, (byte)0x7d, (byte)0x7e, (byte)0x7f,
            (byte)0x80, (byte)0x81, (byte)0x82, (byte)0x83, (byte)0x84, (byte)0x85, (byte)0x86, (byte)0x87,
            (byte)0x88, (byte)0x89, (byte)0x8a, (byte)0x8b, (byte)0x8c, (byte)0x8d, (byte)0x8e, (byte)0x8f,
            (byte)0x90, (byte)0x91, (byte)0x92, (byte)0x93, (byte)0x94, (byte)0x95, (byte)0x96, (byte)0x97,
            (byte)0x98, (byte)0x99, (byte)0x9a, (byte)0x9b, (byte)0x9c, (byte)0x9d, (byte)0x9e, (byte)0x9f,
            (byte)0xa0, (byte)0xa1, (byte)0xa2, (byte)0xa3, (byte)0xa4, (byte)0xa5, (byte)0xa6, (byte)0xa7, 
            (byte)0xa8, (byte)0xa9, (byte)0xaa, (byte)0xab, (byte)0xac, (byte)0xad, (byte)0xae, (byte)0xaf
        };
        final byte[] info = new byte[] { 
            (byte)0xb0, (byte)0xb1, (byte)0xb2, (byte)0xb3, (byte)0xb4, (byte)0xb5, (byte)0xb6, (byte)0xb7,
            (byte)0xb8, (byte)0xb9, (byte)0xba, (byte)0xbb, (byte)0xbc, (byte)0xbd, (byte)0xbe, (byte)0xbf,
            (byte)0xc0, (byte)0xc1, (byte)0xc2, (byte)0xc3, (byte)0xc4, (byte)0xc5, (byte)0xc6, (byte)0xc7,
            (byte)0xc8, (byte)0xc9, (byte)0xca, (byte)0xcb, (byte)0xcc, (byte)0xcd, (byte)0xce, (byte)0xcf,
            (byte)0xd0, (byte)0xd1, (byte)0xd2, (byte)0xd3, (byte)0xd4, (byte)0xd5, (byte)0xd6, (byte)0xd7,
            (byte)0xd8, (byte)0xd9, (byte)0xda, (byte)0xdb, (byte)0xdc, (byte)0xdd, (byte)0xde, (byte)0xdf,
            (byte)0xe0, (byte)0xe1, (byte)0xe2, (byte)0xe3, (byte)0xe4, (byte)0xe5, (byte)0xe6, (byte)0xe7,
            (byte)0xe8, (byte)0xe9, (byte)0xea, (byte)0xeb, (byte)0xec, (byte)0xed, (byte)0xee, (byte)0xef,
            (byte)0xf0, (byte)0xf1, (byte)0xf2, (byte)0xf3, (byte)0xf4, (byte)0xf5, (byte)0xf6, (byte)0xf7,
            (byte)0xf8, (byte)0xf9, (byte)0xfa, (byte)0xfb, (byte)0xfc, (byte)0xfd, (byte)0xfe, (byte)0xff
        };
        final byte[] prk  = new byte[] { 
            (byte)0x06, (byte)0xa6, (byte)0xb8, (byte)0x8c, (byte)0x58, (byte)0x53, (byte)0x36, (byte)0x1a,
            (byte)0x06, (byte)0x10, (byte)0x4c, (byte)0x9c, (byte)0xeb, (byte)0x35, (byte)0xb4, (byte)0x5c,
            (byte)0xef, (byte)0x76, (byte)0x00, (byte)0x14, (byte)0x90, (byte)0x46, (byte)0x71, (byte)0x01,
            (byte)0x4a, (byte)0x19, (byte)0x3f, (byte)0x40, (byte)0xc1, (byte)0x5f, (byte)0xc2, (byte)0x44
        };
        final byte[] okm  = new byte[] { 
            (byte)0xb1, (byte)0x1e, (byte)0x39, (byte)0x8d, (byte)0xc8, (byte)0x03, (byte)0x27, (byte)0xa1,
            (byte)0xc8, (byte)0xe7, (byte)0xf7, (byte)0x8c, (byte)0x59, (byte)0x6a, (byte)0x49, (byte)0x34,
            (byte)0x4f, (byte)0x01, (byte)0x2e, (byte)0xda, (byte)0x2d, (byte)0x4e, (byte)0xfa, (byte)0xd8,
            (byte)0xa0, (byte)0x50, (byte)0xcc, (byte)0x4c, (byte)0x19, (byte)0xaf, (byte)0xa9, (byte)0x7c,
            (byte)0x59, (byte)0x04, (byte)0x5a, (byte)0x99, (byte)0xca, (byte)0xc7, (byte)0x82, (byte)0x72,
            (byte)0x71, (byte)0xcb, (byte)0x41, (byte)0xc6, (byte)0x5e, (byte)0x59, (byte)0x0e, (byte)0x09,
            (byte)0xda, (byte)0x32, (byte)0x75, (byte)0x60, (byte)0x0c, (byte)0x2f, (byte)0x09, (byte)0xb8,
            (byte)0x36, (byte)0x77, (byte)0x93, (byte)0xa9, (byte)0xac, (byte)0xa3, (byte)0xdb, (byte)0x71,
            (byte)0xcc, (byte)0x30, (byte)0xc5, (byte)0x81, (byte)0x79, (byte)0xec, (byte)0x3e, (byte)0x87,
            (byte)0xc1, (byte)0x4c, (byte)0x01, (byte)0xd5, (byte)0xc1, (byte)0xf3, (byte)0x43, (byte)0x4f,
            (byte)0x1d, (byte)0x87
        };

        testHKDF("HKDFSHA256", ikm, salt, info, prk, okm);
    }

    public void testHKDF_RFC5869_TestCase3() throws Exception {

        final byte[] ikm  = new byte[] { 
            (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, 
            (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b,
            (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b };
        final byte[] salt = new byte[] { };
        final byte[] info = new byte[] { };
        final byte[] prk  = new byte[] {
            (byte)0x19, (byte)0xef, (byte)0x24, (byte)0xa3, (byte)0x2c, (byte)0x71, (byte)0x7b, (byte)0x16,
            (byte)0x7f, (byte)0x33, (byte)0xa9, (byte)0x1d, (byte)0x6f, (byte)0x64, (byte)0x8b, (byte)0xdf,
            (byte)0x96, (byte)0x59, (byte)0x67, (byte)0x76, (byte)0xaf, (byte)0xdb, (byte)0x63, (byte)0x77,
            (byte)0xac, (byte)0x43, (byte)0x4c, (byte)0x1c, (byte)0x29, (byte)0x3c, (byte)0xcb, (byte)0x04, };
        final byte[] okm  = new byte[] {
            (byte)0x8d, (byte)0xa4, (byte)0xe7, (byte)0x75, (byte)0xa5, (byte)0x63, (byte)0xc1, (byte)0x8f,
            (byte)0x71, (byte)0x5f, (byte)0x80, (byte)0x2a, (byte)0x06, (byte)0x3c, (byte)0x5a, (byte)0x31,
            (byte)0xb8, (byte)0xa1, (byte)0x1f, (byte)0x5c, (byte)0x5e, (byte)0xe1, (byte)0x87, (byte)0x9e,
            (byte)0xc3, (byte)0x45, (byte)0x4e, (byte)0x5f, (byte)0x3c, (byte)0x73, (byte)0x8d, (byte)0x2d,
            (byte)0x9d, (byte)0x20, (byte)0x13, (byte)0x95, (byte)0xfa, (byte)0xa4, (byte)0xb6, (byte)0x1a, 
            (byte)0x96, (byte)0xc8 };

        testHKDF("HKDFSHA256", ikm, salt, info, prk, okm);
    }

    public void testHKDF_RFC5869_TestCase4() throws Exception {
     
        final byte[] ikm  = new byte[] { 
            (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, 
            (byte)0x0b, (byte)0x0b, (byte)0x0b };
        final byte[] salt = new byte[] { 
            (byte)0x00, (byte)0x01, (byte)0x02, (byte)0x03, (byte)0x04, (byte)0x05, (byte)0x06, (byte)0x07, 
            (byte)0x08, (byte)0x09, (byte)0x0a, (byte)0x0b, (byte)0x0c };
        final byte[] info = new byte[] { 
            (byte)0xf0, (byte)0xf1, (byte)0xf2, (byte)0xf3, (byte)0xf4, (byte)0xf5, (byte)0xf6, (byte)0xf7, 
            (byte)0xf8, (byte)0xf9 };
        final byte[] prk  = new byte[] { 
            (byte)0x9b, (byte)0x6c, (byte)0x18, (byte)0xc4, (byte)0x32, (byte)0xa7, (byte)0xbf, (byte)0x8f, 
            (byte)0x0e, (byte)0x71, (byte)0xc8, (byte)0xeb, (byte)0x88, (byte)0xf4, (byte)0xb3, (byte)0x0b, 
            (byte)0xaa, (byte)0x2b, (byte)0xa2, (byte)0x43 };
        final byte[] okm  = new byte[] {
            (byte)0x08, (byte)0x5a, (byte)0x01, (byte)0xea, (byte)0x1b, (byte)0x10, (byte)0xf3, (byte)0x69, 
            (byte)0x33, (byte)0x06, (byte)0x8b, (byte)0x56, (byte)0xef, (byte)0xa5, (byte)0xad, (byte)0x81,
            (byte)0xa4, (byte)0xf1, (byte)0x4b, (byte)0x82, (byte)0x2f, (byte)0x5b, (byte)0x09, (byte)0x15, 
            (byte)0x68, (byte)0xa9, (byte)0xcd, (byte)0xd4, (byte)0xf1, (byte)0x55, (byte)0xfd, (byte)0xa2,
            (byte)0xc2, (byte)0x2e, (byte)0x42, (byte)0x24, (byte)0x78, (byte)0xd3, (byte)0x05, (byte)0xf3,
            (byte)0xf8, (byte)0x96 };

        testHKDF("HKDFSHA1", ikm, salt, info, prk, okm);
    }
    
    public void testHKDF_RFC5869_TestCase5() throws Exception {

        final byte[] ikm  = new byte[] { 
            (byte)0x00, (byte)0x01, (byte)0x02, (byte)0x03, (byte)0x04, (byte)0x05, (byte)0x06, (byte)0x07, 
            (byte)0x08, (byte)0x09, (byte)0x0a, (byte)0x0b, (byte)0x0c, (byte)0x0d, (byte)0x0e, (byte)0x0f,
            (byte)0x10, (byte)0x11, (byte)0x12, (byte)0x13, (byte)0x14, (byte)0x15, (byte)0x16, (byte)0x17,
            (byte)0x18, (byte)0x19, (byte)0x1a, (byte)0x1b, (byte)0x1c, (byte)0x1d, (byte)0x1e, (byte)0x1f,
            (byte)0x20, (byte)0x21, (byte)0x22, (byte)0x23, (byte)0x24, (byte)0x25, (byte)0x26, (byte)0x27,
            (byte)0x28, (byte)0x29, (byte)0x2a, (byte)0x2b, (byte)0x2c, (byte)0x2d, (byte)0x2e, (byte)0x2f,
            (byte)0x30, (byte)0x31, (byte)0x32, (byte)0x33, (byte)0x34, (byte)0x35, (byte)0x36, (byte)0x37,
            (byte)0x38, (byte)0x39, (byte)0x3a, (byte)0x3b, (byte)0x3c, (byte)0x3d, (byte)0x3e, (byte)0x3f,
            (byte)0x40, (byte)0x41, (byte)0x42, (byte)0x43, (byte)0x44, (byte)0x45, (byte)0x46, (byte)0x47,
            (byte)0x48, (byte)0x49, (byte)0x4a, (byte)0x4b, (byte)0x4c, (byte)0x4d, (byte)0x4e, (byte)0x4f };
        final byte[] salt = new byte[] { 
            (byte)0x60, (byte)0x61, (byte)0x62, (byte)0x63, (byte)0x64, (byte)0x65, (byte)0x66, (byte)0x67,
            (byte)0x68, (byte)0x69, (byte)0x6a, (byte)0x6b, (byte)0x6c, (byte)0x6d, (byte)0x6e, (byte)0x6f,
            (byte)0x70, (byte)0x71, (byte)0x72, (byte)0x73, (byte)0x74, (byte)0x75, (byte)0x76, (byte)0x77,
            (byte)0x78, (byte)0x79, (byte)0x7a, (byte)0x7b, (byte)0x7c, (byte)0x7d, (byte)0x7e, (byte)0x7f,
            (byte)0x80, (byte)0x81, (byte)0x82, (byte)0x83, (byte)0x84, (byte)0x85, (byte)0x86, (byte)0x87,
            (byte)0x88, (byte)0x89, (byte)0x8a, (byte)0x8b, (byte)0x8c, (byte)0x8d, (byte)0x8e, (byte)0x8f,
            (byte)0x90, (byte)0x91, (byte)0x92, (byte)0x93, (byte)0x94, (byte)0x95, (byte)0x96, (byte)0x97,
            (byte)0x98, (byte)0x99, (byte)0x9a, (byte)0x9b, (byte)0x9c, (byte)0x9d, (byte)0x9e, (byte)0x9f,
            (byte)0xa0, (byte)0xa1, (byte)0xa2, (byte)0xa3, (byte)0xa4, (byte)0xa5, (byte)0xa6, (byte)0xa7,
            (byte)0xa8, (byte)0xa9, (byte)0xaa, (byte)0xab, (byte)0xac, (byte)0xad, (byte)0xae, (byte)0xaf };
        final byte[] info = new byte[] {
            (byte)0xb0, (byte)0xb1, (byte)0xb2, (byte)0xb3, (byte)0xb4, (byte)0xb5, (byte)0xb6, (byte)0xb7,
            (byte)0xb8, (byte)0xb9, (byte)0xba, (byte)0xbb, (byte)0xbc, (byte)0xbd, (byte)0xbe, (byte)0xbf,
            (byte)0xc0, (byte)0xc1, (byte)0xc2, (byte)0xc3, (byte)0xc4, (byte)0xc5, (byte)0xc6, (byte)0xc7,
            (byte)0xc8, (byte)0xc9, (byte)0xca, (byte)0xcb, (byte)0xcc, (byte)0xcd, (byte)0xce, (byte)0xcf,
            (byte)0xd0, (byte)0xd1, (byte)0xd2, (byte)0xd3, (byte)0xd4, (byte)0xd5, (byte)0xd6, (byte)0xd7,
            (byte)0xd8, (byte)0xd9, (byte)0xda, (byte)0xdb, (byte)0xdc, (byte)0xdd, (byte)0xde, (byte)0xdf,
            (byte)0xe0, (byte)0xe1, (byte)0xe2, (byte)0xe3, (byte)0xe4, (byte)0xe5, (byte)0xe6, (byte)0xe7,
            (byte)0xe8, (byte)0xe9, (byte)0xea, (byte)0xeb, (byte)0xec, (byte)0xed, (byte)0xee, (byte)0xef,
            (byte)0xf0, (byte)0xf1, (byte)0xf2, (byte)0xf3, (byte)0xf4, (byte)0xf5, (byte)0xf6, (byte)0xf7,
            (byte)0xf8, (byte)0xf9, (byte)0xfa, (byte)0xfb, (byte)0xfc, (byte)0xfd, (byte)0xfe, (byte)0xff };
        final byte[] prk  = new byte[] { 
            (byte)0x8a, (byte)0xda, (byte)0xe0, (byte)0x9a, (byte)0x2a, (byte)0x30, (byte)0x70, (byte)0x59,
            (byte)0x47, (byte)0x8d, (byte)0x30, (byte)0x9b, (byte)0x26, (byte)0xc4, (byte)0x11, (byte)0x5a,
            (byte)0x22, (byte)0x4c, (byte)0xfa, (byte)0xf6 };
        final byte[] okm  = new byte[] { 
            (byte)0x0b, (byte)0xd7, (byte)0x70, (byte)0xa7, (byte)0x4d, (byte)0x11, (byte)0x60, (byte)0xf7,
            (byte)0xc9, (byte)0xf1, (byte)0x2c, (byte)0xd5, (byte)0x91, (byte)0x2a, (byte)0x06, (byte)0xeb,
            (byte)0xff, (byte)0x6a, (byte)0xdc, (byte)0xae, (byte)0x89, (byte)0x9d, (byte)0x92, (byte)0x19,
            (byte)0x1f, (byte)0xe4, (byte)0x30, (byte)0x56, (byte)0x73, (byte)0xba, (byte)0x2f, (byte)0xfe,
            (byte)0x8f, (byte)0xa3, (byte)0xf1, (byte)0xa4, (byte)0xe5, (byte)0xad, (byte)0x79, (byte)0xf3,
            (byte)0xf3, (byte)0x34, (byte)0xb3, (byte)0xb2, (byte)0x02, (byte)0xb2, (byte)0x17, (byte)0x3c,
            (byte)0x48, (byte)0x6e, (byte)0xa3, (byte)0x7c, (byte)0xe3, (byte)0xd3, (byte)0x97, (byte)0xed,
            (byte)0x03, (byte)0x4c, (byte)0x7f, (byte)0x9d, (byte)0xfe, (byte)0xb1, (byte)0x5c, (byte)0x5e,
            (byte)0x92, (byte)0x73, (byte)0x36, (byte)0xd0, (byte)0x44, (byte)0x1f, (byte)0x4c, (byte)0x43,
            (byte)0x00, (byte)0xe2, (byte)0xcf, (byte)0xf0, (byte)0xd0, (byte)0x90, (byte)0x0b, (byte)0x52,
            (byte)0xd3, (byte)0xb4, };

        testHKDF("HKDFSHA1", ikm, salt, info, prk, okm);
    }

    public void testHKDF_RFC5869_TestCase6() throws Exception {

        final byte[] ikm  = new byte[] {
            (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, 
            (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b,
            (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b, (byte)0x0b };
        final byte[] salt = new byte[] { };
        final byte[] info = new byte[] { };
        final byte[] prk  = new byte[] {
            (byte)0xda, (byte)0x8c, (byte)0x8a, (byte)0x73, (byte)0xc7, (byte)0xfa, (byte)0x77, (byte)0x28,
            (byte)0x8e, (byte)0xc6, (byte)0xf5, (byte)0xe7, (byte)0xc2, (byte)0x97, (byte)0x78, (byte)0x6a,
            (byte)0xa0, (byte)0xd3, (byte)0x2d, (byte)0x01 };
        final byte[] okm  = new byte[] {
            (byte)0x0a, (byte)0xc1, (byte)0xaf, (byte)0x70, (byte)0x02, (byte)0xb3, (byte)0xd7, (byte)0x61,
            (byte)0xd1, (byte)0xe5, (byte)0x52, (byte)0x98, (byte)0xda, (byte)0x9d, (byte)0x05, (byte)0x06,
            (byte)0xb9, (byte)0xae, (byte)0x52, (byte)0x05, (byte)0x72, (byte)0x20, (byte)0xa3, (byte)0x06,
            (byte)0xe0, (byte)0x7b, (byte)0x6b, (byte)0x87, (byte)0xe8, (byte)0xdf, (byte)0x21, (byte)0xd0,
            (byte)0xea, (byte)0x00, (byte)0x03, (byte)0x3d, (byte)0xe0, (byte)0x39, (byte)0x84, (byte)0xd3,
            (byte)0x49, (byte)0x18 };

        testHKDF("HKDFSHA1", ikm, salt, info, prk, okm);
    }
    
    private static void testHKDF(String algorithm, byte[] ikm, byte[] salt, byte[] info, byte[] prk, byte[] okm) throws Exception
    {
        HKDF hkdf = HKDF.getInstance(algorithm);
        
        hkdf.beginExtractKey(salt);
        hkdf.addKeyMaterial(ikm);
        hkdf.finishExtractKey();
        
        assertEqual(prk, hkdf.pseudoRandomKey(), "Pseudo-random Key");
        
        byte[] derivedKey = hkdf.expandKey(okm.length, info);
        
        assertEqual(okm, derivedKey, "Derived key");
    }
}