sparrow/services/knowoapi/model/device_test.go

package model

import (
	"sparrow/pkg/mysql"
	"testing"

	"github.com/jinzhu/gorm"
)

func TestGetDeviceCount(t *testing.T) {

	device := newDevice(t)
	count, err := device.GetDeviceCount(1)
	if err != nil {
		t.Fatal(err)
	}
	t.Log(count)
}

func TestGetActiveNumberOfDate(t *testing.T) {
	d := newDevice(t)
	count, err := d.GetActiveNumberOfDate(1, "2018-12-27")
	if err != nil {
		t.Fatal(err)
	}
	t.Log(count)
}

func TestGetLivelyCountOfDate(t *testing.T) {
	d := newDevice(t)
	count, err := d.GetLivelyCountOfDate(1, "2019-01-02")
	if err != nil {
		t.Fatal(err)
	}
	t.Log(count)
}

// func TestGetActiveOf7Days(t *testing.T) {
// 	d := newDevice(t)
// 	datas, err := d.GetActiveOf7Days(1)
// 	if err != nil {
// 		t.Fatal(err)
// 	}
// 	for k, v := range datas {
// 		t.Logf("%d, %v", k, v)
// 	}
// }

// func TestGetActiveOf14Days(t *testing.T) {
// 	d := newDevice(t)
// 	datas, err := d.GetActiveOf14Days(1)
// 	if err != nil {
// 		t.Fatal(err)
// 	}
// 	for k, v := range datas {
// 		t.Logf("%d, %v", k, v)
// 	}
// }
// func TestGetActiveOf30Days(t *testing.T) {
// 	d := newDevice(t)
// 	datas, err := d.GetActiveOf30Days(1)
// 	if err != nil {
// 		t.Fatal(err)
// 	}
// 	for k, v := range datas {
// 		t.Logf("%d, %v", k, v)
// 	}
// }

func TestGetActiveOfNumDays(t *testing.T) {
	d := newDevice(t)
	datas, err := d.GetActiveOfNumDays(1, 30)
	if err != nil {
		t.Fatal(err)
	}
	for k, v := range datas {
		t.Logf("%d, %v", k, v)
	}
}
func TestGetLivelyOfNumDays(t *testing.T) {
	d := newDevice(t)
	datas, err := d.GetLivelyOfNumDays(1, 30)
	if err != nil {
		t.Fatal(err)
	}
	for k, v := range datas {
		t.Logf("%d, %v", k, v)
	}
}
func TestGetDevices(t *testing.T) {
	d := newDevice(t)
	datas, total, err := d.GetDevices(1, 0, 1, 10, "")
	if err != nil {
		t.Fatal(err)
	}
	t.Logf("%v,%d", datas, total)
}
func newDevice(t *testing.T) *Device {
	db, err := getDB()
	if err != nil {
		t.Fatal(err)
	}
	device := new(Device).Init(db)
	return device
}

func getDB() (*gorm.DB, error) {
	db, err := mysql.GetClient("39.98.250.155", "3306", "gxt-iot-db", "root", "gEkYDPloQcp93t4WHr3X")
	if err != nil {
		return nil, err
	}
	gormdb, err := gorm.Open("mysql", db)
	if err != nil {
		return nil, err
	}
	gormdb.SingularTable(true)
	gormdb.LogMode(true)
	return gormdb, nil
}