[trafficcontrol] branch master updated: Added unit tests for cdn folder (#7476)

[sr...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

srijeet0406 pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/trafficcontrol.git


The following commit(s) were added to refs/heads/master by this push:
     new 29a6832dd6 Added unit tests for cdn folder (#7476)
29a6832dd6 is described below

commit 29a6832dd633621a5211f6bb5439d6bbb4fd91ef
Author: Rima Shah <22...@users.noreply.github.com>
AuthorDate: Fri May 5 08:45:30 2023 -0600

    Added unit tests for cdn folder (#7476)
    
    * Added two new test files in cdn folder.
    
    * Added tests
    
    * Added tests-1
---
 .../traffic_ops_golang/cdn/capacity_test.go        | 12 ++--
 .../traffic_ops_golang/cdn/dnssecrefresh_test.go   | 64 +++++++++++++++++
 traffic_ops/traffic_ops_golang/cdn/genksk_test.go  | 84 ++++++++++++++++++++++
 3 files changed, 154 insertions(+), 6 deletions(-)

diff --git a/traffic_ops/traffic_ops_golang/cdn/capacity_test.go b/traffic_ops/traffic_ops_golang/cdn/capacity_test.go
index 74e56f24ec..f1bcc85431 100644
--- a/traffic_ops/traffic_ops_golang/cdn/capacity_test.go
+++ b/traffic_ops/traffic_ops_golang/cdn/capacity_test.go
@@ -1,11 +1,5 @@
 package cdn
 
-import (
-	"testing"
-
-	"github.com/apache/trafficcontrol/lib/go-tc"
-)
-
 /*
  * Licensed to the Apache Software Foundation (ASF) under one
  * or more contributor license agreements.  See the NOTICE file
@@ -25,6 +19,12 @@ import (
  * under the License.
  */
 
+import (
+	"testing"
+
+	"github.com/apache/trafficcontrol/lib/go-tc"
+)
+
 func TestGetStatsFromServiceInterface(t *testing.T) {
 	data1 := tc.ServerStats{
 		Interfaces: nil,
diff --git a/traffic_ops/traffic_ops_golang/cdn/dnssecrefresh_test.go b/traffic_ops/traffic_ops_golang/cdn/dnssecrefresh_test.go
new file mode 100644
index 0000000000..9e8cc46b8e
--- /dev/null
+++ b/traffic_ops/traffic_ops_golang/cdn/dnssecrefresh_test.go
@@ -0,0 +1,64 @@
+package cdn
+
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+import (
+	"testing"
+
+	"github.com/jmoiron/sqlx"
+
+	"gopkg.in/DATA-DOG/go-sqlmock.v1"
+)
+
+func TestGetDNSSECKeyRefreshParams_test(t *testing.T) {
+	mockDB, mock, err := sqlmock.New()
+	if err != nil {
+		t.Fatalf("an error '%s' was not expected when opening a stub database connection", err)
+	}
+	defer mockDB.Close()
+
+	db := sqlx.NewDb(mockDB, "sqlmock")
+	defer db.Close()
+
+	cols := []string{"cdn_name", "cdn_domain", "cdn_dnssec_enabled", "parameter_name", "parameter_value"}
+	rows := sqlmock.NewRows(cols)
+	rows.AddRow("test", "test.com", false, "", "")
+
+	mock.ExpectBegin()
+	mock.ExpectQuery("WITH cdn_profile_ids").WillReturnRows(rows)
+	mock.ExpectCommit()
+
+	params, err := getDNSSECKeyRefreshParams(db.MustBegin().Tx)
+	for _, v := range params {
+		if v.CDNName != "test" {
+			t.Errorf("Expected cdn name: test, got: %s", v.CDNName)
+		}
+		if v.CDNDomain != "test.com" {
+			t.Errorf("Expected cdn domain: test.com, got: %s", v.CDNDomain)
+		}
+		if v.DNSSECEnabled != false {
+			t.Errorf("Expected DNSSEC to not be enabled")
+		}
+		if err != nil {
+			t.Errorf("expected no error, got: %s", err)
+		}
+	}
+
+}
diff --git a/traffic_ops/traffic_ops_golang/cdn/genksk_test.go b/traffic_ops/traffic_ops_golang/cdn/genksk_test.go
new file mode 100644
index 0000000000..25e2f22604
--- /dev/null
+++ b/traffic_ops/traffic_ops_golang/cdn/genksk_test.go
@@ -0,0 +1,84 @@
+package cdn
+
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+import (
+	"testing"
+
+	"github.com/jmoiron/sqlx"
+
+	"gopkg.in/DATA-DOG/go-sqlmock.v1"
+)
+
+func TestGetKSKParams(t *testing.T) {
+	mockDB, mock, err := sqlmock.New()
+	if err != nil {
+		t.Fatalf("an error '%s' was not expected when opening a stub database connection", err)
+	}
+	defer mockDB.Close()
+
+	db := sqlx.NewDb(mockDB, "sqlmock")
+	defer db.Close()
+
+	rows := sqlmock.NewRows([]string{"name", "value"})
+	rows.AddRow("test", "2")
+	mock.ExpectBegin()
+	mock.ExpectQuery("WITH cdn_profile_id").WithArgs("test").WillReturnRows(rows)
+	mock.ExpectCommit()
+
+	ttl, mult, err := getKSKParams(db.MustBegin().Tx, "test")
+	if ttl != nil {
+		t.Errorf("expected: nil, got: %v", ttl)
+	}
+	if *mult != 2 {
+		t.Errorf("expected: 2, got: %v", *mult)
+	}
+	if err != nil {
+		t.Errorf("%s", err)
+	}
+}
+
+func TestGetKSKParamsDNSKey(t *testing.T) {
+	mockDB, mock, err := sqlmock.New()
+	if err != nil {
+		t.Fatalf("an error '%s' was not expected when opening a stub database connection", err)
+	}
+	defer mockDB.Close()
+
+	db := sqlx.NewDb(mockDB, "sqlmock")
+	defer db.Close()
+
+	rows1 := sqlmock.NewRows([]string{"name", "value"})
+	rows1.AddRow("tld.ttls.DNSKEY", "5")
+	mock.ExpectBegin()
+	mock.ExpectQuery("WITH cdn_profile_id").WithArgs("test").WillReturnRows(rows1)
+	mock.ExpectCommit()
+
+	ttl1, mult1, err1 := getKSKParams(db.MustBegin().Tx, "test")
+	if *ttl1 != 5 {
+		t.Errorf("expected: 5, got: %v", *ttl1)
+	}
+	if mult1 != nil {
+		t.Errorf("expected: nil, got: %v", mult1)
+	}
+	if err1 != nil {
+		t.Errorf("%s", err1)
+	}
+}