mono/mcs/class/System.Core/Test/System.Linq/LookupTest.cs

//
// LookupTest.cs
//
// Author:
//   Jb Evain ([email protected])
//
// (C) 2009 Novell, Inc. (http://www.novell.com)
//
// Permission is hereby granted, free of charge, to any person obtaining
// a copy of this software and associated documentation files (the
// "Software"), to deal in the Software without restriction, including
// without limitation the rights to use, copy, modify, merge, publish,
// distribute, sublicense, and/or sell copies of the Software, and to
// permit persons to whom the Software is furnished to do so, subject to
// the following conditions:
//
// The above copyright notice and this permission notice shall be
// included in all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
// LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
// OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
// WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
//

using System;
using System.Collections.Generic;
using System.Linq;

using NUnit.Framework;

namespace MonoTests.System.Linq {

	[TestFixture]
	public class LookupTest {

		class Color {

			public string Name { get; set; }
			public int Value { get; set; }

			public Color (string name, int value)
			{
				Name = name;
				Value = value;
			}
		}

		static IEnumerable<Color> GetColors ()
		{
			yield return new Color ("Red", 0xff0000);
			yield return new Color ("Green", 0x00ff00);
			yield return new Color ("Blue", 0x0000ff);
		}

		[Test]
		public void LookupIgnoreCase ()
		{
			var lookup = GetColors ().ToLookup (
				c => c.Name,
				c => c.Value,
				StringComparer.OrdinalIgnoreCase);

			Assert.AreEqual (0xff0000, lookup ["red"].First ());
			Assert.AreEqual (0x00ff00, lookup ["GrEeN"].First ());
			Assert.AreEqual (0x0000ff, lookup ["Blue"].First ());
		}
		
		[Test]
		public void LookupContains()
		{
			var lookup = new [] { "hi", "bye" }.ToLookup (c => c [0].ToString ());
			
			Assert.IsTrue (lookup.Contains ("h"));
			Assert.IsFalse (lookup.Contains ("d"));
			Assert.IsFalse (lookup.Contains (null));
		}
		
		[Test]
		public void LookupContainsNull()
		{
			var lookup = new [] { "hi", "bye", "42" }.ToLookup (c => (Char.IsNumber (c [0]) ? null : c [0].ToString ()));
			
			Assert.IsTrue (lookup.Contains ("h"));
			Assert.IsTrue (lookup.Contains (null));
			Assert.IsFalse (lookup.Contains ("d"));
		}
		
		[Test]
		public void LookupEnumeratorWithoutNull()
		{
			var lookup = new [] { "hi", "bye" }.ToLookup (c => c [0].ToString ());
			
			Assert.IsTrue (lookup.Any (g => g.Key == "h"));
			Assert.IsTrue (lookup.Any (g => g.Key == "b"));
			Assert.IsFalse (lookup.Any (g => g.Key == null));
		}
		
		[Test]
		public void LookupEnumeratorWithNull()
		{
			var lookup = new [] { "hi", "bye", "42" }.ToLookup (c => (Char.IsNumber (c [0]) ? null : c [0].ToString ()));
			
			Assert.IsTrue (lookup.Any (g => g.Key == "h"));
			Assert.IsTrue (lookup.Any (g => g.Key == "b"));
			Assert.IsTrue (lookup.Any (g => g.Key == null));
		}
		
		[Test]
		public void LookupNullKeyNone()
		{
			var lookup = new [] { "hi", "bye" }.ToLookup (c => c [0].ToString ());
			
			Assert.AreEqual (2, lookup.Count);
			Assert.AreEqual (0, lookup [null].Count ());
		}

		[Test]
		public void EmptyResult ()
		{
			var lookup = GetColors ().ToLookup (
				c => c.Name,
				c => c.Value,
				StringComparer.OrdinalIgnoreCase);

			var l = lookup ["notexist"];
			Assert.IsNotNull (l);
			int [] values = (int []) l;
			Assert.AreEqual (values.Length, 0);
		}
	}
}