Merge pull request #3 from TechConsult/favorite-storage-operations

favorite storage operations

Author: Faustasm

test/test_storage.py

@@ -92,6 +92,18 @@ def test_templatize_storage(self, manager):
         assert storage.title == "my server template"
         assert storage.type == "template"
 
+    @responses.activate
+    def test_add_storage_to_favorites(self, manager):
+        data = Mock.mock_post("storage/01d4fcd4-e446-433b-8a9c-551a1284952e/favorite", empty_content=True)
+        res = manager.add_storage_to_favorites("01d4fcd4-e446-433b-8a9c-551a1284952e")
+        assert res == {}
+
+    @responses.activate
+    def test_remove_storage_from_favorites(self, manager):
+        data = Mock.mock_delete("storage/01d4fcd4-e446-433b-8a9c-551a1284952e/favorite")
+        res = manager.remove_storage_from_favorites("01d4fcd4-e446-433b-8a9c-551a1284952e")
+        assert res == {}
+
     @responses.activate
     def test_storage_update(self, manager):
 

upcloud_api/cloud_manager/storage_mixin.py

@@ -150,3 +150,17 @@ def templatize_storage(self, storage, title):
         body = {'storage': {'title': title}}
         res = self.post_request(url, body)
         return Storage(cloud_manager=self, **res['storage'])
+
+    def add_storage_to_favorites(self, storage):
+        """
+        Adds a storage to the list of favorite storages.
+        """
+        url = '/storage/{0}/favorite'.format(storage)
+        return self.post_request(url)
+
+    def remove_storage_from_favorites(self, storage):
+        """
+        Removes a storage from the list of favorite storages.
+        """
+        url = '/storage/{0}/favorite'.format(storage)
+        return self.request('DELETE', url)