using System ;
using System.IO ;
using System.Linq ;
using System.Text ;
using FluentAssertions ;
using Moq ;
using NUnit.Framework ;
using NzbDrone.Common ;
using NzbDrone.Core.Notifications.Plex ;
using NzbDrone.Core.Test.Framework ;
namespace NzbDrone.Core.Test.NotificationTests
{
[TestFixture]
public class PlexProviderTest : CoreTest
{
private PlexClientSettings _clientSettings ;
public void WithClientCredentials ( )
{
_clientSettings . Username = "plex" ;
_clientSettings . Password = "plex" ;
}
[SetUp]
public void Setup ( )
{
_clientSettings = new PlexClientSettings
{
Host = "localhost" ,
Port = 3000
} ;
}
[Test]
public void GetSectionKeys_should_return_single_section_key_when_only_one_show_section ( )
{
var response = "<MediaContainer size=\"1\" mediaTagPrefix=\"/system/bundle/media/flags/\" mediaTagVersion=\"1329809559\" title1=\"Plex Library\" identifier=\"com.plexapp.plugins.library\"><Directory refreshing=\"0\" key=\"5\" type=\"show\" title=\"TV Shows\" art=\"/:/resources/show-fanart.jpg\" agent=\"com.plexapp.agents.thetvdb\" scanner=\"Plex Series Scanner\" language=\"en\" updatedAt=\"1329810350\"><Location path=\"C:/Test/TV\"/></Directory></MediaContainer>" ;
Stream stream = new MemoryStream ( ASCIIEncoding . Default . GetBytes ( response ) ) ;
Mocker . GetMock < IHttpProvider > ( ) . Setup ( s = > s . DownloadStream ( "http://localhost:32400/library/sections" , null ) )
. Returns ( stream ) ;
var result = Mocker . Resolve < PlexService > ( ) . GetSectionKeys ( new PlexServerSettings { Host = "localhost" , Port = 32400 } ) ;
result . Should ( ) . HaveCount ( 1 ) ;
result . First ( ) . Should ( ) . Be ( 5 ) ;
}
[Test]
public void GetSectionKeys_should_return_single_section_key_when_only_one_show_section_with_other_sections ( )
{
var response = "<MediaContainer size=\"1\" mediaTagPrefix=\"/system/bundle/media/flags/\" mediaTagVersion=\"1329809559\" title1=\"Plex Library\" identifier=\"com.plexapp.plugins.library\"><Directory refreshing=\"0\" key=\"5\" type=\"show\" title=\"TV Shows\" art=\"/:/resources/show-fanart.jpg\" agent=\"com.plexapp.agents.thetvdb\" scanner=\"Plex Series Scanner\" language=\"en\" updatedAt=\"1329810350\"><Location path=\"C:/Test/TV\"/></Directory><Directory refreshing=\"0\" key=\"7\" type=\"movie\" title=\"TV Shows\" art=\"/:/resources/show-fanart.jpg\" agent=\"com.plexapp.agents.thetvdb\" scanner=\"Plex Series Scanner\" language=\"en\" updatedAt=\"1329810350\"><Location path=\"C:/Test/TV\"/></Directory></MediaContainer>" ;
Stream stream = new MemoryStream ( ASCIIEncoding . Default . GetBytes ( response ) ) ;
Mocker . GetMock < IHttpProvider > ( ) . Setup ( s = > s . DownloadStream ( "http://localhost:32400/library/sections" , null ) )
. Returns ( stream ) ;
var result = Mocker . Resolve < PlexService > ( ) . GetSectionKeys ( new PlexServerSettings { Host = "localhost" , Port = 32400 } ) ;
result . Should ( ) . HaveCount ( 1 ) ;
result . First ( ) . Should ( ) . Be ( 5 ) ;
}
[Test]
public void GetSectionKeys_should_return_multiple_section_keys_when_there_are_multiple_show_sections ( )
{
var response = "<MediaContainer size=\"1\" mediaTagPrefix=\"/system/bundle/media/flags/\" mediaTagVersion=\"1329809559\" title1=\"Plex Library\" identifier=\"com.plexapp.plugins.library\"><Directory refreshing=\"0\" key=\"5\" type=\"show\" title=\"TV Shows\" art=\"/:/resources/show-fanart.jpg\" agent=\"com.plexapp.agents.thetvdb\" scanner=\"Plex Series Scanner\" language=\"en\" updatedAt=\"1329810350\"><Location path=\"C:/Test/TV\"/></Directory><Directory refreshing=\"0\" key=\"6\" type=\"show\" title=\"TV Shows\" art=\"/:/resources/show-fanart.jpg\" agent=\"com.plexapp.agents.thetvdb\" scanner=\"Plex Series Scanner\" language=\"en\" updatedAt=\"1329810350\"><Location path=\"C:/Test/TV\"/></Directory></MediaContainer>" ;
Stream stream = new MemoryStream ( ASCIIEncoding . Default . GetBytes ( response ) ) ;
Mocker . GetMock < IHttpProvider > ( ) . Setup ( s = > s . DownloadStream ( "http://localhost:32400/library/sections" , null ) )
. Returns ( stream ) ;
var result = Mocker . Resolve < PlexService > ( ) . GetSectionKeys ( new PlexServerSettings { Host = "localhost" , Port = 32400 } ) ;
result . Should ( ) . HaveCount ( 2 ) ;
result . First ( ) . Should ( ) . Be ( 5 ) ;
result . Last ( ) . Should ( ) . Be ( 6 ) ;
}
[Test]
public void UpdateSection_should_update_section ( )
{
var response = "" ;
Stream stream = new MemoryStream ( ASCIIEncoding . Default . GetBytes ( response ) ) ;
Mocker . GetMock < IHttpProvider > ( ) . Setup ( s = > s . DownloadString ( "http://localhost:32400/library/sections/5/refresh" ) )
. Returns ( response ) ;
Mocker . Resolve < PlexService > ( ) . UpdateSection ( new PlexServerSettings { Host = "localhost" , Port = 32400 } , 5 ) ;
}
[Test]
public void Notify_should_send_notification ( )
{
const string header = "Test Header" ;
const string message = "Test Message" ;
var expectedUrl = String . Format ( "http://localhost:3000/xbmcCmds/xbmcHttp?command=ExecBuiltIn(Notification({0}, {1}))" , header , message ) ;
var fakeHttp = Mocker . GetMock < IHttpProvider > ( ) ;
fakeHttp . Setup ( s = > s . DownloadString ( expectedUrl ) )
. Returns ( "ok" ) ;
Mocker . Resolve < PlexService > ( ) . Notify ( _clientSettings , header , message ) ;
fakeHttp . Verify ( v = > v . DownloadString ( expectedUrl ) , Times . Once ( ) ) ;
}
[Test]
public void Notify_should_send_notification_with_credentials_when_configured ( )
{
WithClientCredentials ( ) ;
const string header = "Test Header" ;
const string message = "Test Message" ;
var expectedUrl = String . Format ( "http://localhost:3000/xbmcCmds/xbmcHttp?command=ExecBuiltIn(Notification({0}, {1}))" , header , message ) ;
var fakeHttp = Mocker . GetMock < IHttpProvider > ( ) ;
fakeHttp . Setup ( s = > s . DownloadString ( expectedUrl , "plex" , "plex" ) )
. Returns ( "ok" ) ;
Mocker . Resolve < PlexService > ( ) . Notify ( _clientSettings , header , message ) ;
fakeHttp . Verify ( v = > v . DownloadString ( expectedUrl , "plex" , "plex" ) , Times . Once ( ) ) ;
}
}
}