wservice/service_test.go at master · vstoianovici/wservice · 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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
package wservice

import (
	"os"
	"sync"
	"testing"

	_ "github.com/lib/pq"
	"github.com/stretchr/testify/assert"
)

func TestNewServiceDefault(t *testing.T) {
	fileName := "./cmd/postgresql.cfg"
	svc, err := getDbConfig(fileName)
	assert.Nil(t, err)
	assert.NotNil(t, svc)
}

func TestNewServiceNoFile(t *testing.T) {
	fileName := "./postgresql.cfg_"
	// Open Postgres configuration file
	file, err := os.Open(fileName)
	assert.NotNil(t, err)
	defer file.Close()
}

func TestNewServiceWrongFormatDelimeter(t *testing.T) {
	fileName := "./cmd/test/postgresql_delim.cfg"
	_, err := getDbConfig(fileName)
	assert.NotNil(t, err)
}

func TestNewServiceWrongFormatFewLines(t *testing.T) {
	fileName := "./cmd/test/postgresql_line_few.cfg"
	_, err := getDbConfig(fileName)
	assert.NotNil(t, err)
}

func TestNewServiceWrongFormatManyLines(t *testing.T) {
	fileName := "./cmd/test/postgresql_line_many.cfg"
	_, err := getDbConfig(fileName)
	assert.NotNil(t, err)
}
func TestGetTable(t *testing.T) {
	svc, _ := getDbConfig("./cmd/postgresql.cfg")
	vSlice, err := svc.GetTable("Accounts")
	assert.Contains(t, vSlice, "Success.")
	assert.Nil(t, err)
	vSlice, err = svc.GetTable("Transfers")
	assert.Contains(t, vSlice, "Success.")
	assert.Nil(t, err)
	vSlice, err = svc.GetTable("someOtherTable")
	assert.NotContains(t, vSlice, "[]")
}

func TestDoTransferRegular(t *testing.T) {
	svc, _ := getDbConfig("./cmd/postgresql.cfg")
	status, err := svc.DoTransfer("bob123", "alice456", "30")
	assert.Contains(t, status, "success")
	assert.Nil(t, err)
	status, err = svc.DoTransfer("alice456", "bob123", "30")
	assert.Contains(t, status, "success")
	assert.Nil(t, err)
}

func TestDoTransferNoDestAccount(t *testing.T) {
	svc, _ := getDbConfig("./cmd/postgresql.cfg")
	status, err := svc.DoTransfer("alice456", "amockaccount123", "30")
	assert.Contains(t, status, "error")
	assert.EqualError(t, err, "The destination account does not exist")
}

func TestDoTransferNoSourceAccount(t *testing.T) {
	svc, _ := getDbConfig("./cmd/postgresql.cfg")
	status, err := svc.DoTransfer("amockaccount123", "alice456", "30")
	assert.Contains(t, status, "error")
	assert.EqualError(t, err, "The source account does not exist")
}

func TestDoTransferSameAccount(t *testing.T) {
	svc, _ := getDbConfig("./cmd/postgresql.cfg")
	status, err := svc.DoTransfer("alice456", "alice456", "30")
	assert.Contains(t, status, "error")
	assert.EqualError(t, err, "the source account is the same as the destination account. ")
}

func TestDoTransferBalanceMinus(t *testing.T) {
	svc, _ := getDbConfig("./cmd/postgresql.cfg")
	status, err := svc.DoTransfer("alice456", "bob123", "300000")
	assert.Contains(t, status, "error")
	assert.EqualError(t, err, "Balance insuficient for transaction")
}

func TestDoTransferWrongCurrency(t *testing.T) {
	svc, _ := getDbConfig("./cmd/postgresql.cfg")
	status, err := svc.DoTransfer("alice456", "marcy789", "30")
	assert.Contains(t, status, "error")
	assert.EqualError(t, err, "Not same currency in transaction source and destination")
}

func TestDoTransferConcurent(t *testing.T) {
	var wg sync.WaitGroup
	svc, _ := getDbConfig("./cmd/postgresql.cfg")
	for i := 0; i < 20; i++ {
		wg.Add(1)
		go func() {
			for j := 0; j < 5; j++ {
				status, err := svc.DoTransfer("bob123", "alice456", "1")
				//log.Println("status", status, " and ", j)
				assert.Contains(t, status, "success")
				assert.Nil(t, err)
			}
			wg.Done()
		}()
	}
	wg.Wait()

}

func TestDoTransferConcurentRevertChanges(t *testing.T) {
	var wg sync.WaitGroup
	svc, _ := getDbConfig("./cmd/postgresql.cfg")
	for i := 0; i < 20; i++ {
		wg.Add(1)
		go func() {
			for j := 0; j < 5; j++ {
				status, err := svc.DoTransfer("alice456", "bob123", "1")
				//log.Println("status", status, " and ", j)
				assert.Contains(t, status, "success")
				assert.Nil(t, err)
			}
			wg.Done()
		}()
	}
	wg.Wait()

}