tests.cpp

#include<assert.h>
#include "tea.cpp"
#pragma GCC diagnostic ignored "-Wwrite-strings"

void strcpyIsWorking()
{
	char * text = "fjdsklg";
	char * dest = new char[10];
	strcopy(dest, text, 7);
	assert( 0 == strcmp(text, dest) );
	strcopy(dest, text);
	assert( 0 == strcmp(text, dest) );
}

void StringSourceIsWorking()
{
	char * text = "abcdeabcdeabcdeabcde";
	StringSource* a = new StringSource(text);
	assert(strcmp(text, a->getValue())==0 && text[0] == a->getc());
}

void StringTargetIsWorking()
{
	char * text = "abcdeabcdeabcdeabcde";
    StringSource*a = new StringSource(text);
    StringTarget*b = new StringTarget(strlen(text));
    while (a->hasNext())
    {
    	b->putc(a->getc());
    }
	assert(strcmp(text, b->getValue()) == 0);
}

void RandomSourceIsWorking()
{
	int n = 30;
	RandomSource* r = new RandomSource(n);
	for(int i = 0; i<n; i++)
	{
		assert(r->hasNext());
		r->getc();
	}
	assert(!r->hasNext());
}

void readBytesIsWorking()
{
	char * text = "jljkljkljjk";
	StringSource* a = new StringSource(text);
	int8_t *s = new int8_t[strlen(text)+1];
	s[strlen(text)]=0;
	readBytes(a, s, strlen(text));
	assert(0 == strcmp((char*)s, text));
}

void putBytesIsWorking()
{
	char * text = "jskfldghdsufdibg";
	StringTarget * t = new StringTarget(strlen(text));
	putBytes(t, (int8_t*)text, strlen(text) );
	assert( 0 == strcmp(text, t->getValue()) );
	
}

void crypt_decryptOfEncryptIsTheSame()
{
	char * text = "n the ea";//8 chars long string
	char * key = "jfkldsajfklaskfj"; //16 chars long string
	StringSource* vs = new StringSource(text);
	StringSource* ks = new StringSource(key);
	int8_t v[8];
	int8_t k[16];
	readBytes(vs, v, 8);
	readBytes(ks, k, 16);
	crypt((int32_t*)v, (int32_t*)k, ENCRYPT, 32);
	crypt((int32_t*)v, (int32_t*)k, DECRYPT, 32);
	StringTarget* vt = new StringTarget(8);
	putBytes(vt, v, 8);
	assert( 0 == strcmp(text, vt->getValue()) );
	
}

void engine_decryptOfEncryptIsTheSame()
{
	char* key = "jfkldsajfklaskfj";
	char* text = "n the ea";
	StringSource* a = new StringSource(text);
	StringTarget* b = new StringTarget(strlen(text)+7);
	StringSource* keySource = new StringSource(key);
	engine(a, b, keySource, ENCRYPT);
	a->setValue(b->getValue(), b->getSize() );
	keySource->setValue(key);
	b->reset();
	engine(a, b, keySource, DECRYPT);
	assert( 0 == strcmp(text, b->getValue()) );
	
}

void filesTesting(Mode mode)
{
	char * in = "in.txt";
	char * t = "temp.txt";
	char * out = "out.txt";
	FileSource * a = new FileSource(mode == ENCRYPT? in : t);
	FileTarget * b = new FileTarget(mode == ENCRYPT? t : out);
	char* key = "jfkldsajfklaskfjakljf";
	StringSource* keySource = new StringSource(key);
	engine(a, b, keySource, mode);
	delete a;
	delete b;
}

void cryptFileTest()
{
	char * in = "in.txt";
	char * t = "temp.txt";
	char * out = "out.txt";
	char * key = "key.txt";
	cryptFile(in, t, key, ENCRYPT);
	cryptFile(t, out, key, DECRYPT);
}

int main()
{
	strcpyIsWorking();
	StringSourceIsWorking();
	StringTargetIsWorking();
	RandomSourceIsWorking();
	readBytesIsWorking();
	putBytesIsWorking();
	crypt_decryptOfEncryptIsTheSame();
	engine_decryptOfEncryptIsTheSame();
//	filesTesting(ENCRYPT);
//	filesTesting(DECRYPT);
	cryptFileTest();
}