CollectionEncoderTest.java

package org.test4j.json.encoder.array;

import java.io.StringWriter;
import java.util.ArrayList;
import java.util.List;

import org.test4j.json.encoder.JSONEncoder;
import org.test4j.json.encoder.beans.test.User;
import org.test4j.json.helper.JSONFeature;
import org.test4j.testng.Test4J;
import org.testng.annotations.Test;

@SuppressWarnings({ "unchecked", "rawtypes" })
@Test(groups = { "test4j", "json" })
public class CollectionEncoderTest extends Test4J {
    @Test
    public void testEncode() throws Exception {
        List<User> users = new ArrayList<User>();
        users.add(User.newInstance(12, "darui.wu"));
        users.add(null);

        JSONEncoder encoder = JSONEncoder.get(users.getClass());
        want.object(encoder).clazIs(CollectionEncoder.class);

        encoder.setFeatures(JSONFeature.UseSingleQuote, JSONFeature.UnMarkClassFlag);
        StringWriter writer = new StringWriter();
        encoder.encode(users, writer, new ArrayList<String>());

        String json = writer.toString();
        want.string(json).eqIgnoreSpace("[{id:12,name:'darui.wu',age:0,salary:0,isFemale:false},null]");
    }

    @Test
    public void testEncode_UserRef() throws Exception {
        List<User> users = new ArrayList<User>();
        User user = User.newInstance(12, "darui.wu");
        users.add(user);
        users.add(user);

        JSONEncoder encoder = JSONEncoder.get(users.getClass());
        want.object(encoder).clazIs(CollectionEncoder.class);

        encoder.setFeatures(JSONFeature.UseSingleQuote);
        StringWriter writer = new StringWriter();
        encoder.encode(users, writer, new ArrayList<String>());

        String json = writer.toString();
        want.string(json).contains("#class:'org.test4j.json.encoder.beans.test.User@").contains("#refer:@");
    }
}