Implemented equality for shortcut objects.

Shortcuts are equal if and only if their key codes and modifier flags are.

1 files changed, 14 insertions, 0 deletions

diff --git a/Framework/MASShortcutTests.m b/Framework/MASShortcutTests.m
index 37a801f..d1cd8c4 100644
--- a/Framework/MASShortcutTests.m
+++ b/Framework/MASShortcutTests.m
@@ -5,6 +5,20 @@
 
 @implementation MASShortcutTests
 
+- (void) testEquality
+{
+    MASShortcut *keyA = [MASShortcut shortcutWithKeyCode:1 modifierFlags:NSControlKeyMask];
+    MASShortcut *keyB = [MASShortcut shortcutWithKeyCode:2 modifierFlags:NSControlKeyMask];
+    MASShortcut *keyC = [MASShortcut shortcutWithKeyCode:1 modifierFlags:NSAlternateKeyMask];
+    MASShortcut *keyD = [MASShortcut shortcutWithKeyCode:1 modifierFlags:NSControlKeyMask];
+    XCTAssertTrue([keyA isEqual:keyA], @"Shortcut is equal to itself.");
+    XCTAssertTrue([keyA isEqual:[keyA copy]], @"Shortcut is equal to its copy.");
+    XCTAssertFalse([keyA isEqual:keyB], @"Shortcuts not equal when key codes differ.");
+    XCTAssertFalse([keyA isEqual:keyC], @"Shortcuts not equal when modifier flags differ.");
+    XCTAssertTrue([keyA isEqual:keyD], @"Shortcuts are equal when key codes and modifiers are.");
+    XCTAssertFalse([keyA isEqual:@"foo"], @"Shortcut not equal to an object of a different class.");
+}
+
 - (void) testShortcutRecorderCompatibility
 {
     MASShortcut *key = [MASShortcut shortcutWithKeyCode:87 modifierFlags:1048576];