secure_stack/UnitTest.cpp at master · Steap2448/secure_stack · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
#include "UnitTest.h"

TortureRoom TortureRoomConstruct()
{
	TortureRoom tr = (TortureRoom) calloc(1, sizeof(elTortureRoom));
	InitElStack_d(&(tr->stk));

	return tr;
}

char TortureRoomPeekLeft(TortureRoom tr, unsigned shift)
{
	if (shift >= sizeof(tr->stk))
	{
		return UNIT_TEST_SHIFT_ERR;
	}

	char* tmp = &(tr->left);
	tmp += shift;
	return *tmp;
}

char TortureRoomPeekRight(TortureRoom tr, unsigned shift)
{
	if (shift >= sizeof(tr->stk))
	{
		return UNIT_TEST_SHIFT_ERR;
	}

	char* tmp = &(tr->right);
	tmp -= shift;
	return *tmp;
}

int TortureRoomModifyLeft(TortureRoom tr, unsigned shift, char value)
{
	if (shift >= sizeof(tr->stk))
	{
		return UNIT_TEST_SHIFT_ERR;
	}

	char* tmp = &(tr->left);
	tmp += shift;
	(*tmp) = value;
	return UNIT_TEST_OK;
}

int TortureRoomModifyRight(TortureRoom tr, unsigned shift, char value)
{
	if (shift >= sizeof(tr->stk))
	{
		return UNIT_TEST_SHIFT_ERR;
	}

	char* tmp = &(tr->right);
	tmp -= shift;
	(*tmp) = value;
	return UNIT_TEST_OK;
}

int UnitTest()
{
	printf("Conducting UnitTest.\n\n");
	TortureRoom tr = TortureRoomConstruct();
	double buf;
	Stack_d stk = &(tr->stk);

	assert(PushStack_d(stk, 10) == STACK_OK);
	assert(PushStack_d(stk, 15) == STACK_OK);
	assert(PushStack_d(stk, 13) == STACK_OK);
	assert(PushStack_d(stk, 16) == STACK_OK);
	assert(TortureRoomModifyLeft(tr, 10, 54) == UNIT_TEST_OK);
	assert(PopStack_d(stk, &buf) == STACK_CORRUPTION_ERR);

	assert(InitElStack_d(stk) == STACK_OK);
	assert(PushStack_d(stk, 10) == STACK_OK);
	assert(PopStack_d(stk, &buf) == STACK_OK);

	assert(InitElStack_d(stk) == STACK_OK);
	assert(PushStack_d(stk, 10) == STACK_OK);
	assert(PushStack_d(stk, 15) == STACK_OK);
	assert(PushStack_d(stk, 13) == STACK_OK);
	assert(PushStack_d(stk, 16) == STACK_OK);
	assert(TortureRoomModifyLeft(tr, 54, 54) == UNIT_TEST_OK);
	assert(PopStack_d(stk, &buf) == STACK_CORRUPTION_ERR);

	assert(InitElStack_d(stk) == STACK_OK);
	for (int i = 0; i < STACK_SIZE; ++i)
	{
		assert(PushStack_d(stk, i) == STACK_OK);
	}
	assert(PushStack_d(stk, 42) == STACK_OVERFLOW_ERR);
	assert(TortureRoomModifyLeft(tr, 54, 54) == UNIT_TEST_OK);
	assert(PopStack_d(stk, &buf) == STACK_CORRUPTION_ERR);

	assert(InitElStack_d(stk) == STACK_OK);
	for (int i = 0; i < STACK_SIZE; ++i)
	{
		assert(PushStack_d(stk, i) == STACK_OK);
	}
	assert(PushStack_d(stk, 42) == STACK_OVERFLOW_ERR);
	stk->size = 42;
	assert(PopStack_d(stk, &buf) == STACK_CORRUPTION_ERR);

	assert(InitElStack_d(stk) == STACK_OK);
	for (int i = 0; i < STACK_SIZE; ++i)
	{
		assert(PushStack_d(stk, i) == STACK_OK);
	}
	assert(PushStack_d(stk, 42) == STACK_OVERFLOW_ERR);
	stk->position = 42;
	assert(PopStack_d(stk, &buf) == STACK_CORRUPTION_ERR);

	assert(InitElStack_d(stk) == STACK_OK);
	assert(PopStack_d(stk, &buf) == STACK_UNDERFLOW_ERR);

	printf("UnitTest completed. Everything seems to be normal.\n");
	return UNIT_TEST_OK;
}