Send a item-not-found error when the “after” value is not in the archive

This commit is contained in:
louiz’
2018-02-11 23:29:58 +01:00
parent 0280343ced
commit d705541435
3 changed files with 11 additions and 3 deletions
+1
View File
@@ -22,6 +22,7 @@ class Database
{ {
public: public:
using time_point = std::chrono::system_clock::time_point; using time_point = std::chrono::system_clock::time_point;
struct RecordNotFound: public std::exception {};
struct Uuid: Column<std::string> { static constexpr auto name = "uuid_"; }; struct Uuid: Column<std::string> { static constexpr auto name = "uuid_"; };
+6 -2
View File
@@ -467,8 +467,12 @@ void BiboumiComponent::handle_iq(const Stanza& stanza)
#ifdef USE_DATABASE #ifdef USE_DATABASE
else if ((query = stanza.get_child("query", MAM_NS))) else if ((query = stanza.get_child("query", MAM_NS)))
{ {
if (this->handle_mam_request(stanza)) try {
stanza_error.disable(); if (this->handle_mam_request(stanza))
stanza_error.disable();
} catch (const Database::RecordNotFound& exc) {
error_name = "item-not-found";
}
} }
else if ((query = stanza.get_child("query", MUC_OWNER_NS))) else if ((query = stanza.get_child("query", MUC_OWNER_NS)))
{ {
+4 -1
View File
@@ -2182,12 +2182,15 @@ if __name__ == '__main__':
"/message/mam:result[@queryid='qid2']/forward:forwarded/client:message[@from='#foo%{irc_server_one}/{nick_one}'][@type='groupchat']/client:body[text()='149']"), "/message/mam:result[@queryid='qid2']/forward:forwarded/client:message[@from='#foo%{irc_server_one}/{nick_one}'][@type='groupchat']/client:body[text()='149']"),
after = partial(save_value, "last_uuid", partial(extract_attribute, "/message/mam:result", "id")) after = partial(save_value, "last_uuid", partial(extract_attribute, "/message/mam:result", "id"))
), ),
# And it should not be marked as complete
partial(expect_stanza, partial(expect_stanza,
("/iq[@type='result'][@id='id2'][@from='#foo%{irc_server_one}'][@to='{jid_one}/{resource_one}']", ("/iq[@type='result'][@id='id2'][@from='#foo%{irc_server_one}'][@to='{jid_one}/{resource_one}']",
"/iq/mam:fin/rsm:set/rsm:last[text()='{last_uuid}']", "/iq/mam:fin/rsm:set/rsm:last[text()='{last_uuid}']",
"/iq//mam:fin[@complete='true']", "/iq//mam:fin[@complete='true']",
"/iq//mam:fin")), "/iq//mam:fin")),
# Send a request with a non-existing ID set as the “after” value.
partial(send_stanza, "<iq to='#foo%{irc_server_one}' from='{jid_one}/{resource_one}' type='set' id='id3'><query xmlns='urn:xmpp:mam:2' queryid='qid3' ><set xmlns='http://jabber.org/protocol/rsm'><after>DUMMY_ID</after></set></query></iq>"),
partial(expect_stanza, "/iq[@id='id3'][@type='error']/error[@type='cancel']/stanza:feature-not-implemented")
]), ]),
Scenario("channel_history_on_fixed_server", Scenario("channel_history_on_fixed_server",
[ [