about summary refs log tree commit diff
path: root/db_actions_test.go
blob: 7b9642638d5a34b18378fee90446fe3d48cf3bac (plain)
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
// db_actions defines actions on the database
// Copyright (C) 2019 Emile Hansmaennel
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.

package db_actions

import (
	"database/sql"
	"reflect"
	"testing"

	"git.darknebu.la/GalaxySimulator/structs"
	_ "github.com/lib/pq"
)

func TestCalcAllForces(t *testing.T) {
	// define a database
	db = ConnectToDB()
	db.SetMaxOpenConns(75)

	type args struct {
		database *sql.DB
		star     structs.Star2D
		theta    float64
	}
	tests := []struct {
		name string
		args args
		want structs.Vec2
	}{
		{
			name: "force acting on a single star",
			args: args{
				database: db,
				star: structs.Star2D{
					C: structs.Vec2{
						X: 100,
						Y: 100,
					},
					V: structs.Vec2{
						X: 0,
						Y: 0,
					},
					M: 1000,
				},
				theta: 0.5,
			},
			want: structs.Vec2{
				X: 0,
				Y: 0,
			},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := CalcAllForces(tt.args.database, tt.args.star, tt.args.theta); !reflect.DeepEqual(got, tt.want) {
				t.Errorf("CalcAllForces() = %v, want %v", got, tt.want)
			}
		})
	}
}