#include <check.h>

#include <config.h>
#include <stasis/transactional.h>
#include <stasis/replacementPolicy.h>
#include <stasis/doubleLinkedList.h>
#include <assert.h>
#include "../check_includes.h"

#define LOG_NAME   "check_replacementPolicy.log"

long myrandom(long x) {
  double xx = x;
  double r = random();
  double max = ((uint64_t)RAND_MAX)+1;
  max /= xx;
  return (long)((r/max));
}

#define OBJECT_COUNT 1000
#define OP_COUNT     10000000

typedef struct LL_ENTRY(node_t) node_t;

typedef struct tracker { 
  long val;
  intptr_t key;
  int inCache;
} tracker;

static node_t * getKey(void * page, void * ignore) { 
  tracker * p = page;
  return (node_t*)p->key;
}
static void setKey(void * page, node_t * n, void * ignore) { 
  tracker * p = page;
  p->key = (intptr_t) n;
}


void randomTest(replacementPolicy * lru) { 
  time_t seed = time(0);
  printf("\nSeed = %ld\n", seed);
  srandom(seed);

  int cachedCount = 0;
  
  tracker * t = calloc(OBJECT_COUNT, sizeof(tracker));
  for(int i = 0; i < OBJECT_COUNT; i++) { 
    t[i].val = i;
  }
  for(unsigned long j = 0; j < 100000000UL; j++) { 
    int op = myrandom(100);

    int i = myrandom(OBJECT_COUNT);
    if(op < 10) { 
      // TOGGLE IN CACHE
      if(!t[i].inCache) { 
	lru->insert(lru, &t[i]);
	t[i].inCache = 1;
	cachedCount ++;
      } else { 
	void * v = lru->remove(lru, &t[i]);
	assert(v == &t[i]);
	t[i].inCache = 0;
	cachedCount --;
      }
    } else if(op < 30) { 
      // Get stale + remove
      tracker * tr = lru->getStale(lru);
      if( tr ) { 
	assert(tr->inCache);
	assert(tr == &t[tr->val]);
	tr = lru->remove(lru, tr);
	assert(tr == &t[tr->val]);
	tr->inCache = 0;
	assert(cachedCount != 0);
	cachedCount --;
      } else { 
	assert(cachedCount == 0);
      }
    } else if(op < 50) { 
      // Get stale
      tracker * tr = lru->getStale(lru);
      if(tr) { 
	assert(tr->inCache);
	assert(tr == &t[tr->val]);
	assert(cachedCount != 0);
      } else { 
	assert(cachedCount == 0);
      }
    } else { 
      // Hit
      if(t[i].inCache) lru->hit(lru, &t[i]);
    }
  }
}
START_TEST(replacementPolicyLRURandomTest) {
  replacementPolicy * lru = lruFastInit(getKey, setKey, 0);
  randomTest(lru);
} END_TEST
START_TEST(replacementPolicyLRUFastRandomTest) {
  replacementPolicy * lru = lruFastInit(getKey, setKey, 0);
  randomTest(lru);
} END_TEST

Suite * check_suite(void) {
  Suite *s = suite_create("replacemenPolicy");
  /* Begin a new test */
  TCase *tc = tcase_create("multithreaded");
  tcase_set_timeout(tc, 1200); // twenty minute timeout
  /* Sub tests are added, one per line, here */
  tcase_add_test(tc, replacementPolicyLRURandomTest); 
  tcase_add_test(tc, replacementPolicyLRUFastRandomTest); 

  /* --------------------------------------------- */
  
  tcase_add_checked_fixture(tc, setup, teardown);


  suite_add_tcase(s, tc);
  return s;
}

#include "../check_setup.h"