wallabag api ok

wallabag_api/wallabag_test.py

@@ -1,15 +1,42 @@
-# -*- coding: utf-8 -*-
+# coding: utf-8
 import unittest
 from wallabag import Wallabag
 
 
 class TestWallabag(unittest.TestCase):
 
+    host = 'http://localhost:8080'
+    client_id = ''
+    client_secret = ''
+    token = ''
+
     def setUp(self):
-        self.host = "http://localhost:5000"
-        self.api_key = '12334567890'
-        self.user_agent = 'WallabagPython/1.0 +https://github.com/foxmask/wallabag-api'
+        access_token = self.test_get_token()
         self.format = 'json'
+        self.w = Wallabag(host=self.host,
+                          token=access_token,
+                          client_id=self.client_id,
+                          client_secret=self.client_secret)
+
+    def test_get_token(self):
+        params = {"grant_type": "password",
+                  "client_id": '1_37e16ub8a62oc4gwcg0o0wssks800kw0ok408kkwo4kosgc88g',
+                  "client_secret": '49etxpn66da8okg4cs40sswsog0sskwg4woc400c4w4w8s4wo4',
+                  "username": 'foxmask',
+                  "password": 'ratatab00m'}
+
+        data = Wallabag.get_token(host='http://localhost:8080', **params)
+        self.assertTrue(isinstance(data, str), True)
+        return data
+
+    def create_entry(self):
+        title = 'foobar title'
+        url = 'https://smcomm.trigger-happy.eu/'
+        tags = ['foo', 'bar']
+        starred = 0
+        archive = 0
+        data = self.w.post_entries(url, title, tags, starred, archive)
+        return data
 
     def test_get_entries(self):
         params = {'archive': 0,
@@ -20,79 +47,64 @@ class TestWallabag(unittest.TestCase):
                   'page': 1,
                   'perPage': 30,
                   'tags': []}
-        w = Wallabag(self.host).get_entries('ABCD', **params)
-        self.assertIsInstance(w, dict)
-
-    def test_post_entries(self):
-        title = 'foobar title'
-        url = ['http://foobar.com/', 'http://barfoo.com/']
-        tags = ['foo', 'bar']
-        self.assertTrue(isinstance(title, str), True)
-        self.assertTrue(isinstance(tags, list), True)
-        w = Wallabag(self.host).post_entries('ABCD', url, title, tags)
-        self.assertTrue(w, True)
+        data = self.w.get_entries(**params)
+        self.assertIsInstance(data, dict)
 
     def test_get_entry(self):
         entry = 1
         self.assertTrue(isinstance(entry, int), True)
-        w = Wallabag(self.host).get_entry('ABCD', entry)
-        self.assertTrue(w, str)
-
-    def test_patch_entries(self):
-        entry = 1
-        params = {'title': '',
-                  'archive': 0,
-                  'tags': [],
-                  'star': 0,
-                  'delete': 0}
-        self.assertTrue(isinstance(entry, int), True)
-        self.assertTrue(isinstance(params, dict), True)
-        w = Wallabag(self.host).patch_entries('ABCD', entry, **params)
-        self.assertTrue(w, True)
-
-    def test_delete_entries(self):
-        entry = 1
-        self.assertTrue(isinstance(entry, int), True)
-        w = Wallabag(self.host).delete_entries('ABCD', entry)
-        self.assertTrue(w, True)
+        data = self.w.get_entry(entry)
+        self.assertTrue(data, str)
 
     def test_get_entry_tags(self):
         entry = 1
         self.assertTrue(isinstance(entry, int), True)
-        w = Wallabag(self.host).get_entry_tags('ABCD', entry)
-        self.assertIsInstance(w, list)
+        data = self.w.get_entry_tags(entry)
+        self.assertIsInstance(data, list)
+
+    def test_get_tags(self):
+        data = self.w.get_tags()
+        self.assertIsInstance(data, list)
+
+    def test_post_entries(self):
+        data = self.create_entry()
+        self.assertTrue(data, True)
+
+    def test_patch_entries(self):
+        entry = 1
+        params = {'title': 'I change the title',
+                  'archive': 0,
+                  'tags': ["bimbo", "pipo"],
+                  'star': 0,
+                  'delete': 0}
+        self.assertTrue(isinstance(entry, int), True)
+        self.assertTrue(isinstance(params, dict), True)
+        data = self.w.patch_entries(entry, **params)
+        self.assertTrue(data, True)
+
+    def test_delete_entries(self):
+        entry = self.create_entry()
+        self.assertTrue(isinstance(entry['id'], int), True)
+        data = self.w.delete_entries(entry['id'])
+        self.assertTrue(data, True)
 
     def test_post_entry_tags(self):
         entry = 1
         self.assertTrue(isinstance(entry, int), True)
         tags = ['foo', 'bar']
         self.assertTrue(isinstance(tags, list), True)
-        w = Wallabag(self.host).post_entry_tags('ABCD', entry, tags)
-        self.assertTrue(w, True)
+        data = self.w.post_entry_tags(entry, tags)
+        self.assertTrue(data, True)
 
+    """
     def test_delete_entry_tag(self):
-        entry = 1
-        tag = 'tag1'
-        self.assertTrue(isinstance(entry, int), True)
+        entry = self.create_entry()
+        tag = 'bar'
+        self.assertTrue(isinstance(entry['id'], int), True)
         self.assertTrue(isinstance(tag, str), True)
-        w = Wallabag(self.host).delete_entry_tag('ABCD', entry, tag)
-        self.assertTrue(w, True)
-
-    def test_get_tags(self):
-        w = Wallabag(self.host).get_tags('ABCD')
-        self.assertTrue(w, True)
-
-    def test_get_tag(self):
-        tag = 'tag1'
-        self.assertTrue(isinstance(tag, str), True)
-        w = Wallabag(self.host).get_tag('ABCD', tag)
-        self.assertTrue(w, True)
-
-    def test_delete_tag(self):
-        tag = 'tag1'
-        self.assertTrue(isinstance(tag, str), True)
-        w = Wallabag(self.host).delete_tag('ABCD', tag)
-        self.assertTrue(w, True)
+        resp = self.w.delete_entry_tag(entry['id'], tag)
+        self.assertTrue(resp, True)
+    """
 
 if __name__ == '__main__':
-    unittest.main()
+    unittest.main()