Influxdata
/
chronograf
mirror of https://github.com/influxdata/chronograf.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Add OrganizationsSourcesStore

pull/5028/head
Michael Desa 2017-10-30 10:49:54 -04:00
parent 11bb189778
commit d5c02afb9b
3 changed files with 387 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
bolt/client.go
View File

@ -25,6 +25,7 @@ type Client struct {
OrganizationsStore *OrganizationsStore
OrganizationUsersStore *OrganizationUsersStore
OrganizationDashboardsStore *OrganizationDashboardsStore
OrganizationSourcesStore *OrganizationSourcesStore
}
// NewClient initializes all stores
@ -44,6 +45,7 @@ func NewClient() *Client {
c.OrganizationsStore = &OrganizationsStore{client: c}
c.OrganizationUsersStore = &OrganizationUsersStore{client: c}
c.OrganizationDashboardsStore = &OrganizationDashboardsStore{client: c}
c.OrganizationSourcesStore = &OrganizationSourcesStore{client: c}
return c
}

77
bolt/organization_sources.go Normal file
View File

@ -0,0 +1,77 @@
package bolt
import (
"context"
"github.com/influxdata/chronograf"
)
type OrganizationSourcesStore struct {
client *Client
}
func (s *OrganizationSourcesStore) All(ctx context.Context) ([]chronograf.Source, error) {
org, err := validOrganization(ctx)
if err != nil {
return nil, err
}
ds, err := s.client.SourcesStore.All(ctx)
if err != nil {
return nil, err
}
dashboards := ds[:0]
for _, d := range ds {
if d.Organization == org {
dashboards = append(dashboards, d)
}
}
return dashboards, nil
}
func (s *OrganizationSourcesStore) Add(ctx context.Context, d chronograf.Source) (chronograf.Source, error) {
org, err := validOrganization(ctx)
if err != nil {
return chronograf.Source{}, err
}
d.Organization = org
return s.client.SourcesStore.Add(ctx, d)
}
func (s *OrganizationSourcesStore) Delete(ctx context.Context, d chronograf.Source) error {
d, err := s.client.SourcesStore.Get(ctx, d.ID)
if err != nil {
return err
}
return s.client.SourcesStore.Delete(ctx, d)
}
func (s *OrganizationSourcesStore) Get(ctx context.Context, id int) (chronograf.Source, error) {
org, err := validOrganization(ctx)
if err != nil {
return chronograf.Source{}, err
}
d, err := s.client.SourcesStore.Get(ctx, id)
if err != nil {
return chronograf.Source{}, err
}
if d.Organization != org {
return chronograf.Source{}, chronograf.ErrSourceNotFound
}
return d, nil
}
func (s *OrganizationSourcesStore) Update(ctx context.Context, d chronograf.Source) error {
_, err := s.client.SourcesStore.Get(ctx, d.ID)
if err != nil {
return err
}
return s.client.SourcesStore.Update(ctx, d)
}

308
bolt/organizations_sources_test.go Normal file
View File

@ -0,0 +1,308 @@
package bolt_test
import (
"context"
"testing"
"github.com/google/go-cmp/cmp"
"github.com/google/go-cmp/cmp/cmpopts"
"github.com/influxdata/chronograf"
)
// IgnoreFields is used because ID is created by BoltDB and cannot be predicted reliably
// EquateEmpty is used because we want nil slices, arrays, and maps to be equal to the empty map
var sourceCmpOptions = cmp.Options{
cmpopts.EquateEmpty(),
cmpopts.IgnoreFields(chronograf.Source{}, "ID"),
cmpopts.IgnoreFields(chronograf.Source{}, "Default"),
}
func TestOrganizationSources_All(t *testing.T) {
type args struct {
organization string
ctx context.Context
}
tests := []struct {
name string
args args
want []chronograf.Source
wantRaw []chronograf.Source
addFirst bool
wantErr bool
}{
{
name: "No Sources",
wantErr: true,
},
{
name: "All Sources",
args: args{
organization: "1337",
ctx: context.Background(),
},
want: []chronograf.Source{
{
Name: "howdy",
Organization: "1337",
},
},
wantRaw: []chronograf.Source{
{
Name: "howdy",
Organization: "1337",
},
{
Name: "doody",
Organization: "1338",
},
},
addFirst: true,
},
}
for _, tt := range tests {
client, err := NewTestClient()
if err != nil {
t.Fatal(err)
}
if err := client.Open(context.TODO()); err != nil {
t.Fatal(err)
}
defer client.Close()
s := client.OrganizationSourcesStore
if tt.addFirst {
for _, d := range tt.wantRaw {
client.SourcesStore.Add(tt.args.ctx, d)
}
}
tt.args.ctx = context.WithValue(tt.args.ctx, "organizationID", tt.args.organization)
gots, err := s.All(tt.args.ctx)
if (err != nil) != tt.wantErr {
t.Errorf("%q. OrganizationSourcesStore.All() error = %v, wantErr %v", tt.name, err, tt.wantErr)
continue
}
for i, got := range gots {
if diff := cmp.Diff(got, tt.want[i], sourceCmpOptions...); diff != "" {
t.Errorf("%q. OrganizationSourcesStore.All():\n-got/+want\ndiff %s", tt.name, diff)
}
}
}
}
func TestOrganizationSources_Add(t *testing.T) {
type args struct {
organization string
ctx context.Context
source chronograf.Source
}
tests := []struct {
name string
args args
want chronograf.Source
wantErr bool
}{
{
name: "Add Source",
args: args{
organization: "1337",
ctx: context.Background(),
source: chronograf.Source{
Name: "howdy",
},
},
want: chronograf.Source{
Name: "howdy",
Organization: "1337",
},
},
}
for _, tt := range tests {
client, err := NewTestClient()
if err != nil {
t.Fatal(err)
}
if err := client.Open(context.TODO()); err != nil {
t.Fatal(err)
}
defer client.Close()
s := client.OrganizationSourcesStore
tt.args.ctx = context.WithValue(tt.args.ctx, "organizationID", tt.args.organization)
d, err := s.Add(tt.args.ctx, tt.args.source)
if (err != nil) != tt.wantErr {
t.Errorf("%q. OrganizationSourcesStore.Add() error = %v, wantErr %v", tt.name, err, tt.wantErr)
continue
}
got, err := s.Get(tt.args.ctx, d.ID)
if diff := cmp.Diff(got, tt.want, sourceCmpOptions...); diff != "" {
t.Errorf("%q. OrganizationSourcesStore.Add():\n-got/+want\ndiff %s", tt.name, diff)
}
}
}
func TestOrganizationSources_Delete(t *testing.T) {
type args struct {
organization string
ctx context.Context
source chronograf.Source
}
tests := []struct {
name string
args args
want []chronograf.Source
addFirst bool
wantErr bool
}{
{
name: "Delete source",
args: args{
organization: "1337",
ctx: context.Background(),
source: chronograf.Source{
Name: "howdy",
Organization: "1337",
},
},
addFirst: true,
},
}
for _, tt := range tests {
client, err := NewTestClient()
if err != nil {
t.Fatal(err)
}
if err := client.Open(context.TODO()); err != nil {
t.Fatal(err)
}
defer client.Close()
s := client.OrganizationSourcesStore
if tt.addFirst {
tt.args.source, _ = client.SourcesStore.Add(tt.args.ctx, tt.args.source)
}
tt.args.ctx = context.WithValue(tt.args.ctx, "organizationID", tt.args.organization)
err = s.Delete(tt.args.ctx, tt.args.source)
if (err != nil) != tt.wantErr {
t.Errorf("%q. OrganizationSourcesStore.All() error = %v, wantErr %v", tt.name, err, tt.wantErr)
continue
}
}
}
func TestOrganizationSources_Get(t *testing.T) {
type args struct {
organization string
ctx context.Context
source chronograf.Source
}
tests := []struct {
name string
args args
want chronograf.Source
addFirst bool
wantErr bool
}{
{
name: "Get Source",
args: args{
organization: "1337",
ctx: context.Background(),
source: chronograf.Source{
Name: "howdy",
Organization: "1337",
},
},
want: chronograf.Source{
Name: "howdy",
Organization: "1337",
},
addFirst: true,
},
}
for _, tt := range tests {
client, err := NewTestClient()
if err != nil {
t.Fatal(err)
}
if err := client.Open(context.TODO()); err != nil {
t.Fatal(err)
}
defer client.Close()
if tt.addFirst {
tt.args.source, _ = client.SourcesStore.Add(tt.args.ctx, tt.args.source)
}
s := client.OrganizationSourcesStore
tt.args.ctx = context.WithValue(tt.args.ctx, "organizationID", tt.args.organization)
if (err != nil) != tt.wantErr {
t.Errorf("%q. OrganizationSourcesStore.Add() error = %v, wantErr %v", tt.name, err, tt.wantErr)
continue
}
got, err := s.Get(tt.args.ctx, tt.args.source.ID)
if diff := cmp.Diff(got, tt.want, sourceCmpOptions...); diff != "" {
t.Errorf("%q. OrganizationSourcesStore.Add():\n-got/+want\ndiff %s", tt.name, diff)
}
}
}
func TestOrganizationSources_Update(t *testing.T) {
type args struct {
organization string
ctx context.Context
source chronograf.Source
name string
}
tests := []struct {
name string
args args
want chronograf.Source
addFirst bool
wantErr bool
}{
{
name: "Update Source Name",
args: args{
organization: "1337",
ctx: context.Background(),
source: chronograf.Source{
Name: "howdy",
Organization: "1337",
},
name: "doody",
},
want: chronograf.Source{
Name: "doody",
Organization: "1337",
},
addFirst: true,
},
}
for _, tt := range tests {
client, err := NewTestClient()
if err != nil {
t.Fatal(err)
}
if err := client.Open(context.TODO()); err != nil {
t.Fatal(err)
}
defer client.Close()
if tt.addFirst {
tt.args.source, _ = client.SourcesStore.Add(tt.args.ctx, tt.args.source)
}
if tt.args.name != "" {
tt.args.source.Name = tt.args.name
}
s := client.OrganizationSourcesStore
tt.args.ctx = context.WithValue(tt.args.ctx, "organizationID", tt.args.organization)
err = s.Update(tt.args.ctx, tt.args.source)
if (err != nil) != tt.wantErr {
t.Errorf("%q. OrganizationSourcesStore.Update() error = %v, wantErr %v", tt.name, err, tt.wantErr)
continue
}
got, err := s.Get(tt.args.ctx, tt.args.source.ID)
if diff := cmp.Diff(got, tt.want, sourceCmpOptions...); diff != "" {
t.Errorf("%q. OrganizationSourcesStore.Update():\n-got/+want\ndiff %s", tt.name, diff)
}
}
}