mediagoblin/tests/test_piwigo.py

   1 # GNU MediaGoblin -- federated, autonomous media hosting
   2 # Copyright (C) 2013 MediaGoblin contributors.  See AUTHORS.
   3 #
   4 # This program is free software: you can redistribute it and/or modify
   5 # it under the terms of the GNU Affero General Public License as published by
   6 # the Free Software Foundation, either version 3 of the License, or
   7 # (at your option) any later version.
   8 #
   9 # This program is distributed in the hope that it will be useful,
  10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 # GNU Affero General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU Affero General Public License
  15 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16
  17 import pytest
  18 from .tools import fixture_add_user
  19
  20
  21 XML_PREFIX = "<?xml version='1.0' encoding='utf-8'?>\n"
  22
  23
  24 class Test_PWG(object):
  25     @pytest.fixture(autouse=True)
  26     def setup(self, test_app):
  27         self.test_app = test_app
  28
  29         fixture_add_user()
  30
  31         self.username = u"chris"
  32         self.password = "toast"
  33
  34     def do_post(self, method, params):
  35         params["method"] = method
  36         return self.test_app.post("/api/piwigo/ws.php", params)
  37
  38     def do_get(self, method, params=None):
  39         if params is None:
  40             params = {}
  41         params["method"] = method
  42         return self.test_app.get("/api/piwigo/ws.php", params)
  43
  44     def test_session(self):
  45         resp = self.do_post("pwg.session.login",
  46             {"username": u"nouser", "password": "wrong"})
  47         assert resp.body == XML_PREFIX \
  48             + '<rsp stat="fail"><err code="999" msg="Invalid username/password"/></rsp>'
  49
  50         resp = self.do_post("pwg.session.login",
  51             {"username": self.username, "password": "wrong"})
  52         assert resp.body == XML_PREFIX \
  53             + '<rsp stat="fail"><err code="999" msg="Invalid username/password"/></rsp>'
  54
  55         resp = self.do_get("pwg.session.getStatus")
  56         assert resp.body == XML_PREFIX \
  57             + '<rsp stat="ok"><username>guest</username></rsp>'
  58
  59         resp = self.do_post("pwg.session.login",
  60             {"username": self.username, "password": self.password})
  61         assert resp.body == XML_PREFIX + '<rsp stat="ok">1</rsp>'
  62
  63         resp = self.do_get("pwg.session.getStatus")
  64         assert resp.body == XML_PREFIX \
  65             + '<rsp stat="ok"><username>chris</username></rsp>'
  66
  67         self.do_get("pwg.session.logout")
  68
  69         resp = self.do_get("pwg.session.getStatus")
  70         assert resp.body == XML_PREFIX \
  71             + '<rsp stat="ok"><username>guest</username></rsp>'