cleaning stuff up
Change-Id: Ib7e91c6e92b6f6b3c93f4d8ca5dcbfcf74f6c671
diff --git a/api/handler_test.go b/api/handler_test.go
index 50c5355..c319e6d 100644
--- a/api/handler_test.go
+++ b/api/handler_test.go
@@ -22,6 +22,7 @@
"gerrit.opencord.org/abstract-olt/api"
"gerrit.opencord.org/abstract-olt/models/abstract"
+ "gerrit.opencord.org/abstract-olt/models/physical"
"golang.org/x/net/context"
)
@@ -40,7 +41,6 @@
t.Fatalf("CreateChassis failed %v\n", err)
}
clli = ret.DeviceID
-
}
func TestHandler_CreateOLTChassis(t *testing.T) {
fmt.Println("in handlerTest_CreateChassis")
@@ -51,6 +51,30 @@
t.Fatalf("CreateOLTChassis failed %v\n", err)
}
fmt.Printf("CreateOLTChassis success %v\n", ret)
+}
+func TestHandler_EnableSlot(t *testing.T) {
+ ctx = context.Background()
+ server = api.Server{}
+ fmt.Println("in handler_test_EnableSlot")
+ // slot number 1 should be provisioned above
+ message := &api.ActivateSlotMessage{CLLI: clli, SlotNumber: 1}
+ ret, err := server.EnableSlot(ctx, message)
+ if err != nil {
+ t.Fatalf("EnableSlot failed with %v\n", err)
+ }
+ fmt.Printf("EnableSlot succeeded with %v\n", ret)
+ // Slot 2 isn't provisioned and should fail
+ message = &api.ActivateSlotMessage{CLLI: clli, SlotNumber: 2}
+ ret, err = server.EnableSlot(ctx, message)
+ if err != nil {
+ switch err.(type) {
+ case *physical.UnprovisionedSlotError:
+ fmt.Printf("EnableSlot failed as it should with %v\n", err)
+ default:
+ t.Fatalf("EnableSlot failed with %v\n", err)
+ }
+ t.Fatalf("EnableSlot should have failed but didn't")
+ }
}
func TestHandler_ProvisionOnt(t *testing.T) {
@@ -65,6 +89,7 @@
}
// this one should fail
fmt.Println("here")
+ //SlotNumber 1 hasn't been provisioned
message = &api.AddOntMessage{CLLI: clli, SlotNumber: 2, PortNumber: 3, OntNumber: 2, SerialNumber: "2033029402"}
ret, err = server.ProvisionOnt(ctx, message)
if err != nil {
@@ -78,5 +103,4 @@
t.Fatalf("ProvsionOnt should have failed but didn't")
}
fmt.Printf("ProvisionOnt success %v\n", ret)
-
}