/*
Copyright (c) 2013 LinkedIn Corp.
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.
*/
package com.linkedin.restli.common;
import com.linkedin.data.DataMap;
import com.linkedin.data.schema.RecordDataSchema;
import com.linkedin.data.template.DataTemplateUtil;
import com.linkedin.data.template.RecordTemplate;
import org.testng.Assert;
import org.testng.annotations.Test;
public class TestComplexResourceKey
{
@Test
public void testEquals() throws CloneNotSupportedException
{
DataMap keyMap = new DataMap();
keyMap.put("keyField1", "keyValue1");
EmptyRecord key1 = new EmptyRecord(keyMap);
DataMap paramMap = new DataMap();
paramMap.put("paramField1", "paramValue1");
EmptyRecord param1 = new EmptyRecord(paramMap);
ComplexResourceKey<EmptyRecord, EmptyRecord> complexKey1 =
new ComplexResourceKey<EmptyRecord, EmptyRecord>(key1, param1);
EmptyRecord key2 = key1.copy();
EmptyRecord param2 = param1.copy();
ComplexResourceKey<EmptyRecord, EmptyRecord> complexKey2 =
new ComplexResourceKey<EmptyRecord, EmptyRecord>(key2, param2);
Assert.assertTrue(complexKey1.equals(complexKey2));
// Different key part
complexKey2.key.data().put("keyField1", "keyValue2");
Assert.assertFalse(complexKey1.equals(complexKey2));
complexKey2.key.data().put("keyField1", "keyValue1");
// Different param part
complexKey2.params.data().put("paramField1", "paramValue2");
Assert.assertFalse(complexKey1.equals(complexKey2));
complexKey2.params.data().put("paramField1", "paramValue1");
// One param null, other not
complexKey1 = new ComplexResourceKey<EmptyRecord, EmptyRecord>(key1, null);
complexKey2 = new ComplexResourceKey<EmptyRecord, EmptyRecord>(key2, param2);
Assert.assertFalse(complexKey1.equals(complexKey2));
Assert.assertFalse(complexKey2.equals(complexKey1));
// Both param null
complexKey2 = new ComplexResourceKey<EmptyRecord, EmptyRecord>(key2, null);
Assert.assertTrue(complexKey1.equals(complexKey2));
}
@Test
public void testMakeReadOnly()
{
DataMap keyDataMap = new DataMap();
keyDataMap.put("key", "key-value");
EmptyRecord key = new EmptyRecord(keyDataMap);
DataMap paramsDataMap = new DataMap();
paramsDataMap.put("params", "params-value");
EmptyRecord params = new EmptyRecord(paramsDataMap);
ComplexResourceKey<EmptyRecord, EmptyRecord> complexResourceKey =
new ComplexResourceKey<EmptyRecord, EmptyRecord>(key, params);
complexResourceKey.makeReadOnly();
try
{
key.data().put("key", "new key value");
Assert.fail("Should not be able to update the key after the ComplexResourceKey has been made read only!");
}
catch (UnsupportedOperationException e)
{
}
try
{
params.data().put("params", "new params value");
Assert.fail("Should not be able to update the params after the ComplexResourceKey has been made read only!");
}
catch (UnsupportedOperationException e)
{
}
}
@Test
public void testReadOnlyWithNullParams()
{
DataMap keyDataMap = new DataMap();
keyDataMap.put("key", "key-value");
EmptyRecord key = new EmptyRecord(keyDataMap);
ComplexResourceKey<EmptyRecord, EmptyRecord> complexResourceKey =
new ComplexResourceKey<EmptyRecord, EmptyRecord>(key, null);
complexResourceKey.makeReadOnly();
try
{
key.data().put("key", "new key value");
Assert.fail("Should not be able to update the key after the ComplexResourceKey has been made read only!");
}
catch (UnsupportedOperationException e)
{
}
}
@Test
public void testKeySchema()
{
RecordDataSchema schema = (RecordDataSchema) DataTemplateUtil.parseSchema
(
"{ \"type\" : \"record\", \"name\" : \"omni\", \"fields\" : [ { \"name\" : \"int\", \"type\" : \"int\" } ] }"
);
TypeSpec<OmniRecord> keyType = new TypeSpec<OmniRecord>(OmniRecord.class, schema);
TypeSpec<OmniRecord> paramsType = new TypeSpec<OmniRecord>(OmniRecord.class, schema);
ComplexKeySpec<OmniRecord, OmniRecord> keySpec = new ComplexKeySpec<OmniRecord, OmniRecord>(keyType, paramsType);
DataMap data = new DataMap();
data.put("int", 1);
ComplexResourceKey<RecordTemplate, RecordTemplate> key = ComplexResourceKey.buildFromDataMap(data, keySpec);
Assert.assertEquals(key.getKey().schema(), schema);
Assert.assertEquals(key.getParams().schema(), schema);
}
public static class OmniRecord extends RecordTemplate {
public OmniRecord(DataMap map, RecordDataSchema schema)
{
super(map, schema);
}
}
}