#import "TestYapDatabase.h"
#import "TestObject.h"

#import "YapDatabase.h"
#import "YapDatabasePrivate.h"
#import "YapDatabaseTransaction+Timestamp.h"

#import <libkern/OSAtomic.h>


@implementation TestYapDatabase

- (NSString *)databasePath:(NSString *)suffix
{
	NSArray *paths = NSSearchPathForDirectoriesInDomains(NSCachesDirectory, NSUserDomainMask, YES);
	NSString *baseDir = ([paths count] > 0) ? [paths objectAtIndex:0] : NSTemporaryDirectory();
	
	NSString *databaseName = [NSString stringWithFormat:@"TestYapDatabase-%@.sqlite", suffix];
	
	return [baseDir stringByAppendingPathComponent:databaseName];
}

- (void)test1
{
	NSString *databasePath = [self databasePath:NSStringFromSelector(_cmd)];
	
	[[NSFileManager defaultManager] removeItemAtPath:databasePath error:NULL];
	YapDatabase *database = [[YapDatabase alloc] initWithPath:databasePath];
	
	STAssertNotNil(database, @"Oops");
	
	YapDatabaseConnection *connection1 = [database newConnection];
	YapDatabaseConnection *connection2 = [database newConnection];
	
	TestObject *object = [TestObject generateTestObject];
	TestObjectMetadata *metadata = [object extractMetadata];
	
	NSString *key1 = @"some-key-1";
	NSString *key2 = @"some-key-2";
	NSString *key3 = @"some-key-3";
	NSString *key4 = @"some-key-4";
	NSString *key5 = @"some-key-5";
	
	__block id aObj;
	__block id aMetadata;
	__block BOOL result;
	
	[connection1 readWriteWithBlock:^(YapDatabaseReadWriteTransaction *transaction){
		
		// Test empty database
		
		STAssertTrue([transaction numberOfKeys] == 0, @"Expected zero key count");
		STAssertTrue([[transaction allKeys] count] == 0, @"Expected empty array");
		
		STAssertNil([transaction objectForKey:@"non-existant-key"], @"Expected nil object");
		STAssertNil([transaction primitiveDataForKey:@"non-existant-key"], @"Expected nil data");
		
		STAssertFalse([transaction hasObjectForKey:@"non-existant-key"], @"Expected NO object for key");
		
		BOOL result = [transaction getObject:&aObj metadata:&aMetadata forKey:@"non-existant-key"];
		
		STAssertFalse(result, @"Expected NO getObject for key");
		STAssertNil(aObj, @"Expected object to be set to nil");
		STAssertNil(aMetadata, @"Expected metadata to be set to nil");
		
		STAssertNil([transaction metadataForKey:@"non-existant-key"], @"Expected nil metadata");
		
		STAssertNoThrow([transaction removeObjectForKey:@"non-existant-key"], @"Expected no issues");
		
		NSArray *keys = @[@"non",@"existant",@"keys"];
		STAssertNoThrow([transaction removeObjectsForKeys:keys], @"Expected no issues");
		
		__block NSUInteger count = 0;
		
		[transaction enumerateKeysAndMetadataUsingBlock:^(NSString *key, id metadata, BOOL *stop){
			count++;
		}];
		
		STAssertTrue(count == 0, @"Expceted zero keys");
		
		[transaction enumerateKeysAndObjectsUsingBlock:^(NSString *key, id object, id metadata, BOOL *stop){
			count++;
		}];
		
		STAssertTrue(count == 0, @"Expceted zero keys");
		
		// Attempt to set metadata for a key that has no associated object.
		// It should silently fail (do nothing).
		// And further queries to fetch metadata for the same key should return nil.
		
		STAssertNoThrow([transaction setMetadata:metadata forKey:@"non-existant-key"], @"Expected nothing to happen");
		
		STAssertNil([transaction metadataForKey:@"non-existant-key"], @"Expected nil metadata since no object");
	}];
	
	[connection2 readWriteWithBlock:^(YapDatabaseReadWriteTransaction *transaction){
		
		// Test object without metadata
		
		[transaction setObject:object forKey:key1];
		
		STAssertTrue([transaction numberOfKeys] == 1, @"Expected 1 key");
		STAssertTrue([[transaction allKeys] count] == 1, @"Expected 1 key");
		
		STAssertNotNil([transaction objectForKey:key1], @"Expected non-nil object");
		STAssertNotNil([transaction primitiveDataForKey:key1], @"Expected non-nil data");
		
		STAssertTrue([transaction hasObjectForKey:key1], @"Expected YES");
		
		result = [transaction getObject:&aObj metadata:&aMetadata forKey:key1];
		
		STAssertTrue(result, @"Expected YES");
		STAssertNotNil(aObj, @"Expected non-nil object");
		STAssertNil(aMetadata, @"Expected nil metadata");
		
		STAssertNil([transaction metadataForKey:key1], @"Expected nil metadata");
		
		[transaction enumerateKeysAndMetadataUsingBlock:^(NSString *key, id metadata, BOOL *stop){
			
			STAssertNil(metadata, @"Expected nil metadata");
		}];
		
		[transaction enumerateKeysAndObjectsUsingBlock:^(NSString *key, id object, id metadata, BOOL *stop){
			
			STAssertNotNil(aObj, @"Expected non-nil object");
			STAssertNil(metadata, @"Expected nil metadata");
		}];
	}];

	[connection1 readWriteWithBlock:^(YapDatabaseReadWriteTransaction *transaction){
		
		// Test remove object
		
		[transaction removeObjectForKey:key1];
		
		STAssertTrue([transaction numberOfKeys] == 0, @"Expected 0 keys");
		STAssertTrue([[transaction allKeys] count] == 0, @"Expected 0 keys");
		
		STAssertNil([transaction objectForKey:key1], @"Expected nil object");
		STAssertNil([transaction primitiveDataForKey:key1], @"Expected nil data");
		
		STAssertFalse([transaction hasObjectForKey:key1], @"Expected NO");
	}];

	[connection2 readWriteWithBlock:^(YapDatabaseReadWriteTransaction *transaction){
		
		// Test object with metadata
		
		[transaction setObject:object forKey:key1 withMetadata:metadata];
		
		STAssertTrue([transaction numberOfKeys] == 1, @"Expected 1 key");
		STAssertTrue([[transaction allKeys] count] == 1, @"Expected 1 key");
		
		STAssertNotNil([transaction objectForKey:key1], @"Expected non-nil object");
		STAssertNotNil([transaction primitiveDataForKey:key1], @"Expected non-nil data");
		
		STAssertTrue([transaction hasObjectForKey:key1], @"Expected YES");
		
		result = [transaction getObject:&aObj metadata:&aMetadata forKey:key1];
		
		STAssertTrue(result, @"Expected YES");
		STAssertNotNil(aObj, @"Expected non-nil object");
		STAssertNotNil(aMetadata, @"Expected non-nil metadata");
		
		STAssertNotNil([transaction metadataForKey:key1], @"Expected non-nil metadata");
		
		[transaction enumerateKeysAndMetadataUsingBlock:^(NSString *key, id metadata, BOOL *stop){
			
			STAssertNotNil(metadata, @"Expected non-nil metadata");
		}];
		
		[transaction enumerateKeysAndObjectsUsingBlock:^(NSString *key, id object, id metadata, BOOL *stop){
			
			STAssertNotNil(aObj, @"Expected non-nil object");
			STAssertNotNil(metadata, @"Expected non-nil metadata");
		}];
	}];

	[connection1 readWriteWithBlock:^(YapDatabaseReadWriteTransaction *transaction){
		
		// Test multiple objects
		
		[transaction setObject:object forKey:key2 withMetadata:metadata];
		[transaction setObject:object forKey:key3 withMetadata:metadata];
		[transaction setObject:object forKey:key4 withMetadata:metadata];
		[transaction setObject:object forKey:key5 withMetadata:metadata];
		
		STAssertTrue([transaction numberOfKeys] == 5, @"Expected 5 keys");
		STAssertTrue([[transaction allKeys] count] == 5, @"Expected 5 keys");
		
		STAssertNotNil([transaction objectForKey:key1], @"Expected non-nil object");
		STAssertNotNil([transaction objectForKey:key2], @"Expected non-nil object");
		STAssertNotNil([transaction objectForKey:key3], @"Expected non-nil object");
		STAssertNotNil([transaction objectForKey:key4], @"Expected non-nil object");
		STAssertNotNil([transaction objectForKey:key5], @"Expected non-nil object");
		
		STAssertTrue([transaction hasObjectForKey:key1], @"Expected YES");
		STAssertTrue([transaction hasObjectForKey:key2], @"Expected YES");
		STAssertTrue([transaction hasObjectForKey:key3], @"Expected YES");
		STAssertTrue([transaction hasObjectForKey:key4], @"Expected YES");
		STAssertTrue([transaction hasObjectForKey:key5], @"Expected YES");
		
		STAssertNotNil([transaction metadataForKey:key1], @"Expected non-nil metadata");
		STAssertNotNil([transaction metadataForKey:key2], @"Expected non-nil metadata");
		STAssertNotNil([transaction metadataForKey:key3], @"Expected non-nil metadata");
		STAssertNotNil([transaction metadataForKey:key4], @"Expected non-nil metadata");
		STAssertNotNil([transaction metadataForKey:key5], @"Expected non-nil metadata");
	}];

	[connection2 readWriteWithBlock:^(YapDatabaseReadWriteTransaction *transaction){
		
		// Test remove multiple objects
		
		[transaction removeObjectsForKeys:@[ key1, key2, key3 ]];
		
		STAssertTrue([transaction numberOfKeys] == 2, @"Expected 2 keys");
		STAssertTrue([[transaction allKeys] count] == 2, @"Expected 2 keys");
		
		STAssertNil([transaction objectForKey:key1], @"Expected nil object");
		STAssertNil([transaction objectForKey:key2], @"Expected nil object");
		STAssertNil([transaction objectForKey:key3], @"Expected nil object");
		STAssertNotNil([transaction objectForKey:key4], @"Expected non-nil object");
		STAssertNotNil([transaction objectForKey:key5], @"Expected non-nil object");
		
		STAssertFalse([transaction hasObjectForKey:key1], @"Expected NO");
		STAssertFalse([transaction hasObjectForKey:key2], @"Expected NO");
		STAssertFalse([transaction hasObjectForKey:key3], @"Expected NO");
		STAssertTrue([transaction hasObjectForKey:key4], @"Expected YES");
		STAssertTrue([transaction hasObjectForKey:key5], @"Expected YES");
		
		STAssertNil([transaction metadataForKey:key1], @"Expected nil metadata");
		STAssertNil([transaction metadataForKey:key2], @"Expected nil metadata");
		STAssertNil([transaction metadataForKey:key3], @"Expected nil metadata");
		STAssertNotNil([transaction metadataForKey:key4], @"Expected non-nil metadata");
		STAssertNotNil([transaction metadataForKey:key5], @"Expected non-nil metadata");
	}];

	[connection1 readWriteWithBlock:^(YapDatabaseReadWriteTransaction *transaction){
		
		// Test remove all objects
		
		[transaction removeAllObjects];
		
		STAssertNil([transaction objectForKey:key1], @"Expected nil object");
		STAssertNil([transaction objectForKey:key2], @"Expected nil object");
		STAssertNil([transaction objectForKey:key3], @"Expected nil object");
		STAssertNil([transaction objectForKey:key4], @"Expected nil object");
		STAssertNil([transaction objectForKey:key5], @"Expected nil object");
		
		STAssertFalse([transaction hasObjectForKey:key1], @"Expected NO");
		STAssertFalse([transaction hasObjectForKey:key2], @"Expected NO");
		STAssertFalse([transaction hasObjectForKey:key3], @"Expected NO");
		STAssertFalse([transaction hasObjectForKey:key4], @"Expected NO");
		STAssertFalse([transaction hasObjectForKey:key5], @"Expected NO");
		
		STAssertNil([transaction metadataForKey:key1], @"Expected nil metadata");
		STAssertNil([transaction metadataForKey:key2], @"Expected nil metadata");
		STAssertNil([transaction metadataForKey:key3], @"Expected nil metadata");
		STAssertNil([transaction metadataForKey:key4], @"Expected nil metadata");
		STAssertNil([transaction metadataForKey:key5], @"Expected nil metadata");
	}];
	
	[connection2 readWithBlock:^(YapDatabaseReadTransaction *transaction) {
		
		// Test all objects have been removed
		
		STAssertTrue([transaction numberOfKeys] == 0, @"Expected 0 objects in database");
	}];
	
	[connection1 readWriteWithBlock:^(YapDatabaseReadWriteTransaction *transaction){
		
		// Test add object again
		
		[transaction setObject:object forKey:key1];
		
		STAssertTrue([transaction numberOfKeys] == 1, @"Expected 1 key");
		STAssertTrue([[transaction allKeys] count] == 1, @"Expected 1 key");
		
		STAssertNotNil([transaction objectForKey:key1], @"Expected non-nil object");
		STAssertNotNil([transaction primitiveDataForKey:key1], @"Expected non-nil data");
		
		STAssertTrue([transaction hasObjectForKey:key1], @"Expected YES");
	}];
	
	[connection1 readWriteWithBlock:^(YapDatabaseReadWriteTransaction *transaction){
		
		// Test object back again
		
		STAssertTrue([transaction numberOfKeys] == 1, @"Expected 1 key");
		STAssertTrue([[transaction allKeys] count] == 1, @"Expected 1 key");
		
		STAssertNotNil([transaction objectForKey:key1], @"Expected non-nil object");
		STAssertNotNil([transaction primitiveDataForKey:key1], @"Expected non-nil data");
		
		STAssertTrue([transaction hasObjectForKey:key1], @"Expected YES");
	}];
}

- (void)testPropertyListSerializerDeserializer
{
	NSData *(^propertyListSerializer)(id) = [YapDatabase propertyListSerializer];
	id (^propertyListDeserializer)(NSData *) = [YapDatabase propertyListDeserializer];
	
	NSDictionary *originalDict = @{ @"date":[NSDate date], @"string":@"string" };
	
	NSData *data = propertyListSerializer(originalDict);
	
	NSDictionary *deserializedDictionary = propertyListDeserializer(data);
	
	STAssertTrue([originalDict isEqualToDictionary:deserializedDictionary], @"PropertyList serialization broken");
}

- (void)testTimestampSerializerDeserializer
{
	NSData *(^timestampSerializer)(id) = [YapDatabase timestampSerializer];
	id (^timestampDeserializer)(NSData *) = [YapDatabase timestampDeserializer];
	
	NSDate *originalDate = [NSDate date];
	
	NSData *data = timestampSerializer(originalDate);
	
	NSDate *deserializedDate = timestampDeserializer(data);
	
	STAssertTrue([originalDate isEqual:deserializedDate], @"Timestamp serialization broken");
}

- (void)test2
{
	NSString *databasePath = [self databasePath:NSStringFromSelector(_cmd)];
	
	[[NSFileManager defaultManager] removeItemAtPath:databasePath error:NULL];
	YapDatabase *database = [[YapDatabase alloc] initWithPath:databasePath];
	
	STAssertNotNil(database, @"Oops");
	
	/// Test concurrent connections.
	///
	/// Ensure that a read-only transaction can continue while a read-write transaction starts.
	/// Ensure that a read-only transaction can start while a read-write transaction is in progress.
	/// Ensure that a read-only transaction picks up the changes after a read-write transaction.
	
	YapDatabaseConnection *connection1 = [database newConnection];
	YapDatabaseConnection *connection2 = [database newConnection];
	
	NSString *key = @"some-key";
	TestObject *object = [TestObject generateTestObject];
	TestObjectMetadata *metadata = [object extractMetadata];
	
	dispatch_queue_t concurrentQueue = dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0);
	dispatch_async(concurrentQueue, ^{
		
		[NSThread sleepForTimeInterval:0.1]; // Zz
		
		[connection1 readWriteWithBlock:^(YapDatabaseReadWriteTransaction *transaction){
			
			[transaction setObject:object forKey:key withMetadata:metadata];
			
			[NSThread sleepForTimeInterval:0.4]; // Zzzzzzzzzzzzzzzzzzzzzzzzzz
		}];
		
	});
	
	// This transaction should start before the read-write transaction has started
	[connection2 readWithBlock:^(YapDatabaseReadTransaction *transaction){
		
		STAssertNil([transaction objectForKey:key], @"Expected nil object");
		STAssertNil([transaction metadataForKey:key], @"Expected nil metadata");
	}];
	
	[NSThread sleepForTimeInterval:0.2]; // Zzzzzz
	
	// This transaction should start after the read-write transaction has started, but before it has committed
	[connection2 readWithBlock:^(YapDatabaseReadTransaction *transaction){
		
		STAssertNil([transaction objectForKey:key], @"Expected nil object");
		STAssertNil([transaction metadataForKey:key], @"Expected nil metadata");
	}];
	
	[NSThread sleepForTimeInterval:0.4]; // Zzzzzzzzzzzzzzzzzzzzzzzzzz
	
	// This transaction should start after the read-write transaction has completed
	[connection2 readWithBlock:^(YapDatabaseReadTransaction *transaction){
		
		STAssertNotNil([transaction objectForKey:key], @"Expected non-nil object");
		STAssertNotNil([transaction metadataForKey:key], @"Expected non-nil metadata");
	}];
}


- (void)test3
{
	NSString *databasePath = [self databasePath:NSStringFromSelector(_cmd)];
	
	[[NSFileManager defaultManager] removeItemAtPath:databasePath error:NULL];
	YapDatabase *database = [[YapDatabase alloc] initWithPath:databasePath];
	
	STAssertNotNil(database, @"Oops");
	
	/// Test concurrent connections.
	///
	/// Ensure that a read-only transaction properly unblocks a blocked read-write transaction.
	/// Need to turn on logging to check this.
	
	YapDatabaseConnection *connection1 = [database newConnection];
	YapDatabaseConnection *connection2 = [database newConnection];
	
	NSString *key = @"some-key";
	TestObject *object = [TestObject generateTestObject];
	TestObjectMetadata *metadata = [object extractMetadata];
	
	dispatch_queue_t concurrentQueue = dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0);
	dispatch_async(concurrentQueue, ^{
		
		[NSThread sleepForTimeInterval:0.2]; // Zz
		
		[connection1 readWriteWithBlock:^(YapDatabaseReadWriteTransaction *transaction){
			
			[transaction setObject:object forKey:key withMetadata:metadata];
		}];
		
	});
	
	// This transaction should before the read-write transaction
	[connection2 readWithBlock:^(YapDatabaseReadTransaction *transaction){
		
		STAssertNil([transaction objectForKey:key], @"Expected nil object");
		STAssertNil([transaction metadataForKey:key], @"Expected nil metadata");
		
		[NSThread sleepForTimeInterval:2.0]; // Zzzzzzzzzzz
	}];
	
	[NSThread sleepForTimeInterval:0.2]; // Zz
	
	// This transaction should start after the read-write transaction
	[connection2 readWithBlock:^(YapDatabaseReadTransaction *transaction){
		
		STAssertNotNil([transaction objectForKey:key], @"Expected non-nil object");
		STAssertNotNil([transaction metadataForKey:key], @"Expected non-nil metadata");
	}];
}


- (void)test4
{
	NSString *databasePath = [self databasePath:NSStringFromSelector(_cmd)];
	
	[[NSFileManager defaultManager] removeItemAtPath:databasePath error:NULL];
	YapDatabase *database = [[YapDatabase alloc] initWithPath:databasePath];
	
	STAssertNotNil(database, @"Oops");
	
	/// Ensure large write doesn't block concurrent read operations on other connections.
	
	YapDatabaseConnection *connection1 = [database newConnection];
	YapDatabaseConnection *connection2 = [database newConnection];
	
	__block int32_t doneWritingFlag = 0;
	
	dispatch_queue_t concurrentQueue = dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0);
	dispatch_async(concurrentQueue, ^{
		
		NSTimeInterval start = [NSDate timeIntervalSinceReferenceDate];
		
		[connection1 readWriteWithBlock:^(YapDatabaseReadWriteTransaction *transaction){
			
			int i;
			for (i = 0; i < 100; i++)
			{
				NSString *key = [NSString stringWithFormat:@"some-key-%d", i];
				TestObject *object = [TestObject generateTestObject];
				TestObjectMetadata *metadata = [object extractMetadata];
				
				[transaction setObject:object forKey:key withMetadata:metadata];
			}
		}];
		
		NSTimeInterval elapsed = [NSDate timeIntervalSinceReferenceDate] - start;
		NSLog(@"Write operation: %.6f", elapsed);
		
		OSAtomicAdd32(1, &doneWritingFlag);
	});
	
	while (OSAtomicAdd32(0, &doneWritingFlag) == 0)
	{
		NSTimeInterval start = [NSDate timeIntervalSinceReferenceDate];
		
		[connection2 readWithBlock:^(YapDatabaseReadTransaction *transaction){
			
			(void)[transaction objectForKey:@"some-key-0"];
		}];
		
		NSTimeInterval elapsed = [NSDate timeIntervalSinceReferenceDate] - start;
		
		STAssertTrue(elapsed < 0.05, @"Read-Only transaction taking too long...");
	}
}

@end