github/lidarr
1
0
Fork 0
mirror of https://github.com/lidarr/lidarr.git synced 2025-07-16 10:03:51 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

IHttpController/HttpController Added (So we can Mock SABnzbd requests)

Browse source 
More unit tests for SabController
This commit is contained in:
markus101 2010-09-27 17:49:34 -07:00
parent 50f97e824e
commit a2967f4658
9 changed files with 407 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

36
NzbDrone.Core/Controllers/SabController.cs
View file

@ -1,5 +1,6 @@
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Net;
using System.Text;
@ -15,11 +16,13 @@ namespace NzbDrone.Core.Controllers
{
private readonly IConfigController _config;
private readonly ILog _logger;
private readonly IHttpController _http;
public SabController(IConfigController config, ILog logger)
public SabController(IConfigController config, ILog logger, IHttpController http)
{
_config = config;
_logger = logger;
_http = http;
}
#region IDownloadClientController Members
@ -28,15 +31,19 @@ namespace NzbDrone.Core.Controllers
{
const string mode = "addurl";
const string cat = "tv";
string priority = _config.GetValue("Priority", String.Empty, false);
string name = nzb.Link.ToString().Replace("&", "%26");
string nzbName = HttpUtility.UrlEncode(nzb.Title);
string action = string.Format("mode={0}&name={1}&cat={2}&nzbname={3}", mode, name, cat, nzbName);
string action = string.Format("mode={0}&name={1}&priority={2}&cat={3}&nzbname={4}", mode, name, priority, cat, nzbName);
string request = GetSabRequest(action);
_logger.DebugFormat("Adding report [{0}] to the queue.", nzbName);
if (SendRequest(request) == "ok")
string response = _http.GetRequest(request).Replace("\n", String.Empty);
_logger.DebugFormat("Queue Repsonse: [{0}]", response);
if (response == "ok")
return true;
return false;
@ -45,18 +52,20 @@ namespace NzbDrone.Core.Controllers
public bool IsInQueue(Episode epsiode)
{
string action = "mode=queue&output=xml";
string request = GetSabRequest(action);
string response = _http.GetRequest(request);
XDocument xDoc = XDocument.Load(GetSabRequest(action));
XDocument xDoc = XDocument.Parse(response);
//If an Error Occurred, retuyrn
//If an Error Occurred, retuyrn)
if (xDoc.Descendants("error").Count() != 0)
return false;
if (xDoc.Descendants("queue").Count() == 0)
return false;
//Get the Count of Items in Queue where 'filename' is Equal to goodName, if not zero, return true (isInQueue)
if ((from s in xDoc.Descendants("slot") where s.Element("filename").ToString().Equals(epsiode.FeedItem.TitleFix, StringComparison.InvariantCultureIgnoreCase) select s).Count() != 0)
//Get the Count of Items in Queue where 'filename' is Equal to goodName, if not zero, return true (isInQueue)))
if ((from s in xDoc.Descendants("slot") where s.Element("filename").Value.Equals(epsiode.FeedItem.TitleFix, StringComparison.InvariantCultureIgnoreCase) select s).Count() != 0)
{
_logger.DebugFormat("Episode in queue - '{0}'", epsiode.FeedItem.TitleFix);
@ -74,19 +83,10 @@ namespace NzbDrone.Core.Controllers
string username = _config.GetValue("Username", String.Empty, false);
string password = _config.GetValue("Password", String.Empty, false);
string apiKey = _config.GetValue("ApiKey", String.Empty, false);
string priority = _config.GetValue("Priority", String.Empty, false);
return string.Format(
@"http://{0}/sabnzbd/api?$Action&priority={1}&apikey={2}&ma_username={3}&ma_password={4}",
sabnzbdInfo, priority, apiKey, username, password).Replace("$Action", action);
}
private string SendRequest(string request)
{
var webClient = new WebClient();
string response = webClient.DownloadString(request).Replace("\n", String.Empty);
_logger.DebugFormat("Queue Repsonse: [{0}]", response);
return response;
@"http://{0}/sabnzbd/api?$Action&apikey={1}&ma_username={2}&ma_password={3}",
sabnzbdInfo, apiKey, username, password).Replace("$Action", action);
}
}
}