Add set_flag method

1 files changed, 31 insertions, 0 deletions

diff --git a/test/test_api.py b/test/test_api.py
index 5a46fb9..3fbf349 100644
--- a/test/test_api.py
+++ b/test/test_api.py
@@ -4,6 +4,12 @@ import os
 from . import test_dir
 from obsapi import ObsApi
 from lxml import etree
+try:
+    from urllib.parse import urlparse
+    from urllib.parse import parse_qs
+except Exception:
+    from urlparse import urlparse
+    from urlparse import parse_qs
 
 apiurl = 'https://api.opensuse.org'
 api = ObsApi(apiurl=apiurl)
@@ -514,3 +520,28 @@ def test_get_project_spec_files(requests_mock, project_directory, package_direct
     expected_specfiles = [spec_file_1, spec_file_2, spec_file_3]
     specfiles = api.get_spec_files(prj)
     assert list(specfiles) == expected_specfiles
+
+
+# POST /source/<project>?cmd=set_flag&repository=:opt&arch=:opt&flag=flag&status=status
+def test_set_project_flags(requests_mock):
+    matcher = re.compile(r'/source/.*')
+    requests_mock.post(matcher)
+    params = dict(cmd='set_flag', flag='publish', status='enable')
+
+    response = api.set_flag(prj, flag=params)
+    assert response.status_code == 200
+
+    url = requests_mock.request_history[0].url
+    query = parse_qs(urlparse(url).query)
+    assert query.keys() == params.keys()
+    for key, values in query.items():
+        assert len(values) == 1
+        assert params.get(key, None) == query.get(key, [])[0]
+
+
+# POST /source/<project>/<package>?cmd=set_flag&repository=:opt&arch=:opt&flag=flag&status=status
+def test_set_package_flags(requests_mock):
+    pass
+
+# POST /source/<project>/<package>?cmd=remove_flag&repository=:opt&arch=:opt&flag=flag
+# POST /source/<project>?cmd=remove_flag&repository=:opt&arch=:opt&flag=flag