| % 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. |
| |
| -module(chttpd_db_doc_size_tests). |
| |
| -include_lib("couch/include/couch_eunit.hrl"). |
| -include_lib("couch/include/couch_db.hrl"). |
| |
| -define(USER, "chttpd_db_test_admin"). |
| -define(PASS, "pass"). |
| -define(AUTH, {basic_auth, {?USER, ?PASS}}). |
| -define(CONTENT_JSON, {"Content-Type", "application/json"}). |
| -define(CONTENT_MULTI_RELATED, {"Content-Type", "multipart/related;boundary=\"bound\""}). |
| -define(CONTENT_MULTI_FORM, {"Content-Type", "multipart/form-data;boundary=\"bound\""}). |
| |
| setup() -> |
| Hashed = couch_passwords:hash_admin_password(?PASS), |
| ok = config:set("admins", ?USER, ?b2l(Hashed), _Persist = false), |
| ok = config:set("couchdb", "max_document_size", "50"), |
| TmpDb = ?tempdb(), |
| Addr = config:get("chttpd", "bind_address", "127.0.0.1"), |
| Port = mochiweb_socket_server:get(chttpd, port), |
| Url = lists:concat(["http://", Addr, ":", Port, "/", ?b2l(TmpDb)]), |
| create_db(Url), |
| Url. |
| |
| teardown(Url) -> |
| delete_db(Url), |
| ok = config:delete("admins", ?USER, _Persist = false), |
| ok = config:delete("couchdb", "max_document_size"). |
| |
| create_db(Url) -> |
| {ok, Status, _, _} = test_request:put(Url, [?CONTENT_JSON, ?AUTH], "{}"), |
| case Status of |
| 201 -> ok; |
| 202 -> ok; |
| _ -> io:format(user, "~n HTTP Status Code: ~p~n", [Status]) |
| end, |
| ?assert(Status =:= 201 orelse Status =:= 202). |
| |
| delete_db(Url) -> |
| {ok, 200, _, _} = test_request:delete(Url, [?AUTH]). |
| |
| all_test_() -> |
| { |
| "chttpd db max_document_size tests", |
| { |
| setup, |
| fun chttpd_test_util:start_couch/0, |
| fun chttpd_test_util:stop_couch/1, |
| { |
| foreach, |
| fun setup/0, |
| fun teardown/1, |
| [ |
| fun post_single_doc/1, |
| fun put_single_doc/1, |
| fun bulk_doc/1, |
| fun put_post_doc_attach_inline/1, |
| fun put_multi_part_related/1, |
| fun post_multi_part_form/1 |
| ] |
| } |
| } |
| }. |
| |
| post_single_doc(Url) -> |
| NewDoc = |
| "{\"post_single_doc\": \"some_doc\",\n" |
| " \"_id\": \"testdoc\", \"should_be\" : \"too_large\"}", |
| {ok, _, _, ResultBody} = test_request:post( |
| Url, |
| [?CONTENT_JSON, ?AUTH], |
| NewDoc |
| ), |
| {[ErrorMsg | _]} = ?JSON_DECODE(ResultBody), |
| ?_assertEqual({<<"error">>, <<"document_too_large">>}, ErrorMsg). |
| |
| put_single_doc(Url) -> |
| NewDoc = |
| "{\"post_single_doc\": \"some_doc\",\n" |
| " \"_id\": \"testdoc\", \"should_be\" : \"too_large\"}", |
| {ok, _, _, ResultBody} = test_request:put( |
| Url ++ "/" ++ "testid", |
| [?CONTENT_JSON, ?AUTH], |
| NewDoc |
| ), |
| {[ErrorMsg | _]} = ?JSON_DECODE(ResultBody), |
| ?_assertEqual({<<"error">>, <<"document_too_large">>}, ErrorMsg). |
| |
| bulk_doc(Url) -> |
| NewDoc = |
| "{\"docs\": [{\"doc1\": 1}, {\"errordoc\":\n" |
| " \"this_should_be_the_too_large_error_document\"}]}", |
| {ok, _, _, ResultBody} = test_request:post( |
| Url ++ "/_bulk_docs/", |
| [?CONTENT_JSON, ?AUTH], |
| NewDoc |
| ), |
| ResultJson = ?JSON_DECODE(ResultBody), |
| Expect = {[{<<"error">>, <<"document_too_large">>}, {<<"reason">>, <<>>}]}, |
| ?_assertEqual(Expect, ResultJson). |
| |
| put_post_doc_attach_inline(Url) -> |
| Body1 = "{\"body\":\"This is a body.\",", |
| Body2 = lists:concat([ |
| "{\"body\":\"This is a body it should fail", |
| "because there are too many characters.\"," |
| ]), |
| DocRest = lists:concat([ |
| "\"_attachments\":{\"foo.txt\":{", |
| "\"content_type\":\"text/plain\",", |
| "\"data\": \"VGhpcyBpcyBhIGJhc2U2NCBlbmNvZGVkIHRleHQ=\"}}}" |
| ]), |
| Doc1 = lists:concat([Body1, DocRest]), |
| Doc2 = lists:concat([Body2, DocRest]), |
| |
| {ok, _, _, ResultBody} = test_request:post( |
| Url, |
| [?CONTENT_JSON, ?AUTH], |
| Doc1 |
| ), |
| {[Msg | _]} = ?JSON_DECODE(ResultBody), |
| {ok, _, _, ResultBody1} = test_request:post( |
| Url, |
| [?CONTENT_JSON, ?AUTH], |
| Doc2 |
| ), |
| {[Msg1 | _]} = ?JSON_DECODE(ResultBody1), |
| |
| {ok, _, _, ResultBody2} = test_request:put( |
| Url ++ "/" ++ "accept", |
| [?CONTENT_JSON, ?AUTH], |
| Doc1 |
| ), |
| {[Msg2 | _]} = ?JSON_DECODE(ResultBody2), |
| {ok, _, _, ResultBody3} = test_request:put( |
| Url ++ "/" ++ "fail", |
| [?CONTENT_JSON, ?AUTH], |
| Doc2 |
| ), |
| {[Msg3 | _]} = ?JSON_DECODE(ResultBody3), |
| [ |
| ?_assertEqual({<<"ok">>, true}, Msg), |
| ?_assertEqual({<<"error">>, <<"document_too_large">>}, Msg1), |
| ?_assertEqual({<<"ok">>, true}, Msg2), |
| ?_assertEqual({<<"error">>, <<"document_too_large">>}, Msg3) |
| ]. |
| |
| put_multi_part_related(Url) -> |
| Body1 = "{\"body\":\"This is a body.\",", |
| Body2 = lists:concat([ |
| "{\"body\":\"This is a body it should fail", |
| "because there are too many characters.\"," |
| ]), |
| DocBeg = "--bound\r\nContent-Type: application/json\r\n\r\n", |
| DocRest = lists:concat([ |
| "\"_attachments\":{\"foo.txt\":{\"follows\":true,", |
| "\"content_type\":\"text/plain\",\"length\":21},\"bar.txt\":", |
| "{\"follows\":true,\"content_type\":\"text/plain\",", |
| "\"length\":20}}}\r\n--bound\r\n\r\nthis is 21 chars long", |
| "\r\n--bound\r\n\r\nthis is 20 chars lon\r\n--bound--epilogue" |
| ]), |
| Doc1 = lists:concat([DocBeg, Body1, DocRest]), |
| Doc2 = lists:concat([DocBeg, Body2, DocRest]), |
| {ok, _, _, ResultBody} = test_request:put( |
| Url ++ "/" ++ "accept", |
| [?CONTENT_MULTI_RELATED, ?AUTH], |
| Doc1 |
| ), |
| {[Msg | _]} = ?JSON_DECODE(ResultBody), |
| {ok, _, _, ResultBody1} = test_request:put( |
| Url ++ "/" ++ "faildoc", |
| [?CONTENT_MULTI_RELATED, ?AUTH], |
| Doc2 |
| ), |
| {[Msg1 | _]} = ?JSON_DECODE(ResultBody1), |
| [ |
| ?_assertEqual({<<"ok">>, true}, Msg), |
| ?_assertEqual({<<"error">>, <<"document_too_large">>}, Msg1) |
| ]. |
| |
| post_multi_part_form(Url) -> |
| Port = mochiweb_socket_server:get(chttpd, port), |
| Host = lists:concat(["http://127.0.0.1:", Port]), |
| Referer = {"Referer", Host}, |
| Body1 = "{\"body\":\"This is a body.\"}", |
| Body2 = lists:concat([ |
| "{\"body\":\"This is a body it should fail", |
| "because there are too many characters.\"}" |
| ]), |
| DocBeg = "--bound\r\nContent-Disposition: form-data; name=\"_doc\"\r\n\r\n", |
| DocRest = lists:concat([ |
| "\r\n--bound\r\nContent-Disposition:", |
| "form-data; name=\"_attachments\"; filename=\"file.txt\"\r\n", |
| "Content-Type: text/plain\r\n\r\ncontents of file.txt\r\n\r\n", |
| "--bound--" |
| ]), |
| Doc1 = lists:concat([DocBeg, Body1, DocRest]), |
| Doc2 = lists:concat([DocBeg, Body2, DocRest]), |
| {ok, _, _, ResultBody} = test_request:post( |
| Url ++ "/" ++ "accept", |
| [?CONTENT_MULTI_FORM, ?AUTH, Referer], |
| Doc1 |
| ), |
| {[Msg | _]} = ?JSON_DECODE(ResultBody), |
| {ok, _, _, ResultBody1} = test_request:post( |
| Url ++ "/" ++ "fail", |
| [?CONTENT_MULTI_FORM, ?AUTH, Referer], |
| Doc2 |
| ), |
| {[Msg1 | _]} = ?JSON_DECODE(ResultBody1), |
| [ |
| ?_assertEqual({<<"ok">>, true}, Msg), |
| ?_assertEqual({<<"error">>, <<"document_too_large">>}, Msg1) |
| ]. |