jiffy/test/jiffy_18_preencode_tests.erl

-module(jiffy_18_preencode_tests).


-include_lib("eunit/include/eunit.hrl").
-include("jiffy_util.hrl").


preencode_success_test_() ->
    [gen(ok, Case) || Case <- cases(ok)].


%% preencode_failure_test_() ->
%%     [gen(error, Case) || Case <- cases(error)].


gen(ok, {E1, J, E2}) ->
    {msg("~p", [E1]), [
        {"Encode", ?_assertEqual(J, enc(E1))},
        {"Decode", ?_assertEqual(E2, dec(J))}
    ]};

gen(ok, {E, J}) ->
    {msg("~p", [E]), [
        {"Encode", ?_assertEqual(J, enc(E))}
    ]}.

%% gen(error, E) ->
%%     {msg("Error: ~p", [E]), [
%%         ?_assertThrow({error, _}, enc(E))
%%     ]}.


cases(ok) ->
    TopTests =
        lists:map(
          fun (EJSON) ->
                  JSON = enc(EJSON),
                  {{json, JSON}, JSON, EJSON}
          end, [ 123
               , <<"hello world">>
               , true
               , false
               , {[ {<<"a">>, <<"apple">>}, {<<"b">>, <<"banana">>} ]}
               ]),
    EJSON = [ 1, <<"a">> ],
    JSON = enc(EJSON),
    BuriedTests =
        [ { [ {json, JSON} ], <<"[[1,\"a\"]]">>, [ EJSON ]}
        , { [ 1, {json, JSON}, 3 ], <<"[1,[1,\"a\"],3]">>, [ 1, EJSON, 3 ]}
        , { [ {json, JSON}, {json, JSON} ], <<"[[1,\"a\"],[1,\"a\"]]">>, [ EJSON, EJSON ]}
        , { {[ {<<"a">>, {json, JSON}} ]}, <<"{\"a\":[1,\"a\"]}">>, {[ {<<"a">>, EJSON} ]}}
        ],

    PartialArray1 = jiffy:partial_encode([ 2, 3 ], []),
    PartialArray2 = jiffy:partial_encode([], []),
    PartialArray3 = jiffy:partial_encode([ 5 ], []),
    PartialArrayTests =
        [ {[ PartialArray1 ], <<"[2,3]">>}
        , {[ 1, PartialArray1 ], <<"[1,2,3]">>}
        , {[ PartialArray1, 4 ], <<"[2,3,4]">>}
        , {[ 1, PartialArray1, 4 ], <<"[1,2,3,4]">>}
        , {[ PartialArray2 ], <<"[]">>}
        , {[ 1, PartialArray2 ], <<"[1]">>}
        , {[ PartialArray2, 4 ], <<"[4]">>}
        , {[ 1, PartialArray2, 4 ], <<"[1,4]">>}
        , {[ PartialArray1, PartialArray2 ], <<"[2,3]">>}
        , {[ PartialArray2, PartialArray1 ], <<"[2,3]">>}
        , {[ PartialArray1, PartialArray1 ], <<"[2,3,2,3]">>}
        , {[ PartialArray2, PartialArray2 ], <<"[]">>}
        , {[ PartialArray1, PartialArray3 ], <<"[2,3,5]">>}
        , {[ 1, PartialArray1, 4, PartialArray3, 6 ], <<"[1,2,3,4,5,6]">>}
        ],

    PartialObject1 = jiffy:partial_encode({[ {<<"ii">>, <<"two">>}, {<<"iii">>, 3} ]}, []),
    PartialObject2 = jiffy:partial_encode({[]}, []),
    PartialObject3 = jiffy:partial_encode({[ {<<"v">>, [ 1, 2, 3, 4, 5 ]} ]}, []),
    PartialObjectTests =
        [ {{[ PartialObject1 ]}, <<"{\"ii\":\"two\",\"iii\":3}">>}
        , {{[ {<<"i">>, 1}, PartialObject1 ]}, <<"{\"i\":1,\"ii\":\"two\",\"iii\":3}">>}
        , {{[ PartialObject1, {<<"iv">>, 4} ]}, <<"{\"ii\":\"two\",\"iii\":3,\"iv\":4}">>}
        , {{[ {<<"i">>, 1}, PartialObject1, {<<"iv">>, 4} ]}, <<"{\"i\":1,\"ii\":\"two\",\"iii\":3,\"iv\":4}">>}
        , {{[ PartialObject2 ]}, <<"{}">>}
        , {{[ {<<"i">>, 1}, PartialObject2 ]}, <<"{\"i\":1}">>}
        , {{[ PartialObject2, {<<"iv">>, 4} ]}, <<"{\"iv\":4}">>}
        , {{[ {<<"i">>, 1}, PartialObject2, {<<"iv">>, 4} ]}, <<"{\"i\":1,\"iv\":4}">>}
        , {{[ PartialObject1, PartialObject2 ]}, <<"{\"ii\":\"two\",\"iii\":3}">>}
        , {{[ PartialObject2, PartialObject1 ]}, <<"{\"ii\":\"two\",\"iii\":3}">>}
        , {{[ PartialObject2, PartialObject2 ]}, <<"{}">>}
        , {{[ PartialObject1, PartialObject3 ]}, <<"{\"ii\":\"two\",\"iii\":3,\"v\":[1,2,3,4,5]}">>}
        ],

    TopTests ++ BuriedTests ++ PartialArrayTests ++ PartialObjectTests.

%% cases(error) ->
%%     [ {json, true}
%%     , {json, "true"}
%%     ].